From 099615c69040bbb30f0bbadd82148fc07835b240 Mon Sep 17 00:00:00 2001 From: collerek Date: Thu, 10 Dec 2020 16:27:19 +0100 Subject: [PATCH] fix tests --- tests/test_saving_nullable_fields.py | 27 +++++++++++++-------------- 1 file changed, 13 insertions(+), 14 deletions(-) diff --git a/tests/test_saving_nullable_fields.py b/tests/test_saving_nullable_fields.py index cc8eb36..061e578 100644 --- a/tests/test_saving_nullable_fields.py +++ b/tests/test_saving_nullable_fields.py @@ -49,19 +49,18 @@ def create_test_database(): @pytest.mark.asyncio async def test_create_models(): - primary = await PrimaryModel( - name="Foo", some_text="Bar", some_other_text="Baz" - ).save() - assert primary.id == 1 + async with db: + async with db.transaction(force_rollback=True): + primary = await PrimaryModel( + name="Foo", some_text="Bar", some_other_text="Baz" + ).save() + assert primary.id == 1 - secondary = await SecondaryModel(name="Foo", primary_model=primary).save() - assert secondary.id == 1 - assert secondary.primary_model.id == 1 + secondary = await SecondaryModel(name="Foo", primary_model=primary).save() + assert secondary.id == 1 + assert secondary.primary_model.id == 1 - -@pytest.mark.asyncio -async def test_update_secondary(): - secondary = await SecondaryModel.objects.get(id=1) - assert secondary.name == "Foo" - await secondary.update(name="Updated") - assert secondary.name == "Updated" + secondary = await SecondaryModel.objects.get() + assert secondary.name == "Foo" + await secondary.update(name="Updated") + assert secondary.name == "Updated"