Fix coverage

GeotrekCE · Aug 2, 2022 · 771d178 · 771d178
1 parent 7d53f16
commit 771d178
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/geotrek/core/models.py b/geotrek/core/models.py
@@ -244,8 +244,8 @@ def save(self, *args, **kwargs):
                 aggr.save()
             self._is_reversed = False
 
-        # If draft is True, send email to managers
-        if (self.pk and self.__class__.objects.get(pk=self.pk).draft != self.draft and self.draft) \
+        # If draft is set from False to True, and setting ALERT_DRAFT is True, send email to managers
+        if (self.pk and self.draft and self.__class__.objects.get(pk=self.pk).draft != self.draft) \
                 and settings.ALERT_DRAFT:
             subject = _("{obj} has been set to draft").format(obj=self)
             message = render_to_string('core/draft_email_message.txt', {"obj": self})

diff --git a/geotrek/core/tests/test_models.py b/geotrek/core/tests/test_models.py
@@ -110,8 +110,8 @@ def test_status_draft_alert(self):
         p1.save()
         self.assertEqual(len(mail.outbox), 1)
 
-    @override_settings(ALERT_REVIEW=True)
-    @mock.patch('geotrek.common.mixins.models.mail_managers')
+    @override_settings(ALERT_DRAFT=True)
+    @mock.patch('geotrek.core.models.mail_managers')
     def test_status_draft_fail_mail(self, mock_mail):
         mock_mail.side_effect = Exception("Test")
         p1 = PathFactory.create()