Fix broken tests

mozilla · Aug 18, 2021 · 74dad3f · 74dad3f
1 parent 000da4f
commit 74dad3f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/emails/tests/models_tests.py b/emails/tests/models_tests.py
@@ -123,6 +123,9 @@ def test_make_relay_address_with_specified_domain(self):
         assert relay_address.get_domain_display() == 'MOZMAIL_DOMAIN'
         assert relay_address.domain_value == 'test.com'
 
+    @override_settings(TEST_MOZMAIL=False)
+    @patch('emails.models.DOMAINS', TEST_DOMAINS)
+    @patch('emails.models.DEFAULT_DOMAIN', TEST_DOMAINS['RELAY_FIREFOX_DOMAIN'])
     def test_delete_adds_deleted_address_object(self):
         relay_address = baker.make(RelayAddress)
         address_hash = sha256(
@@ -134,6 +137,7 @@ def test_delete_adds_deleted_address_object(self):
         ).count()
         assert deleted_count == 1
 
+    @patch('emails.models.DOMAINS', TEST_DOMAINS)
     def test_delete_mozmail_deleted_address_object(self):
         relay_address = baker.make(RelayAddress, domain=2)
         address_hash = sha256(