Page MenuHomeSoftware Heritage
Files F9697744

test_pagination.py
No OneTemporary
Actions

test_pagination.py
View Options

# Copyright (C) 2022 The Software Heritage developers
# See the AUTHORS file at the top-level directory of this distribution
# License: GNU General Public License version 3, or any later version
# See top-level LICENSE file for more information
from ..data import get_origins
from .utils import get_query_response
# Using Origin object to run functional tests for pagination
def get_origin_nodes(client, first=1, after=""):
query_str = """
{
origins(first: %s, %s) {
nodes {
id
}
pageInfo {
hasNextPage
endCursor
}
}
}
""" % (
first,
after,
)
return get_query_response(client, query_str)
def test_pagination(client):
# requesting the max number of nodes available
# endCursor must be None
data, _ = get_origin_nodes(client, len(get_origins()))
assert len(data["origins"]["nodes"]) == len(get_origins())
assert data["origins"]["pageInfo"] == {"hasNextPage": False, "endCursor": None}
def test_first_arg(client):
data, _ = get_origin_nodes(client, 1)
assert len(data["origins"]["nodes"]) == 1
assert data["origins"]["pageInfo"]["hasNextPage"] is True
def test_invalid_first_arg(client):
data, errors = get_origin_nodes(client, -1)
assert data["origins"] is None
assert (len(errors)) == 2 # one error for origins and anotehr one for pageInfo
assert (
errors[0]["message"]
== "Pagination error: Value for argument 'first' is invalid; it must be between 0 and 1000" # noqa: B950
)
def test_too_big_first_arg(client):
data, errors = get_origin_nodes(client, 1001) # max page size is 1000
assert data["origins"] is None
assert (len(errors)) == 2
assert (
errors[0]["message"]
== "Pagination error: Value for argument 'first' is invalid; it must be between 0 and 1000" # noqa: B950
)
def test_after_arg(client):
first_data, _ = get_origin_nodes(client)
end_cursor = first_data["origins"]["pageInfo"]["endCursor"]
# get again with endcursor as the after argument
data, _ = get_origin_nodes(client, 1, f'after: "{end_cursor}"')
assert len(data["origins"]["nodes"]) == 1
assert data["origins"]["pageInfo"] == {"hasNextPage": False, "endCursor": None}
def test_invalid_after_arg(client):
data, errors = get_origin_nodes(client, 1, 'after: "invalid"')
assert data["origins"] is None
assert (len(errors)) == 2
assert (
errors[0]["message"] == "Pagination error: Invalid value for argument 'after'"
)
def test_edge_cursor(client):
origins = get_origin_nodes(client)[0]["origins"]
# end cursor here must be the item cursor for the second item
end_cursor = origins["pageInfo"]["endCursor"]
query_str = (
"""
{
origins(first: 1, after: "%s") {
edges {
cursor
node {
id
}
}
nodes {
id
}
}
}
"""
% end_cursor
)
data, _ = get_query_response(client, query_str)
origins = data["origins"]
assert [edge["node"] for edge in origins["edges"]] == origins["nodes"]
assert origins["edges"][0]["cursor"] == end_cursor

File Metadata

Mime Type
text/x-python
Expires
Tue, Aug 19, 12:46 AM (3 w, 1 d ago)
Storage Engine
blob
Storage Format
Raw Data
Storage Handle
3330330

Event Timeline

About · Developer information · Legal