tests

Author: marcoacierno

backend/custom_admin/widgets.py

@@ -0,0 +1,5 @@
+from django.forms import Textarea
+
+
+class RichEditorWidget(Textarea):
+    template_name = "astro/widgets/rich-editor.html"

backend/notifications/admin/admins.py

@@ -4,6 +4,7 @@
 from django.urls import reverse
 from django.contrib import admin
 from django.urls.resolvers import URLPattern
+from custom_admin.widgets import RichEditorWidget
 from notifications.admin.views import (
     view_sent_email,
     view_email_template,
@@ -16,10 +17,6 @@
 from django.forms import Textarea
 
 
-class RichEditorWidget(Textarea):
-    template_name = "astro/widgets/rich-editor.html"
-
-
 class SentEmailEventInline(admin.TabularInline):
     model = SentEmailEvent
     extra = 0

backend/notifications/tests/admin/test_email_template_admin.py

@@ -1,5 +1,6 @@
 from django.urls import reverse
 
+from custom_admin.widgets import RichEditorWidget
 from notifications.tests.factories import EmailTemplateFactory
 import pytest
 from notifications.models import EmailTemplate, EmailTemplateIdentifier
@@ -12,7 +13,6 @@
     [
         ("subject", {"rows": 2, "cols": 200}),
         ("preview_text", {"rows": 2, "cols": 200}),
-        ("body", {"rows": 50, "cols": 200}),
     ],
 )
 def test_textarea_size_for_templates(rf, field_name, expected_attrs):
@@ -27,6 +27,18 @@ def test_textarea_size_for_templates(rf, field_name, expected_attrs):
     assert form_field.widget.attrs == expected_attrs
 
 
+def test_email_template_body_uses_rich_editor(rf):
+    db_field = EmailTemplate._meta.get_field("body")
+
+    admin = EmailTemplateAdmin(
+        model=EmailTemplate,
+        admin_site=AdminSite(),
+    )
+
+    form_field = admin.formfield_for_dbfield(db_field, request=rf.get("/"))
+    assert isinstance(form_field.widget, RichEditorWidget)
+
+
 def test_cannot_edit_essential_fields_on_existing_object(rf):
     admin = EmailTemplateAdmin(
         model=EmailTemplate,