def test_is_type_of_used_to_resolve_runtime_type_for_union():
DogType = GraphQLObjectType(
name='Dog',
is_type_of=is_type_of(Dog),
fields={
'name': GraphQLField(GraphQLString),
'woofs': GraphQLField(GraphQLBoolean)
}
)
CatType = GraphQLObjectType(
name='Cat',
is_type_of=is_type_of(Cat),
fields={
'name': GraphQLField(GraphQLString),
'meows': GraphQLField(GraphQLBoolean)
}
)
PetType = GraphQLUnionType(
name='Pet',
types=[CatType, DogType]
)
schema = GraphQLSchema(
query=GraphQLObjectType(
name='Query',
fields={
'pets': GraphQLField(
GraphQLList(PetType),
resolver=lambda *_: [Dog('Odie', True), Cat('Garfield', False)]
)
}
),
types=[CatType, DogType]
)
query = '''
{
pets {
... on Dog {
name
woofs
}
... on Cat {
name
meows
}
}
}
'''
result = graphql(schema, query)
assert not result.errors
assert result.data == {'pets': [{'woofs': True, 'name': 'Odie'}, {'name': 'Garfield', 'meows': False}]}
def test_is_type_of_used_to_resolve_runtime_type_for_union():
# type: () -> None
DogType = GraphQLObjectType(
name="Dog",
is_type_of=is_type_of(Dog),
fields={
"name": GraphQLField(GraphQLString),
"woofs": GraphQLField(GraphQLBoolean),
},
)
CatType = GraphQLObjectType(
name="Cat",
is_type_of=is_type_of(Cat),
fields={
"name": GraphQLField(GraphQLString),
"meows": GraphQLField(GraphQLBoolean),
},
)
PetType = GraphQLUnionType(name="Pet", types=[CatType, DogType])
schema = GraphQLSchema(
query=GraphQLObjectType(
name="Query",
fields={
"pets": GraphQLField(
GraphQLList(PetType),
resolver=lambda *_: [Dog("Odie", True), Cat("Garfield", False)],
)
},
),
types=[CatType, DogType],
)
query = """
{
pets {
... on Dog {
name
woofs
}
... on Cat {
name
meows
}
}
}
"""
result = graphql(schema, query)
assert not result.errors
assert result.data == {
"pets": [{"woofs": True, "name": "Odie"}, {"name": "Garfield", "meows": False}]
}
def test_resolve_type_on_union_yields_useful_error():
DogType = GraphQLObjectType(name='Dog',
fields={
'name': GraphQLField(GraphQLString),
'woofs': GraphQLField(GraphQLBoolean)
})
HumanType = GraphQLObjectType(name='Human',
fields={
'name': GraphQLField(GraphQLString),
})
CatType = GraphQLObjectType(name='Cat',
fields={
'name': GraphQLField(GraphQLString),
'meows': GraphQLField(GraphQLBoolean)
})
PetType = GraphQLUnionType(
name='Pet',
types=[DogType, CatType],
resolve_type=make_type_resolver(
lambda: [(Dog, DogType), (Cat, CatType), (Human, HumanType)]))
schema = GraphQLSchema(
query=GraphQLObjectType(name='Query',
fields={
'pets':
GraphQLField(GraphQLList(PetType),
resolver=lambda *_: [
Dog('Odie', True),
Cat('Garfield', False),
Human('Jon')
])
}))
query = '''
{
pets {
... on Dog {
name
woofs
}
... on Cat {
name
meows
}
}
}
'''
result = graphql(schema, query)
assert result.errors[
0].message == 'Runtime Object type "Human" is not a possible type for "Pet".'
assert result.data == {
'pets': [{
'woofs': True,
'name': 'Odie'
}, {
'name': 'Garfield',
'meows': False
}, None]
}
def test_resolve_type_on_union_yields_useful_error():
# type: () -> None
DogType = GraphQLObjectType(
name="Dog",
fields={
"name": GraphQLField(GraphQLString),
"woofs": GraphQLField(GraphQLBoolean),
},
)
HumanType = GraphQLObjectType(
name="Human", fields={"name": GraphQLField(GraphQLString)}
)
CatType = GraphQLObjectType(
name="Cat",
fields={
"name": GraphQLField(GraphQLString),
"meows": GraphQLField(GraphQLBoolean),
},
)
PetType = GraphQLUnionType(
name="Pet",
types=[DogType, CatType],
resolve_type=make_type_resolver(
lambda: [(Dog, DogType), (Cat, CatType), (Human, HumanType)]
),
)
schema = GraphQLSchema(
query=GraphQLObjectType(
name="Query",
fields={
"pets": GraphQLField(
GraphQLList(PetType),
resolver=lambda *_: [
Dog("Odie", True),
Cat("Garfield", False),
Human("Jon"),
],
)
},
)
)
query = """
{
pets {
... on Dog {
name
woofs
}
... on Cat {
name
meows
}
}
}
"""
result = graphql(schema, query)
assert (
result.errors[0].message
== 'Runtime Object type "Human" is not a possible type for "Pet".'
)
assert result.data == {
"pets": [
{"woofs": True, "name": "Odie"},
{"name": "Garfield", "meows": False},
None,
]
}