diff --git a/tests/test_search.py b/tests/test_search.py index 32743f0a8cd30ac0e562c659f52bb476d0734e59..a77314f3e9489b4725e3b570dd830cb570467400 100644 --- a/tests/test_search.py +++ b/tests/test_search.py @@ -1,7 +1,7 @@ import pytest from openlobby.core.api.paginator import Paginator, encode_cursor -from openlobby.core.search import query_reports +from openlobby.core.search import query_reports, reports_by_author from .dummy import prepare_reports @@ -42,3 +42,23 @@ def test_query_reports__pagination(first, after, expected_ids): paginator = Paginator(first=first, after=after) response = query_reports(query, paginator) assert expected_ids == [int(r.meta.id) for r in response] + + +def test_reports_by_author(): + prepare_reports() + author_id = 1 + paginator = Paginator() + response = reports_by_author(author_id, paginator) + assert [3, 1] == [int(r.meta.id) for r in response] + + +@pytest.mark.parametrize('first, after, expected_ids', [ + (1, None, [3]), + (1, encode_cursor(1), [1]), +]) +def test_reports_by_author__pagination(first, after, expected_ids): + prepare_reports() + author_id = 1 + paginator = Paginator(first=first, after=after) + response = reports_by_author(author_id, paginator) + assert expected_ids == [int(r.meta.id) for r in response]