From ea06ea88e2b746c3db12a6c3ebae136ce184bef5 Mon Sep 17 00:00:00 2001 From: Luna Date: Mon, 13 Sep 2021 23:52:27 -0300 Subject: [PATCH] tests: add test for 'before' and 'after' params --- tests/test_messages.py | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) diff --git a/tests/test_messages.py b/tests/test_messages.py index 535eadb..4494c18 100644 --- a/tests/test_messages.py +++ b/tests/test_messages.py @@ -45,3 +45,26 @@ async def test_message_listing(test_cli_user): fetched_ids = [m["id"] for m in rjson] for message in messages: assert str(message.id) in fetched_ids + + # assert all messages are below given id if its on 'before' param + + resp = await test_cli_user.get( + f"/api/v6/channels/{channel.id}/messages", + query_string={"before": middle_message_id}, + ) + assert resp.status_code == 200 + rjson = await resp.json + + for message_json in rjson: + assert int(message_json["id"]) <= middle_message_id + + # assert all message are above given id if its on 'after' param + resp = await test_cli_user.get( + f"/api/v6/channels/{channel.id}/messages", + query_string={"after": middle_message_id}, + ) + assert resp.status_code == 200 + rjson = await resp.json + + for message_json in rjson: + assert int(message_json["id"]) >= middle_message_id