Add requested tests for html_body template

django-otp · Jun 16, 2023 · d5086b1 · d5086b1
1 parent 47e3b55
commit d5086b1
Show file tree

Hide file tree

Showing 2 changed files with 53 additions and 0 deletions.
diff --git a/src/django_otp/plugins/otp_email/tests.py b/src/django_otp/plugins/otp_email/tests.py
@@ -115,6 +115,32 @@ def test_settings(self):
         with self.subTest(field='body'):
             self.assertEqual(msg.body, "Test template 2: {}".format(self.device.token))
 
+    @override_settings(
+        OTP_EMAIL_SENDER="webmaster@example.com",
+        OTP_EMAIL_SUBJECT="Test subject",
+        OTP_EMAIL_BODY_HTML_TEMPLATE="<div>{{token}}</div>",
+    )
+    def test_settings_html_template(self):
+        self.device.generate_challenge()
+
+        self.assertEqual(len(mail.outbox), 1)
+
+        msg = mail.outbox[0]
+
+        with self.subTest(field='from_email'):
+            self.assertEqual(msg.from_email, "webmaster@example.com")
+        with self.subTest(field='subject'):
+            self.assertEqual(msg.subject, "Test subject")
+        with self.subTest(field='body'):
+            self.assertEqual(
+                msg.body, 'Test template 1: {}\n'.format(self.device.token)
+            )
+        with self.subTest(field='alternatives'):
+            self.assertEqual(
+                msg.alternatives[0],
+                ('<div>{}</div>'.format(self.device.token), 'text/html'),
+            )
+
     @override_settings(
         OTP_EMAIL_SENDER="webmaster@example.com",
         OTP_EMAIL_SUBJECT="Test subject",
@@ -136,6 +162,32 @@ def test_settings_template_path(self):
                 msg.body, "Test template 3: {}\n".format(self.device.token)
             )
 
+    @override_settings(
+        OTP_EMAIL_SENDER="webmaster@example.com",
+        OTP_EMAIL_SUBJECT="Test subject",
+        OTP_EMAIL_BODY_HTML_TEMPLATE_PATH="otp/email/custom_html.html",
+    )
+    def test_settings_html_template_path(self):
+        self.device.generate_challenge()
+
+        self.assertEqual(len(mail.outbox), 1)
+
+        msg = mail.outbox[0]
+
+        with self.subTest(field='from_email'):
+            self.assertEqual(msg.from_email, "webmaster@example.com")
+        with self.subTest(field='subject'):
+            self.assertEqual(msg.subject, "Test subject")
+        with self.subTest(field='body'):
+            self.assertEqual(
+                msg.body, 'Test template 1: {}\n'.format(self.device.token)
+            )
+        with self.subTest(field='alternatives'):
+            self.assertEqual(
+                msg.alternatives[0],
+                ('<p>{}</p>'.format(self.device.token), 'text/html'),
+            )
+
     @override_settings(
         OTP_EMAIL_SENDER="webmaster@example.com",
         OTP_EMAIL_SUBJECT="Test subject",

diff --git a/test/test_project/templates/otp/email/custom_html.html b/test/test_project/templates/otp/email/custom_html.html
@@ -0,0 +1 @@
+<p>{{ token }}</p>