1 files changed, 12 insertions, 2 deletions
diff --git a/tests/composite_pk/test_create.py b/tests/composite_pk/test_create.py
index a007952f66..38ad9690fb 100644
--- a/tests/composite_pk/test_create.py
+++ b/tests/composite_pk/test_create.py
@@ -1,6 +1,7 @@
+from django.db import IntegrityError
 from django.test import TestCase, skipUnlessDBFeature
 
-from .models import Tenant, User
+from .models import Post, Tenant, User
 
 
 class CompositePKCreateTests(TestCase):
@@ -8,7 +9,7 @@ class CompositePKCreateTests(TestCase):
 
     @classmethod
     def setUpTestData(cls):
-        cls.tenant = Tenant.objects.create()
+        cls.tenant = Tenant.objects.create(id=1)
         cls.user = User.objects.create(
             tenant=cls.tenant,
             id=1,
@@ -151,3 +152,12 @@ class CompositePKCreateTests(TestCase):
                 self.assertEqual(user.email, fields["defaults"]["email"])
                 self.assertEqual(user.email, f"user{user.id}@example.com")
                 self.assertEqual(count + 1, User.objects.count())
+
+    def test_save_default_pk_not_set(self):
+        with self.assertNumQueries(1):
+            Post().save()
+
+    def test_save_default_pk_set(self):
+        post = Post.objects.create()
+        with self.assertRaises(IntegrityError):
+            Post(tenant_id=post.tenant_id, id=post.id).save()