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
|
from dataclasses import dataclass
from graphql import graphql_sync
from graphql.utilities import print_schema
from apischema.graphql import graphql_schema
from apischema.tagged_unions import Tagged, TaggedUnion
@dataclass
class Bar:
field: str
class Foo(TaggedUnion):
bar: Tagged[Bar]
baz: Tagged[int]
def query(foo: Foo) -> Foo:
return foo
schema = graphql_schema(query=[query])
schema_str = """\
type Query {
query(foo: FooInput!): Foo!
}
type Foo {
bar: Bar
baz: Int
}
type Bar {
field: String!
}
input FooInput {
bar: BarInput
baz: Int
}
input BarInput {
field: String!
}"""
assert print_schema(schema) == schema_str
query_str = """
{
query(foo: {bar: {field: "value"}}) {
bar {
field
}
baz
}
}"""
assert graphql_sync(schema, query_str).data == {
"query": {"bar": {"field": "value"}, "baz": None}
}
|
