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