Skip to content

Commit 4c62d62

Browse files
More explicitly test data returned and make it so data is not stored between tests
1 parent 73b70db commit 4c62d62

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

tests/test_dynamodb.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -243,7 +243,7 @@ def alias_query_data(alias_table):
243243
AliasKeyModel.delete(datum["name"])
244244

245245

246-
@pytest.fixture(scope="module")
246+
@pytest.fixture
247247
def nested_query_data(nested_table):
248248
presets = [dict()] * 5
249249
data = [datum for datum in [nested_model_data_generator(**i) for i in presets]]
@@ -257,7 +257,7 @@ def nested_query_data(nested_table):
257257
NestedModel.delete((datum[NestedModel.Config.hash_key], datum[NestedModel.Config.range_key]))
258258

259259

260-
@pytest.fixture(scope="module")
260+
@pytest.fixture
261261
def nested_query_data_empty_ticket(nested_table):
262262
presets = [dict()] * 5
263263
data = [datum for datum in [nested_model_data_generator(include_ticket=False, **i) for i in presets]]
@@ -460,14 +460,15 @@ def test_query_scan_complex(dynamo, complex_query_data):
460460

461461
def test_query_with_nested_model(dynamo, nested_query_data):
462462
res = NestedModel.query()
463-
res_data = [m.ticket for m in res]
464-
assert any(elem is not None for elem in res_data)
463+
for m in res:
464+
assert m.ticket.created_time is not None
465+
assert m.ticket.number is not None
466+
assert type(m.ticket) is Ticket
465467

466468

467469
def test_query_with_nested_model_optional(dynamo, nested_query_data_empty_ticket):
468470
res = NestedModel.query()
469-
res_data = [m.ticket for m in res]
470-
assert any(elem is None for elem in res_data)
471+
assert all([m.ticket is None for m in res])
471472

472473

473474
def test_query_alias_save(dynamo):

0 commit comments

Comments
 (0)