def test_response_without_filters(session):
add_users(session)
session.commit()
with SQLAlchemyQueryCounter(session, 2):
execution_result = schema.execute(
'{field{edges{node{username}}}}', context={'session': session}
)
assert not execution_result.errors
assert not execution_result.invalid
assert execution_result.data
assert 'field' in execution_result.data
field = execution_result.data['field']
assert 'edges' in field
edges = field['edges']
assert len(edges) == 2
node = edges[0]['node']
assert node == {'username': '******'}
node = edges[1]['node']
assert node == {'username': '******'}
def test_response_with_filters(session):
add_users(session)
session.commit()
request_string = (
'{field(filters:{username: "******"}){edges{node{username balance}}}}'
)
with SQLAlchemyQueryCounter(session, 2):
execution_result = schema.execute(
request_string, context={'session': session}
)
assert not execution_result.errors
assert not execution_result.invalid
assert execution_result.data
assert 'field' in execution_result.data
field = execution_result.data['field']
assert 'edges' in field
edges = field['edges']
assert len(edges) == 1
node = edges[0]['node']
assert node == {'username': '******', 'balance': 0}
def test_nested_response_with_recursive_model(session):
add_groups(session, with_parent_group=True)
session.commit()
request_string = """{
allGroups(filters:{parentGroupIdIsNull: true}){
edges{
node{
subGroups(
filters: {
or: [{name: "group_1"}, {name: "group_2"}]
},
sort: NAME_DESC
){
edges{
node{
name
}
}
}
}
}
}
}"""
with SQLAlchemyQueryCounter(session, 3):
execution_result = schema.execute(
request_string, context={'session': session}
)
assert not execution_result.errors
assert not execution_result.invalid
assert execution_result.data
assert 'allGroups' in execution_result.data
field = execution_result.data['allGroups']
assert 'edges' in field
edges = field['edges']
assert len(edges) == 1
group_0 = edges[0]['node']
group_0_sub_groups_edges = group_0['subGroups']['edges']
assert len(group_0_sub_groups_edges) == 2
sub_group_name = group_0_sub_groups_edges[0]['node']['name']
assert sub_group_name == 'group_2'
def test_nested_response_with_filters(session):
users = add_users(session)
add_users_to_new_groups(session, users)
session.commit()
request_string = """{
field(filters:{username: "******"}){
edges{
node{
username
memberships(filters:{isModerator: true}){
edges{
node{
isModerator
}
}
}
}
}
}
}"""
with SQLAlchemyQueryCounter(session, 3):
execution_result = schema.execute(
request_string, context={'session': session}
)
assert not execution_result.errors
assert not execution_result.invalid
assert execution_result.data
assert 'field' in execution_result.data
field = execution_result.data['field']
assert 'edges' in field
edges = field['edges']
assert len(edges) == 1
user_0 = edges[0]['node']
user_0_memberships_edges = user_0['memberships']['edges']
assert len(user_0_memberships_edges) == 1
is_moderator_value = user_0_memberships_edges[0]['node']['isModerator']
assert is_moderator_value is True
def test_nested_response_without_filters(session):
users = add_users(session)
add_users_to_new_groups(session, users)
session.commit()
request_string = """{
field{
edges{
node{
username
groups{
edges{
node{
name
}
}
}
SomeMemberships: memberships{
edges{
node{
isModerator
}
}
}
createdMemberships{
edges{
node{
isModerator
}
}
}
}
}
}
}"""
if graphene_sqlalchemy_version_lt_2_1_2:
query_count = 8 # default
else:
query_count = 5 # graphene_sqlalchemy_filter.ModelLoader
with SQLAlchemyQueryCounter(session, query_count):
execution_result = schema.execute(
request_string, context={'session': session}
)
assert not execution_result.errors
assert not execution_result.invalid
assert execution_result.data
assert 'field' in execution_result.data
field = execution_result.data['field']
assert 'edges' in field
edges = field['edges']
assert len(edges) == 2
user_0 = edges[0]['node']
user_1 = edges[1]['node']
user_0_memberships_edges = user_0['SomeMemberships']['edges']
is_moderator_values_1 = [
e['node']['isModerator'] for e in user_0_memberships_edges
]
assert is_moderator_values_1 == [False, True, False]
user_0_created_memberships_edges = user_0['createdMemberships']['edges']
is_moderator_values_2 = [
e['node']['isModerator'] for e in user_0_created_memberships_edges
]
assert is_moderator_values_2 == [False, True, False, True]
user_0_groups_edges = user_0['groups']['edges']
groups = [e['node']['name'] for e in user_0_groups_edges]
assert groups == ['group_1', 'group_2', 'group_3']
user_1_memberships_edges = user_1['SomeMemberships']['edges']
is_moderator_values = [
e['node']['isModerator'] for e in user_1_memberships_edges
]
assert is_moderator_values == [True, False, False]