Differential D8077 Diff 31369 swh/graphql/tests/functional/test_query_cost.py

Changeset View

Standalone View

swh/graphql/tests/functional/test_query_cost.py

This file was added.

				# 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 . import utils
				from ..data import get_snapshots


				def test_valid_query(client):
				query_str = """
				{
				origins(first: 2) {
				nodes {
				url
				}
				}
				}
				"""
				response, _ = utils.get_query_response(client, query_str)
				assert len(response["origins"]["nodes"]) == 2


				def test_query_cost_simple(client):
				query_str = """
				{
				origins(first: 1000) {
				nodes {
				url
				}
				}
				}
				"""
				errors = utils.get_error_response(client, query_str)
				assert (
				"The query exceeds the maximum cost of 100. Actual cost is 1000"
				in errors[0]["message"]
				)


				def test_query_cost_origin(client):
				query_str = """
				{
				origins(first: 10) {
				nodes {
				url
				latestVisit {
				date
				}
				visits(first: 5) {
				nodes {
				date
				statuses {
				nodes {
				date
				}
				}
				}
				}
				snapshots(first: 5) {
				nodes {
				swhid
				}
				}
				}
				}
				}
				"""
				# Total cost here is 170
				# 10 (origin) + 10 (latestVisit) + 105 (visits) + 10 5 * 3 (status) +
				# 10 * 5*2 (snapshots) = 320
				errors = utils.get_error_response(client, query_str)
				assert (
				"The query exceeds the maximum cost of 100. Actual cost is 320"
				in errors[0]["message"]
				)


				def test_query_cost_snapshots(client):
				query_str = """
				{
				snapshot(swhid: "%s") {
				branches(first: 50) {
				nodes {
				target {
				...on Revision {
				swhid
				}
				...on Directory {
				swhid
				entries(first: 3) {
				nodes {
				targetType
				}
				}
				}
				}
				}
				}
				}
				}
				"""
				# Total cost here is 157
				# 1 (snapshot) + 2 50 (branches) + 50 1 (revision or Directory) + 3 * 2 = 157
				# parent multiplier is not applied when schema introspection is used
				# ie: directory entry connection cost is 3 * 2 and not 50 * 3 * 2
				errors = utils.get_error_response(client, query_str % get_snapshots()[0].swhid())
				assert (
				"The query exceeds the maximum cost of 100. Actual cost is 157"
				in errors[0]["message"]
				)