graphene/examples/starwars_relay/tests/test_objectidentification.py

173 lines
3.0 KiB
Python
Raw Normal View History

from ..data import setup
from ..schema import schema
2015-10-27 09:54:51 +03:00
setup()
2015-09-26 09:25:10 +03:00
2015-10-03 08:17:51 +03:00
def test_str_schema():
assert str(schema) == '''schema {
query: Query
mutation: Mutation
}
type Faction implements Node {
id: ID!
name: String
ships(before: String, after: String, first: Int, last: Int): ShipConnection
}
input IntroduceShipInput {
shipName: String!
factionId: String!
clientMutationId: String
}
type IntroduceShipPayload {
ship: Ship
faction: Faction
clientMutationId: String
}
type Mutation {
introduceShip(input: IntroduceShipInput!): IntroduceShipPayload
}
interface Node {
id: ID!
}
type PageInfo {
hasNextPage: Boolean!
hasPreviousPage: Boolean!
startCursor: String
endCursor: String
}
type Query {
rebels: Faction
empire: Faction
node(id: ID!): Node
}
type Ship implements Node {
id: ID!
name: String
}
type ShipConnection {
pageInfo: PageInfo!
2016-10-27 20:35:31 +03:00
edges: [ShipEdge]!
}
type ShipEdge {
node: Ship
cursor: String!
}
'''
2015-09-26 09:25:10 +03:00
def test_correctly_fetches_id_name_rebels():
query = '''
query RebelsQuery {
rebels {
id
name
}
}
'''
expected = {
2015-10-03 08:17:51 +03:00
'rebels': {
'id': 'RmFjdGlvbjox',
'name': 'Alliance to Restore the Republic'
}
2015-09-26 09:25:10 +03:00
}
result = schema.execute(query)
assert not result.errors
2015-09-26 09:25:10 +03:00
assert result.data == expected
2015-10-03 08:17:51 +03:00
2015-09-26 09:25:10 +03:00
def test_correctly_refetches_rebels():
query = '''
query RebelsRefetchQuery {
node(id: "RmFjdGlvbjox") {
id
... on Faction {
name
}
}
}
'''
expected = {
2015-10-03 08:17:51 +03:00
'node': {
'id': 'RmFjdGlvbjox',
'name': 'Alliance to Restore the Republic'
}
2015-09-26 09:25:10 +03:00
}
result = schema.execute(query)
assert not result.errors
2015-09-26 09:25:10 +03:00
assert result.data == expected
2015-10-03 08:17:51 +03:00
2015-09-26 09:25:10 +03:00
def test_correctly_fetches_id_name_empire():
query = '''
query EmpireQuery {
empire {
id
name
}
}
'''
expected = {
2015-10-03 08:17:51 +03:00
'empire': {
'id': 'RmFjdGlvbjoy',
'name': 'Galactic Empire'
}
2015-09-26 09:25:10 +03:00
}
result = schema.execute(query)
assert not result.errors
2015-09-26 09:25:10 +03:00
assert result.data == expected
2015-10-03 08:17:51 +03:00
2015-09-26 09:25:10 +03:00
def test_correctly_refetches_empire():
query = '''
query EmpireRefetchQuery {
node(id: "RmFjdGlvbjoy") {
id
... on Faction {
name
}
}
}
'''
expected = {
2015-10-03 08:17:51 +03:00
'node': {
'id': 'RmFjdGlvbjoy',
'name': 'Galactic Empire'
}
2015-09-26 09:25:10 +03:00
}
result = schema.execute(query)
assert not result.errors
2015-09-26 09:25:10 +03:00
assert result.data == expected
2015-10-03 08:17:51 +03:00
2015-09-26 09:25:10 +03:00
def test_correctly_refetches_xwing():
query = '''
query XWingRefetchQuery {
node(id: "U2hpcDox") {
id
... on Ship {
name
}
}
}
'''
expected = {
2015-10-03 08:17:51 +03:00
'node': {
'id': 'U2hpcDox',
'name': 'X-Wing'
}
2015-09-26 09:25:10 +03:00
}
result = schema.execute(query)
assert not result.errors
2015-09-26 09:25:10 +03:00
assert result.data == expected