Refs #36075 -- Used field in pk_fields over field.primary_key.

5 files changed, 46 insertions, 2 deletions

diff --git a/tests/composite_pk/fixtures/tenant.json b/tests/composite_pk/fixtures/tenant.json
index 66a25e94f6..b9b2588ada 100644
--- a/tests/composite_pk/fixtures/tenant.json
+++ b/tests/composite_pk/fixtures/tenant.json
@@ -77,7 +77,8 @@
         "model": "composite_pk.timestamped",
         "fields": {
             "id": 1,
-            "created": "2022-01-12T05:55:14.956"
+            "created": "2022-01-12T05:55:14.956",
+            "text": ""
         }
     }
 ]
diff --git a/tests/composite_pk/models/tenant.py b/tests/composite_pk/models/tenant.py
index 9817b40fac..6286ed2354 100644
--- a/tests/composite_pk/models/tenant.py
+++ b/tests/composite_pk/models/tenant.py
@@ -56,3 +56,4 @@ class TimeStamped(models.Model):
     pk = models.CompositePrimaryKey("id", "created")
     id = models.SmallIntegerField(unique=True)
     created = models.DateTimeField(auto_now_add=True)
+    text = models.TextField(default="", blank=True)
diff --git a/tests/composite_pk/test_models.py b/tests/composite_pk/test_models.py
index ca6ad8b5dc..27157a52ad 100644
--- a/tests/composite_pk/test_models.py
+++ b/tests/composite_pk/test_models.py
@@ -118,6 +118,10 @@ class CompositePKModelsTests(TestCase):
 
                 self.assertSequenceEqual(ctx.exception.messages, messages)
 
+    def test_full_clean_update(self):
+        with self.assertNumQueries(1):
+            self.comment_1.full_clean()
+
     def test_field_conflicts(self):
         test_cases = (
             ({"pk": (1, 1), "id": 2}, (1, 1)),
diff --git a/tests/composite_pk/test_update.py b/tests/composite_pk/test_update.py
index 6748aa2834..4d45e906cf 100644
--- a/tests/composite_pk/test_update.py
+++ b/tests/composite_pk/test_update.py
@@ -1,7 +1,7 @@
 from django.db import connection
 from django.test import TestCase
 
-from .models import Comment, Tenant, Token, User
+from .models import Comment, Tenant, TimeStamped, Token, User
 
 
 class CompositePKUpdateTests(TestCase):
@@ -57,6 +57,28 @@ class CompositePKUpdateTests(TestCase):
         self.assertEqual(user.email, email)
         self.assertEqual(count, User.objects.count())
 
+    def test_update_fields_deferred(self):
+        c = Comment.objects.defer("text", "user_id").get(pk=self.comment_1.pk)
+        c.text = "Hello"
+
+        with self.assertNumQueries(1) as ctx:
+            c.save()
+
+        sql = ctx[0]["sql"]
+        self.assertEqual(sql.count(connection.ops.quote_name("tenant_id")), 1)
+        self.assertEqual(sql.count(connection.ops.quote_name("comment_id")), 1)
+
+        c = Comment.objects.get(pk=self.comment_1.pk)
+        self.assertEqual(c.text, "Hello")
+
+    def test_update_fields_pk_field(self):
+        msg = (
+            "The following fields do not exist in this model, are m2m fields, "
+            "or are non-concrete fields: id"
+        )
+        with self.assertRaisesMessage(ValueError, msg):
+            self.user_1.save(update_fields=["id"])
+
     def test_bulk_update_comments(self):
         comment_1 = Comment.objects.get(pk=self.comment_1.pk)
         comment_2 = Comment.objects.get(pk=self.comment_2.pk)
@@ -77,6 +99,11 @@ class CompositePKUpdateTests(TestCase):
         self.assertEqual(comment_2.text, "bar")
         self.assertEqual(comment_3.text, "baz")
 
+    def test_bulk_update_primary_key_fields(self):
+        message = "bulk_update() cannot be used with primary key fields."
+        with self.assertRaisesMessage(ValueError, message):
+            Comment.objects.bulk_update([self.comment_1, self.comment_2], ["id"])
+
     def test_update_or_create_user(self):
         test_cases = (
             {
@@ -110,6 +137,16 @@ class CompositePKUpdateTests(TestCase):
                 self.assertEqual(user.email, fields["defaults"]["email"])
                 self.assertEqual(count, User.objects.count())
 
+    def test_update_or_create_with_pre_save_pk_field(self):
+        t = TimeStamped.objects.create(id=1)
+        self.assertEqual(TimeStamped.objects.count(), 1)
+        t, created = TimeStamped.objects.update_or_create(
+            pk=t.pk, defaults={"text": "new text"}
+        )
+        self.assertIs(created, False)
+        self.assertEqual(TimeStamped.objects.count(), 1)
+        self.assertEqual(t.text, "new text")
+
     def test_update_comment_by_user_email(self):
         result = Comment.objects.filter(user__email=self.user_1.email).update(
             text="foo"
diff --git a/tests/composite_pk/tests.py b/tests/composite_pk/tests.py
index 303c6592fb..6b09480fb0 100644
--- a/tests/composite_pk/tests.py
+++ b/tests/composite_pk/tests.py
@@ -340,6 +340,7 @@ class CompositePKFixturesTests(TestCase):
                     "fields": {
                         "id": 1,
                         "created": "2022-01-12T05:55:14.956",
+                        "text": "",
                     },
                 },
             ],