File: test_no_schema_introspection.py

package info (click to toggle)
graphql-core 3.2.6-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 5,384 kB
  • sloc: python: 45,812; makefile: 26; sh: 13
file content (145 lines) | stat: -rw-r--r-- 3,950 bytes parent folder | download | duplicates (2)
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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
from functools import partial

from graphql.utilities import build_schema
from graphql.validation import NoSchemaIntrospectionCustomRule

from .harness import assert_validation_errors

schema = build_schema(
    """
    type Query {
      someQuery: SomeType
    }

    type SomeType {
      someField: String
      introspectionField: __EnumValue
    }
    """
)

assert_errors = partial(
    assert_validation_errors, NoSchemaIntrospectionCustomRule, schema=schema
)

assert_valid = partial(assert_errors, errors=[])


def describe_validate_prohibit_introspection_queries():
    def ignores_valid_fields_including_typename():
        assert_valid(
            """
            {
              someQuery {
                __typename
                someField
              }
            }
            """
        )

    def ignores_fields_not_in_the_schema():
        assert_valid(
            """
            {
              __introspect
            }
            """
        )

    def reports_error_when_a_field_with_an_introspection_type_is_requested():
        assert_errors(
            """
            {
              __schema {
                queryType {
                  name
                }
              }
            }
            """,
            [
                {
                    "message": "GraphQL introspection has been disabled,"
                    " but the requested query contained the field '__schema'.",
                    "locations": [(3, 15)],
                },
                {
                    "message": "GraphQL introspection has been disabled,"
                    " but the requested query contained the field 'queryType'.",
                    "locations": [(4, 17)],
                },
            ],
        )

    def reports_error_when_a_field_with_introspection_type_is_requested_and_aliased():
        assert_errors(
            """
            {
              s: __schema {
                queryType {
                  name
                }
              }
            }
            """,
            [
                {
                    "message": "GraphQL introspection has been disabled,"
                    " but the requested query contained the field '__schema'.",
                    "locations": [(3, 15)],
                },
                {
                    "message": "GraphQL introspection has been disabled,"
                    " but the requested query contained the field 'queryType'.",
                    "locations": [(4, 17)],
                },
            ],
        )

    def reports_error_when_using_a_fragment_with_a_field_with_an_introspection_type():
        assert_errors(
            """
            {
              ...QueryFragment
            }

            fragment QueryFragment on Query {
              __schema {
                queryType {
                  name
                }
              }
            }
            """,
            [
                {
                    "message": "GraphQL introspection has been disabled,"
                    " but the requested query contained the field '__schema'.",
                    "locations": [(7, 15)],
                },
                {
                    "message": "GraphQL introspection has been disabled,"
                    " but the requested query contained the field 'queryType'.",
                    "locations": [(8, 17)],
                },
            ],
        )

    def reports_error_for_non_standard_introspection_fields():
        assert_errors(
            """
            {
              someQuery {
                introspectionField
              }
            }
            """,
            [
                {
                    "message": "GraphQL introspection has been disabled, but"
                    " the requested query contained the field 'introspectionField'.",
                    "locations": [(4, 17)],
                },
            ],
        )