Skip to content

Commit 2e77898

Browse files
committed
test: Use cmp to compare dict result instead of json.dumps
1 parent c70adb0 commit 2e77898

File tree

1 file changed

+13
-15
lines changed

1 file changed

+13
-15
lines changed

graphene_mongo/tests/test_query.py

Lines changed: 13 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,7 @@ def resolve_editors(self, *args, **kwargs):
4242
expected = {
4343
'editor': {
4444
'firstName': 'Penny',
45-
'company': {"name": "Newsco"}
45+
'company': {'name': 'Newsco'}
4646
},
4747
'editors': [{
4848
'firstName': 'Penny',
@@ -55,15 +55,17 @@ def resolve_editors(self, *args, **kwargs):
5555
'lastName': 'Rodman'
5656
}]
5757
}
58-
expected_metadata = '{"age": "20", "nickname": "$1"}'
58+
expected_metadata = {
59+
'age': '20',
60+
'nickname': "$1"
61+
}
5962

6063
schema = graphene.Schema(query=Query)
6164
result = schema.execute(query)
6265
assert not result.errors
6366
metadata = result.data['editor'].pop('metadata')
64-
assert (json.loads(metadata)) == dict(json.loads(expected_metadata))
65-
assert json.dumps(result.data, sort_keys=True) == \
66-
json.dumps(expected, sort_keys=True)
67+
assert cmp(json.loads(metadata), expected_metadata) == 0
68+
assert cmp(result.data, expected) == 0
6769

6870

6971
def test_should_query_reporter(fixtures):
@@ -125,7 +127,7 @@ def resolve_reporter(self, *args, **kwargs):
125127
schema = graphene.Schema(query=Query)
126128
result = schema.execute(query)
127129
assert not result.errors
128-
assert dict(result.data['reporter']) == expected['reporter']
130+
assert cmp(result.data, expected) == 0
129131

130132

131133
def test_should_custom_kwargs(fixtures):
@@ -163,7 +165,7 @@ def resolve_editors(self, *args, **kwargs):
163165
schema = graphene.Schema(query=Query)
164166
result = schema.execute(query)
165167
assert not result.errors
166-
assert all(item in result.data['editors'] for item in expected['editors'])
168+
assert cmp(result.data, expected) == 0
167169

168170

169171
def test_should_self_reference(fixtures):
@@ -232,8 +234,7 @@ def resolve_all_players(self, *args, **kwargs):
232234
schema = graphene.Schema(query=Query)
233235
result = schema.execute(query)
234236
assert not result.errors
235-
assert json.dumps(result.data, sort_keys=True) == \
236-
json.dumps(expected, sort_keys=True)
237+
assert cmp(result.data, expected) == 0
237238

238239

239240
def test_should_query_with_embedded_document(fixtures):
@@ -266,8 +267,7 @@ def resolve_professor_vector(self, info, id):
266267
schema = graphene.Schema(query=Query, types=[ProfessorVectorType])
267268
result = schema.execute(query)
268269
assert not result.errors
269-
assert json.dumps(result.data, sort_keys=True) == \
270-
json.dumps(expected, sort_keys=True)
270+
assert cmp(result.data, expected) == 0
271271

272272

273273
def test_should_query_child(fixtures):
@@ -311,8 +311,7 @@ def resolve_children(self, *args, **kwargs):
311311
schema = graphene.Schema(query=Query)
312312
result = schema.execute(query)
313313
assert not result.errors
314-
assert json.dumps(result.data, sort_keys=True) == \
315-
json.dumps(expected, sort_keys=True)
314+
assert cmp(result.data, expected) == 0
316315

317316

318317
def test_should_query_cell_tower(fixtures):
@@ -372,5 +371,4 @@ def resolve_cell_towers(self, *args, **kwargs):
372371
schema = graphene.Schema(query=Query)
373372
result = schema.execute(query)
374373
assert not result.errors
375-
assert json.dumps(result.data, sort_keys=True) == \
376-
json.dumps(expected, sort_keys=True)
374+
assert cmp(result.data, expected) == 0

0 commit comments

Comments
 (0)