Add get email domain test

When the TEST_MOZMAIL and ADDITIONAL_DOMAINS are set return the first domain from ADDITIONAL_DOMAINS
mozilla · Aug 12, 2021 · b93e80f · b93e80f
1 parent 55b43b2
commit b93e80f
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/emails/tests/utils_tests.py b/emails/tests/utils_tests.py
@@ -75,3 +75,8 @@ def test_get_email_domain_from_settings_on_heroku_test_mozmail_false(self):
     def test_get_email_domain_from_settings_not_on_heroku_test_mozmail_false(self):
         email_domain = get_email_domain_from_settings()
         assert 'test.com' == email_domain
+
+    @override_settings(ON_HEROKU=True, ADDITIONAL_DOMAINS=['mozmail-test.com'], TEST_MOZMAIL=True)
+    def test_get_email_domain_from_settings_test_mozmail_true(self):
+        email_domain = get_email_domain_from_settings()
+        assert 'mozmail-test.com' == email_domain