File: test_disable_introspection.py

package info (click to toggle)
python-graphene 3.4.3-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 1,120 kB
  • sloc: python: 8,935; makefile: 214; sh: 18
file content (37 lines) | stat: -rw-r--r-- 842 bytes parent folder | download
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
from graphql import parse, validate

from ...types import Schema, ObjectType, String
from ..disable_introspection import DisableIntrospection


class Query(ObjectType):
    name = String(required=True)

    @staticmethod
    def resolve_name(root, info):
        return "Hello world!"


schema = Schema(query=Query)


def run_query(query: str):
    document = parse(query)

    return validate(
        schema=schema.graphql_schema,
        document_ast=document,
        rules=(DisableIntrospection,),
    )


def test_disallows_introspection_queries():
    errors = run_query("{ __schema { queryType { name } } }")

    assert len(errors) == 1
    assert errors[0].message == "Cannot query '__schema': introspection is disabled."


def test_allows_non_introspection_queries():
    errors = run_query("{ name }")
    assert len(errors) == 0