1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
|
from pathlib import Path
import pytest
from graphql import parse
from pytest_codspeed.plugin import BenchmarkFixture
from strawberry import Schema
from .api import Query
ROOT = Path(__file__).parent / "queries"
large_query = (ROOT / "large_items.graphql").read_text()
@pytest.mark.benchmark
def test_parse_large_query(benchmark: BenchmarkFixture):
benchmark(parse, large_query)
@pytest.mark.benchmark
@pytest.mark.parametrize(
"use_parser_cache",
[False, True],
ids=["baseline", "with_parser_cache"],
)
def test_execute_large_query(
benchmark: BenchmarkFixture,
use_parser_cache: bool,
):
extension_instances = []
if use_parser_cache:
from strawberry.extensions import ParserCache
extension_instances.append(ParserCache())
schema = Schema(query=Query, extensions=extension_instances)
benchmark(
schema.execute_sync,
large_query,
variable_values={"count": 1},
)
|
