From cf3ec2e2a55058b47b154f43b3bf414299129ad1 Mon Sep 17 00:00:00 2001 From: EspenAlbert Date: Tue, 24 Nov 2020 09:09:54 +0000 Subject: [PATCH] Refactor tests to cover all lines. --- tests/test_foreign_keys.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/tests/test_foreign_keys.py b/tests/test_foreign_keys.py index ae814d3..09e3e82 100644 --- a/tests/test_foreign_keys.py +++ b/tests/test_foreign_keys.py @@ -383,7 +383,6 @@ async def test_bulk_update_model_with_no_children(): async with database: async with database.transaction(force_rollback=True): album = await Album.objects.create(name="Test") - track = await Track.objects.create(album=album, title="Test1", position=1) album.name = "Test2" await Album.objects.bulk_update([album], columns=["name"]) @@ -396,12 +395,13 @@ async def test_bulk_update_model_with_children(): async with database: async with database.transaction(force_rollback=True): album = await Album.objects.create(name="Test") - track = await Track.objects.create(album=album, title="Test1", position=1) - album_with_tracks = await Album.objects.select_related("tracks").get( - name="Test" - ) - album_with_tracks.name = "Test2" - await Album.objects.bulk_update([album_with_tracks], columns=["name"]) + track = await Track.objects.create(title="Test1", position=1) + album.tracks = [track] + album.name = "Test2" + await Album.objects.bulk_update([album], columns=["name"]) - updated_album = await Album.objects.get(id=album.id) + updated_album = await Album.objects.select_related("tracks").get( + id=album.id + ) assert updated_album.name == "Test2" + assert len(updated_album.tracks) == 0