Fix get email domain from settings

When the TEST_MOZMAIL and ADDITIONAL_DOMAINS are not set in settings return the domain based on the ON_HEROKU setting
mozilla · Aug 12, 2021 · 55b43b2 · 55b43b2
1 parent 7d1f550
commit 55b43b2
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/emails/tests/utils_tests.py b/emails/tests/utils_tests.py
@@ -66,12 +66,12 @@ def test_generate_relay_From_with_linebreak_chars(self):
         )
         assert formatted_from_address == expected_formatted_from
 
-    @override_settings(ON_HEROKU=True, SITE_ORIGIN='https://test.com')
-    def test_get_email_domain_from_settings_on_heroku(self):
+    @override_settings(ON_HEROKU=True, SITE_ORIGIN='https://test.com', TEST_MOZMAIL=False)
+    def test_get_email_domain_from_settings_on_heroku_test_mozmail_false(self):
         email_domain = get_email_domain_from_settings()
         assert 'mail.test.com' == email_domain
 
-    @override_settings(ON_HEROKU=False, SITE_ORIGIN='https://test.com')
-    def test_get_email_domain_from_settings_not_on_heroku(self):
+    @override_settings(ON_HEROKU=False, SITE_ORIGIN='https://test.com', TEST_MOZMAIL=False)
+    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