From d1299b058d08434c6966c28a826255db69cc5cd8 Mon Sep 17 00:00:00 2001 From: collerek Date: Thu, 22 Oct 2020 12:58:03 +0200 Subject: [PATCH] add more test with alternative column names --- .coverage | Bin 53248 -> 53248 bytes tests/test_aliases.py | 22 ++++++++++++++++++++++ 2 files changed, 22 insertions(+) diff --git a/.coverage b/.coverage index e491190b6ca1563327e0366abff0a37c93180b7a..0adcea0b240d514ef5be6d5eb340ae010424570d 100644 GIT binary patch delta 158 zcmV;P0Ac@tpaX!Q1F%3Z2QoS}Haavovqvw#P+SH85BU$?56}UEv-yt)K>CR^zW@LL delta 155 zcmV;M0A&AwpaX!Q1F%3Z2QfM{IXX2kvqvw#P+SQB5BU$_57Q6G54;bm50DRo4^t0D z4?YhW4(ATv4%H6K4#N(*4ucM14onU+4k`{E4hs$X4dxBg4aN diff --git a/tests/test_aliases.py b/tests/test_aliases.py index 8ca4136..f169c30 100644 --- a/tests/test_aliases.py +++ b/tests/test_aliases.py @@ -133,3 +133,25 @@ async def test_bulk_operations_and_fields(): await children[0].load() await children[0].delete() children = await Child.objects.all() + assert len(children) == 1 + + +@pytest.mark.asyncio +async def test_working_with_aliases_get_or_create(): + async with database: + async with database.transaction(force_rollback=True): + artist = await Artist.objects.get_or_create(first_name='Teddy', last_name='Bear', born_year=2020) + assert artist.pk is not None + + artist2 = await Artist.objects.get_or_create(first_name='Teddy', last_name='Bear', born_year=2020) + assert artist == artist2 + + art3 = artist2.dict() + art3['born_year'] = 2019 + await Artist.objects.update_or_create(**art3) + + artist3 = await Artist.objects.get(last_name='Bear') + assert artist3.born_year == 2019 + + artists = await Artist.objects.all() + assert len(artists) == 1