add test for Add request.FILES to post_comment form

Author: DominicLGit

tests/custom_comments/__init__.py

@@ -1,7 +1,7 @@
 from django.urls import reverse
 
 from . import views
-from .forms import CustomCommentForm
+
 
 
 def get_model():
@@ -10,6 +10,7 @@ def get_model():
 
 
 def get_form():
+    from .forms import CustomCommentForm
     return CustomCommentForm
 
 

tests/custom_comments/forms.py

@@ -1,5 +1,12 @@
 from django import forms
 
+from django_comments.forms import CommentForm
 
-class CustomCommentForm(forms.Form):
-    pass
+
+class CustomCommentForm(CommentForm):
+    file = forms.FileField()
+
+    def get_comment_create_data(self, site_id=None):
+        data = super().get_comment_create_data(site_id=site_id)
+        data["file"] = self.cleaned_data["file"]
+        return data

tests/custom_comments/migrations/0001_initial.py

@@ -0,0 +1,40 @@
+# Generated by Django 4.2.1 on 2023-05-17 22:32
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+    initial = True
+
+    dependencies = [
+        ("django_comments", "0004_add_object_pk_is_removed_index"),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name="CustomComment",
+            fields=[
+                (
+                    "comment_ptr",
+                    models.OneToOneField(
+                        auto_created=True,
+                        on_delete=django.db.models.deletion.CASCADE,
+                        parent_link=True,
+                        primary_key=True,
+                        serialize=False,
+                        to="django_comments.comment",
+                    ),
+                ),
+                ("file", models.FileField(upload_to="")),
+            ],
+            options={
+                "verbose_name": "comment",
+                "verbose_name_plural": "comments",
+                "ordering": ("submit_date",),
+                "permissions": [("can_moderate", "Can moderate comments")],
+                "abstract": False,
+            },
+            bases=("django_comments.comment",),
+        ),
+    ]

tests/custom_comments/migrations/__init__.py

@@ -1,5 +1,7 @@
 from django.db import models
 
+from django_comments.models import Comment
 
-class CustomComment(models.Model):
-    pass
+
+class CustomComment(Comment):
+    file = models.FileField()

tests/custom_comments/models.py

@@ -7,6 +7,7 @@
 import os
 import sys
 import django
+from django.core.management import call_command
 
 here = os.path.dirname(os.path.abspath(__file__))
 parent = os.path.dirname(here)
@@ -27,6 +28,7 @@
         "testapp",
         "custom_comments",
     ],
+    MEDIA_ROOT=os.path.join(here, 'media'),
     MIDDLEWARE=(
         'django.contrib.sessions.middleware.SessionMiddleware',
         'django.contrib.auth.middleware.AuthenticationMiddleware',

tests/runtests.py

@@ -1,3 +1,6 @@
+import shutil
+
+from django.conf import settings
 from django.contrib.auth.models import User
 from django.contrib.contenttypes.models import ContentType
 from django.contrib.sites.models import Site
@@ -23,6 +26,10 @@ class CommentTestCase(TestCase):
     """
     fixtures = ["comment_tests"]
 
+    def tearDown(self):
+        super().tearDown()
+        shutil.rmtree(settings.MEDIA_ROOT, ignore_errors=True)
+
     def createSomeComments(self):
         # Two anonymous comments on two different objects
         c1 = Comment.objects.create(

tests/testapp/tests/__init__.py

@@ -1,6 +1,10 @@
 from django.conf import settings
+
 from django.contrib.auth.models import User
+from django.core.files.uploadedfile import SimpleUploadedFile
+from django.test.utils import override_settings
 
+import django_comments
 from django_comments import signals
 from django_comments.abstracts import COMMENT_MAX_LENGTH
 from django_comments.models import Comment
@@ -85,6 +89,20 @@ def testPostTooLongComment(self):
             response, "Ensure this value has at most %d characters" % COMMENT_MAX_LENGTH
         )
 
+    @override_settings(
+        COMMENTS_APP='custom_comments',
+    )
+    def testPostCommentWithFile(self):
+        a = Article.objects.get(pk=1)
+        data = self.getValidData(a)
+        test_file = SimpleUploadedFile("test_file.txt", b"file_content")
+        data["file"] = test_file
+        response = self.client.post("/post/", data)
+        self.assertEqual(response.status_code, 302)
+        custom_comment = django_comments.get_model().objects.first()
+        self.assertTrue(custom_comment.file)
+        self.assertEqual(custom_comment.file.read(), b"file_content")
+
     def testCommentPreview(self):
         a = Article.objects.get(pk=1)
         data = self.getValidData(a)