1 files changed, 61 insertions, 0 deletions
diff --git a/tests/api/test_pagination.py b/tests/api/test_pagination.py
new file mode 100644
index 0000000..67854c3
--- /dev/null
+++ b/tests/api/test_pagination.py
@@ -0,0 +1,61 @@
+import pytest
+from conftest import DB, Response
+
+
+@pytest.fixture
+def query_pagination(schema_execute):
+    query = """
+    query artists($pagination: Pagination) {
+        artists(pagination: $pagination) {
+            __typename
+            count
+            edges {
+                id
+            }
+        }
+    }
+    """
+
+    async def _execute(pagination=None):
+        return await schema_execute(
+            query, {"pagination": pagination} if pagination else None
+        )
+
+    return _execute
+
+
+@pytest.mark.parametrize(
+    "pagination,count,length",
+    [
+        (None, 4, 4),
+        ({"items": 3, "page": 1}, 4, 3),
+        ({"items": 3, "page": 2}, 4, 1),
+        ({"items": 3, "page": 3}, 0, 0),
+        ({"items": 10, "page": 1}, 4, 4),
+        ({"items": 0, "page": 1}, 0, 0),
+        ({"items": 2, "page": 0}, 0, 0),
+        ({"items": -1, "page": 0}, 0, 0),
+        ({"items": 0, "page": -1}, 0, 0),
+    ],
+    ids=[
+        "is missing and lists all",
+        "lists first page",
+        "lists last page",
+        "lists none (no more items)",
+        "lists all",
+        "lists none (zero items)",
+        "lists none (page zero)",
+        "lists none (negative items)",
+        "lists none (negative page)",
+    ],
+)
+@pytest.mark.anyio
+async def test_pagination(query_pagination, gen_artist, pagination, count, length):
+    await DB.add_all(*gen_artist)
+
+    response = Response(await query_pagination(pagination))
+    response.assert_is("ArtistFilterResult")
+
+    assert response.count == count
+    assert isinstance((response.edges), list)
+    assert len(response.edges) == length