Update get_or_create method

tests/test_fastapi/test_m2m_forwardref.py

@@ -102,7 +102,7 @@ def test_payload():
             "native_name": "Thailand",
         }
         resp = client.post("/", json=payload, headers={"application-type": "json"})
-        print(resp.content)
+        # print(resp.content)
         assert resp.status_code == 201
 
         resp_country = Country(**resp.json())

tests/test_model_definition/test_aliases.py

@@ -157,15 +157,17 @@ async def test_bulk_operations_and_fields():
 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(
+            artist, created = await Artist.objects.get_or_create(
                 first_name="Teddy", last_name="Bear", born_year=2020
             )
             assert artist.pk is not None
+            assert created is True
 
-            artist2 = await Artist.objects.get_or_create(
+            artist2, created = await Artist.objects.get_or_create(
                 first_name="Teddy", last_name="Bear", born_year=2020
             )
             assert artist == artist2
+            assert created is False
 
             art3 = artist2.dict()
             art3["born_year"] = 2019

tests/test_model_definition/test_save_status.py

@@ -195,12 +195,14 @@ async def test_queryset_methods():
             comps = await Company.objects.all()
             assert [comp.saved for comp in comps]
 
-            comp2 = await Company.objects.get_or_create(name="Banzai_new", founded=2001)
+            comp2, created = await Company.objects.get_or_create(name="Banzai_new", founded=2001)
             assert comp2.saved
+            assert created is True
 
-            comp3 = await Company.objects.get_or_create(name="Banzai", founded=1988)
+            comp3, created = await Company.objects.get_or_create(name="Banzai", founded=1988)
             assert comp3.saved
             assert comp3.pk == comp.pk
+            assert created is False
 
             update_dict = comp.dict()
             update_dict["founded"] = 2010

tests/test_queries/test_queryproxy_on_m2m_models.py

@@ -102,20 +102,23 @@ async def test_queryset_methods():
             await post.categories.add(news)
             await post.categories.add(breaking)
 
-            category = await post.categories.get_or_create(name="News")
+            category, created = await post.categories.get_or_create(name="News")
             assert category == news
             assert len(post.categories) == 1
+            assert created is False
 
-            category = await post.categories.get_or_create(name="Breaking News")
+            category, created = await post.categories.get_or_create(name="Breaking News")
             assert category != breaking
             assert category.pk is not None
             assert len(post.categories) == 2
+            assert created is True
 
             await post.categories.update_or_create(pk=category.pk, name="Urgent News")
             assert len(post.categories) == 2
-            cat = await post.categories.get_or_create(name="Urgent News")
+            cat, created = await post.categories.get_or_create(name="Urgent News")
             assert cat.pk == category.pk
             assert len(post.categories) == 1
+            assert created is False
 
             await post.categories.remove(cat)
             await cat.delete()

tests/test_queries/test_queryset_level_methods.py

@@ -166,17 +166,18 @@ async def test_delete_and_update():
 @pytest.mark.asyncio
 async def test_get_or_create():
     async with database:
-        tom = await Book.objects.get_or_create(
+        tom, created = await Book.objects.get_or_create(
             title="Volume I", author="Anonymous", genre="Fiction"
         )
         assert await Book.objects.count() == 1
+        assert created is True
 
-        assert (
-            await Book.objects.get_or_create(
-                title="Volume I", author="Anonymous", genre="Fiction"
-            )
-            == tom
+        second_tom, created = await Book.objects.get_or_create(
+            title="Volume I", author="Anonymous", genre="Fiction"
         )
+
+        assert second_tom.pk == tom.pk
+        assert created is False
         assert await Book.objects.count() == 1
 
         assert await Book.objects.create(
@@ -188,6 +189,28 @@ async def test_get_or_create():
             )
 
 
+@pytest.mark.asyncio
+async def test_get_or_create_with_defaults():
+    async with database:
+        book, created = await Book.objects.get_or_create(
+            title="Nice book", _defaults={"author": "Mojix", "genre": "Historic"}
+        )
+        assert created is True
+        assert book.author == "Mojix"
+        assert book.title == "Nice book"
+        assert book.genre == "Historic"
+
+        book2, created = await Book.objects.get_or_create(
+            author="Mojix", _defaults={"title": "Book2"}
+        )
+        assert created is False
+        assert book2 == book
+        assert book2.title == "Nice book"
+        assert book2.author == "Mojix"
+        assert book2.genre == "Historic"
+        assert await Book.objects.count() == 1
+        
+
 @pytest.mark.asyncio
 async def test_update_or_create():
     async with database:

tests/test_queries/test_reverse_fk_queryset.py

@@ -87,22 +87,26 @@ async def test_quering_by_reverse_fk():
             assert await album.tracks.exists()
             assert await album.tracks.count() == 3
 
-            track = await album.tracks.get_or_create(
+            track, created = await album.tracks.get_or_create(
                 title="The Bird", position=1, play_count=30
             )
             assert track == track1
+            assert created is False
             assert len(album.tracks) == 1
 
-            track = await album.tracks.get_or_create(
-                title="The Bird2", position=4, play_count=5
+            track, created = await album.tracks.get_or_create(
+                title="The Bird2", _defaults={"position": 4, "play_count": 5}
             )
             assert track != track1
+            assert created is True
             assert track.pk is not None
+            assert track.position == 4 and track.play_count == 5
             assert len(album.tracks) == 2
 
             await album.tracks.update_or_create(pk=track.pk, play_count=50)
             assert len(album.tracks) == 2
-            track = await album.tracks.get_or_create(title="The Bird2")
+            track, created = await album.tracks.get_or_create(title="The Bird2")
+            assert created is False
             assert track.play_count == 50
             assert len(album.tracks) == 1