def test_get_query(db, factory):
schema = build_graphql_schema()
movie_id = factory.Movie.create().id
result = execute_query(schema, get_movie_query, variables={'id': movie_id})
assert result.errors is None
movie = result.data['getMovie']
assert movie.get('id') is not None
def test_custom_resolver(db):
schema = build_graphql_schema()
result = execute_query(schema,
update_or_create_store_query,
variables={'name': 'a'})
assert result.errors is None
assert result.data['updateOrCreateStore']['id'] is not None
assert result.data['updateOrCreateStore']['name'] == 'override'
def test_reverse_query(db, factory):
schema = build_graphql_schema()
movies = factory.Movie.create_batch(5)
factory.Movie.create(star=movies[0].star)
result = execute_query(schema, all_actors_query)
assert result.errors is None
assert len(result.data['allActors']) == 5
for actor in result.data['allActors']:
assert actor.get('moviesByStar') is not None
def test_query(db, factory):
schema = build_graphql_schema()
factory.Movie.create_batch(5)
result = execute_query(schema, all_movies_query)
assert result.errors is None
assert len(result.data['allMovies']) == 5
for movie in result.data['allMovies']:
assert movie.get('star') is not None
assert movie['star'].get('address') is not None
def test_create_reverse(db):
schema = build_graphql_schema()
result = execute_query(schema, create_actor_and_movies_query)
assert result.errors is None
data = result.data['createActor']
assert data['id'] is not None
assert data['moviesByStar'] is not None
assert len(data['moviesByStar']) == 2
for movie in data['moviesByStar']:
assert movie['id'] is not None
def test_mutation(db, factory):
schema = build_graphql_schema()
factory.User.create(email='test', password='******')
result = execute_query(schema,
authenticate_query, {
'email': 'test',
'password': '******'
},
reraise=True)
assert result.errors is None
assert len(result.data['authenticate']['token']) > 0
def test_delete(db, factory):
schema = build_graphql_schema()
movie = factory.Movie.create()
result = execute_query(schema,
delete_movie_query,
variables={'id': movie.id})
assert result.errors is None
data = result.data['deleteMovie']
assert data['id'] == movie.id
result = Q(movie).filter(id=movie.id).get()
assert result is None
def test_update(db):
schema = build_graphql_schema()
result = execute_query(schema, create_actors_query)
result = execute_query(schema,
update_actors_query,
variables={
'id1': result.data['firstActor']['id'],
'id2': result.data['secondActor']['id']
})
assert result.errors is None
assert result.data['firstActor']['age'] == 60
assert result.data['secondActor']['age'] == 80
def test_update_or_create(db):
schema = build_graphql_schema()
result = execute_query(schema,
update_or_create_actor_query,
variables={'firstName': 'a'})
assert result.errors is None
assert result.data['updateOrCreateActor']['id'] is not None
assert result.data['updateOrCreateActor']['firstName'] == 'a'
result = execute_query(schema,
update_or_create_actor_query,
variables={
'id': result.data['updateOrCreateActor']['id'],
'firstName': 'b'
})
assert result.errors is None
assert result.data['updateOrCreateActor']['firstName'] == 'b'
def test_omit():
schema = build_graphql_schema()
# print_schema(schema)
omitted_types = ('AllOmitted', )
for type in omitted_types:
assert type not in schema.type_map
queries = schema.type_map['Query']
omitted_queries = ('allAllOmitted', 'getAllOmitted', 'allListOmitted',
'getGetOmitted')
for query in omitted_queries:
assert query not in queries.fields
mutations = schema.type_map['Mutation']
omitted_mutations = ('createAllOmitted', 'updateAllOmitted',
'createCreateOmitted', 'updateUpdateOmitted')
for mutation in omitted_mutations:
assert mutation not in mutations.fields
def test_set_role(db):
# TOOD: Really need to check the kind of error returned.
schema = build_graphql_schema()
result = execute_query(
schema,
all_addresses_query,
context={'session': Session(DefaultRole.Meta.dbrole)})
assert len(result.errors) > 0
result = execute_query(schema,
all_addresses_query,
context={'session': Session(UserRole.Meta.dbrole)})
assert result.errors is None
result = execute_query(
schema,
get_address_query,
context={'session': Session(DefaultRole.Meta.dbrole)})
assert len(result.errors) > 0
def test_create(db):
schema = build_graphql_schema()
result = execute_query(schema, create_actors_query)
assert result.errors is None
assert result.data['firstActor'] is not None
assert result.data['secondActor'] is not None
def test_schema():
schema = build_graphql_schema()