add tests to the interchangable keys

Author: PetrDlouhy

tests/testapp/fixtures/comment_tests.json

@@ -27,6 +27,7 @@
     "pk" : 1,
     "fields" : {
         "author" : 1,
+        "uuid" : "336384ea-b04f-4a3a-a06a-1f25a8048f8f",
         "headline" : "Man Bites Dog"
     }
   },
@@ -35,6 +36,7 @@
     "pk" : 2,
     "fields" : {
         "author" : 2,
+        "uuid" : "d77c5d7d-1b0d-467b-814b-96697bd9a686",
         "headline" : "Dog Bites Man"
     }
   },

tests/testapp/models.py

@@ -15,6 +15,7 @@ def __str__(self):
 
 
 class Article(models.Model):
+    uuid = models.UUIDField(editable=False, null=True)
     author = models.ForeignKey(Author, on_delete=models.CASCADE)
     headline = models.CharField(max_length=100)
 

tests/testapp/tests/test_comment_form.py

@@ -1,14 +1,19 @@
 import time
+from datetime import datetime
 
 from django.conf import settings
+from django.contrib.contenttypes.models import ContentType
 from django.contrib.sites.models import Site
+from django.test.utils import override_settings
+from freezegun import freeze_time
+from testapp.models import Article
 
 from django_comments.forms import CommentForm
 from django_comments.models import Comment
 
 from . import CommentTestCase
-from testapp.models import Article
 
+CT = ContentType.objects.get_for_model
 
 class CommentFormTests(CommentTestCase):
 
@@ -75,6 +80,92 @@ def testGetCommentObject(self):
         c = f.get_comment_object(site_id=self.site_2.id)
         self.assertEqual(c.site_id, self.site_2.id)
 
+    @freeze_time("2012-01-14 13:21:34")
+    def test_get_comment_create_data_uuid(self):
+        """
+        The get_comment_create_data() method returns
+        uuid field as  object_pk if overriden by settings
+        """
+        a = Article.objects.get(pk=1)
+        d = self.getValidData(a)
+        d["comment"] = "testGetCommentObject with a site"
+        f = CommentForm(Article.objects.get(pk=1), data=d)
+        self.assertTrue(f.is_valid())
+        with override_settings(
+            COMMENTS_ID_OVERRIDES={
+                "testapp.Article": "uuid",
+            }
+        ):
+            c = f.get_comment_create_data(site_id=self.site_2.id)
+            self.assertDictEqual(
+                c,
+                {
+                    "comment": "testGetCommentObject with a site",
+                    "content_type": CT(Article),
+                    "is_public": True,
+                    "is_removed": False,
+                    "object_pk": "336384ea-b04f-4a3a-a06a-1f25a8048f8f",  # uuid is returned
+                    "site_id": 2,
+                    "submit_date": datetime(2012, 1, 14, 13, 21, 34),
+                    "user_email": "[email protected]",
+                    "user_name": "Jim Bob",
+                    "user_url": "",
+                },
+            )
+        c = f.get_comment_create_data(site_id=self.site_2.id)
+        self.assertDictEqual(
+            c,
+            {
+                "comment": "testGetCommentObject with a site",
+                "content_type": CT(Article),
+                "is_public": True,
+                "is_removed": False,
+                "object_pk": "1",  # pk is returned as object_pk
+                "site_id": 2,
+                "submit_date": datetime(2012, 1, 14, 13, 21, 34),
+                "user_email": "[email protected]",
+                "user_name": "Jim Bob",
+                "user_url": "",
+            },
+        )
+
+    @freeze_time("2012-01-14 13:21:34")
+    def test_generate_security_data_uuid(self):
+        """
+        The generate_security_data() method returns
+        uuid field as  object_pk if overriden by settings
+        """
+        a = Article.objects.get(pk=1)
+        d = self.getValidData(a)
+        d["comment"] = "testGetCommentObject with a site"
+        f = CommentForm(Article.objects.get(pk=1), data=d)
+        self.assertTrue(f.is_valid())
+        with override_settings(
+            COMMENTS_ID_OVERRIDES={
+                "testapp.Article": "uuid",
+            }
+        ):
+            c = f.generate_security_data()
+            self.assertDictEqual(
+                c,
+                {
+                    "content_type": "testapp.article",
+                    "object_pk": "336384ea-b04f-4a3a-a06a-1f25a8048f8f",
+                    "security_hash": "b89ebc7c1c6ed757991fa06027405aecdf8d51f1",
+                    "timestamp": "1326547294",
+                },
+            )
+        c = f.generate_security_data()
+        self.assertDictEqual(
+            c,
+            {
+                "content_type": "testapp.article",
+                "object_pk": "1",
+                "security_hash": "2f4a55f47e58b22791d4d26f3b1a2e594302fb61",
+                "timestamp": "1326547294",
+            },
+        )
+
     def testProfanities(self):
         """Test COMMENTS_ALLOW_PROFANITIES and PROFANITIES_LIST settings"""
         a = Article.objects.get(pk=1)

tests/testapp/tests/test_comment_views.py

@@ -1,5 +1,6 @@
 from django.conf import settings
 from django.contrib.auth.models import User
+from django.test.utils import override_settings
 
 from django_comments import signals
 from django_comments.abstracts import COMMENT_MAX_LENGTH
@@ -333,6 +334,28 @@ def testCommentPostRedirectWithInvalidIntegerPK(self):
         response = self.client.get(broken_location)
         self.assertEqual(response.status_code, 200)
 
+    @override_settings(
+        COMMENTS_ID_OVERRIDES={
+            "testapp.Article": "uuid",
+        }
+    )
+    def testCommentPostWithUUID(self):
+        """
+        Tests that attempting to retrieve the location specified in the
+        post redirect, after adding some invalid data to the expected
+        querystring it ends with, doesn't cause a server error.
+        """
+        a = Article.objects.get(pk=1)
+        data = self.getValidData(a)
+        data["comment"] = "This is another comment"
+        self.assertEqual(data["object_pk"], "336384ea-b04f-4a3a-a06a-1f25a8048f8f")
+        response = self.client.post("/post/", data)
+        self.assertEqual(response.status_code, 302)
+        self.assertEqual(Comment.objects.count(), 1)
+        self.assertEqual(
+            Comment.objects.first().object_pk, "336384ea-b04f-4a3a-a06a-1f25a8048f8f"
+        )
+
     def testCommentNextWithQueryStringAndAnchor(self):
         """
         The `next` key needs to handle already having an anchor. Refs #13411.

tox.ini

@@ -27,3 +27,4 @@ deps=
     django-40: Django>=4.0a1,<4.1
     django-41: Django>=4.1a1,<4.2
     django-main: https://github.com/django/django/archive/main.tar.gz
+    freezegun