fix tests

This commit is contained in:
collerek
2020-12-10 16:27:19 +01:00
parent f4aedbfb2b
commit 099615c690

View File

@ -49,19 +49,18 @@ def create_test_database():
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_create_models(): async def test_create_models():
primary = await PrimaryModel( async with db:
name="Foo", some_text="Bar", some_other_text="Baz" async with db.transaction(force_rollback=True):
).save() primary = await PrimaryModel(
assert primary.id == 1 name="Foo", some_text="Bar", some_other_text="Baz"
).save()
assert primary.id == 1
secondary = await SecondaryModel(name="Foo", primary_model=primary).save() secondary = await SecondaryModel(name="Foo", primary_model=primary).save()
assert secondary.id == 1 assert secondary.id == 1
assert secondary.primary_model.id == 1 assert secondary.primary_model.id == 1
secondary = await SecondaryModel.objects.get()
@pytest.mark.asyncio assert secondary.name == "Foo"
async def test_update_secondary(): await secondary.update(name="Updated")
secondary = await SecondaryModel.objects.get(id=1) assert secondary.name == "Updated"
assert secondary.name == "Foo"
await secondary.update(name="Updated")
assert secondary.name == "Updated"