feat: add tests for SnippetPluginForm

Author: michaeljcollinsuk

djangocms_snippet/forms.py

@@ -79,7 +79,7 @@ def __init__(self, *args, **kwargs):
         this the get_related_url method on the widget is overridden to build a URL for the Snippet admin instead of
         the SnippetGrouper, as this is not enabled in the admin.
         """
-        super(SnippetPluginForm, self).__init__(*args, **kwargs)
+        super().__init__(*args, **kwargs)
         self.fields["snippet_grouper"].widget.can_add_related = True
         self.fields["snippet_grouper"].widget.get_related_url = self.get_related_url_for_snippet
 

tests/test_forms.py

@@ -5,6 +5,7 @@
 from cms.test_utils.testcases import CMSTestCase
 
 from djangocms_snippet import cms_config, forms
+from djangocms_snippet.forms import SnippetPluginForm
 from djangocms_snippet.models import Snippet, SnippetGrouper
 
 from .utils.factories import SnippetWithVersionFactory
@@ -188,3 +189,23 @@ def test_snippet_form_validation_multiple_version_states_in_grouper(self):
         form = forms.SnippetForm(form_data)
 
         self.assertTrue(form.is_valid())
+
+
+class SnippetPluginFormTestCase(CMSTestCase):
+
+    def setUp(self):
+        self.form = SnippetPluginForm()
+
+    def test_get_related_url_for_snippet(self):
+        """
+        Check that the url to add a snippet in the admin is returned
+        """
+        self.assertEqual(self.form.get_related_url_for_snippet("", "add"), "/en/admin/djangocms_snippet/snippet/add/")
+
+    def test_get_related_url_for_snippet_used(self):
+        """
+        Checks that the get_related_url widget is overridden
+        """
+        snippet_grouper_widget = self.form.fields["snippet_grouper"].widget
+        self.assertEqual(snippet_grouper_widget.get_related_url, self.form.get_related_url_for_snippet)
+        self.assertTrue(snippet_grouper_widget.can_add_related)