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
146
147
148
149
150
|
from typing import Any, Dict, Optional
from graphql.execution import ExecutionResult, execute
from graphql.language import parse
from graphql.pyutils import AwaitableOrValue
from graphql.utilities import build_schema
schema = build_schema(
"""
type Query {
test(input: TestInputObject!): TestObject
}
input TestInputObject @oneOf {
a: String
b: Int
}
type TestObject {
a: String
b: Int
}
"""
)
def execute_query(
query: str, root_value: Any, variable_values: Optional[Dict[str, Any]] = None
) -> AwaitableOrValue[ExecutionResult]:
return execute(schema, parse(query), root_value, variable_values=variable_values)
def describe_execute_handles_one_of_input_objects():
def describe_one_of_input_objects():
root_value = {
"test": lambda _info, input: input, # noqa: A002
}
def accepts_a_good_default_value():
query = """
query ($input: TestInputObject! = {a: "abc"}) {
test(input: $input) {
a
b
}
}
"""
result = execute_query(query, root_value)
assert result == ({"test": {"a": "abc", "b": None}}, None)
def rejects_a_bad_default_value():
query = """
query ($input: TestInputObject! = {a: "abc", b: 123}) {
test(input: $input) {
a
b
}
}
"""
result = execute_query(query, root_value)
assert result == (
{"test": None},
[
{
# This type of error would be caught at validation-time
# hence the vague error message here.
"message": "Argument 'input' of non-null type"
" 'TestInputObject!' must not be null.",
"locations": [(3, 31)],
"path": ["test"],
}
],
)
def accepts_a_good_variable():
query = """
query ($input: TestInputObject!) {
test(input: $input) {
a
b
}
}
"""
result = execute_query(query, root_value, {"input": {"a": "abc"}})
assert result == ({"test": {"a": "abc", "b": None}}, None)
def accepts_a_good_variable_with_an_undefined_key():
query = """
query ($input: TestInputObject!) {
test(input: $input) {
a
b
}
}
"""
result = execute_query(query, root_value, {"input": {"a": "abc"}})
assert result == ({"test": {"a": "abc", "b": None}}, None)
def rejects_a_variable_with_multiple_non_null_keys():
query = """
query ($input: TestInputObject!) {
test(input: $input) {
a
b
}
}
"""
result = execute_query(query, root_value, {"input": {"a": "abc", "b": 123}})
assert result == (
None,
[
{
"message": "Variable '$input' got invalid value"
" {'a': 'abc', 'b': 123}; Exactly one key must be specified"
" for OneOf type 'TestInputObject'.",
"locations": [(2, 24)],
}
],
)
def rejects_a_variable_with_multiple_nullable_keys():
query = """
query ($input: TestInputObject!) {
test(input: $input) {
a
b
}
}
"""
result = execute_query(
query, root_value, {"input": {"a": "abc", "b": None}}
)
assert result == (
None,
[
{
"message": "Variable '$input' got invalid value"
" {'a': 'abc', 'b': None}; Exactly one key must be specified"
" for OneOf type 'TestInputObject'.",
"locations": [(2, 24)],
}
],
)
|
