Prodream/ormar
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactor tests to cover all lines.

Browse Source
This commit is contained in:
EspenAlbert
2020-11-24 09:09:54 +00:00
parent a668b20f1a
commit cf3ec2e2a5
1 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
tests/test_foreign_keys.py
View File

@ -383,7 +383,6 @@ async def test_bulk_update_model_with_no_children():
async with database: async with database:
async with database.transaction(force_rollback=True): async with database.transaction(force_rollback=True):
album = await Album.objects.create(name="Test") album = await Album.objects.create(name="Test")
track = await Track.objects.create(album=album, title="Test1", position=1)
album.name = "Test2" album.name = "Test2"
await Album.objects.bulk_update([album], columns=["name"]) 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:
async with database.transaction(force_rollback=True): async with database.transaction(force_rollback=True):
album = await Album.objects.create(name="Test") album = await Album.objects.create(name="Test")
track = await Track.objects.create(album=album, title="Test1", position=1) track = await Track.objects.create(title="Test1", position=1)
album_with_tracks = await Album.objects.select_related("tracks").get( album.tracks = [track]
name="Test" album.name = "Test2"
) await Album.objects.bulk_update([album], columns=["name"])
album_with_tracks.name = "Test2"
await Album.objects.bulk_update([album_with_tracks], 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 updated_album.name == "Test2"
assert len(updated_album.tracks) == 0