From 5c64e0cd5d1bc55a4118430de3fd28474bebca3c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?P=C4=93teris=20Caune?= <cuu508@gmail.com>
Date: Sun, 15 Oct 2017 20:25:04 +0300
Subject: [PATCH] If profile has next_nag_date set in future then
 set_next_nag_date does not touch it.

---
 hc/accounts/models.py           |  2 ++
 hc/api/tests/test_sendalerts.py | 32 ++++++++++++++++++++++++--------
 2 files changed, 26 insertions(+), 8 deletions(-)

diff --git a/hc/accounts/models.py b/hc/accounts/models.py
index e8885fc0..5a7d6970 100644
--- a/hc/accounts/models.py
+++ b/hc/accounts/models.py
@@ -203,6 +203,8 @@ class Profile(models.Model):
         is_member = models.Q(user__memberships__team=self)
         q = Profile.objects.filter(is_owner | is_member)
         q = q.exclude(nag_period=NO_NAG)
+        # Exclude profiles with next_nag_date in future
+        q = q.exclude(next_nag_date__gt=timezone.now())
 
         q.update(next_nag_date=timezone.now() + models.F("nag_period"))
 
diff --git a/hc/api/tests/test_sendalerts.py b/hc/api/tests/test_sendalerts.py
index 7ed585e9..8464f9a9 100644
--- a/hc/api/tests/test_sendalerts.py
+++ b/hc/api/tests/test_sendalerts.py
@@ -2,7 +2,7 @@ from datetime import timedelta
 from mock import Mock, patch
 
 from django.core.management import call_command
-from django.utils import timezone
+from django.utils.timezone import now
 from hc.api.management.commands.sendalerts import Command, notify
 from hc.api.models import Check
 from hc.test import BaseTestCase
@@ -13,7 +13,7 @@ class SendAlertsTestCase(BaseTestCase):
     def test_it_handles_grace_period(self):
         check = Check(user=self.alice, status="up")
         # 1 day 30 minutes after ping the check is in grace period:
-        check.last_ping = timezone.now() - timedelta(days=1, minutes=30)
+        check.last_ping = now() - timedelta(days=1, minutes=30)
         check.alert_after = check.get_alert_after()
         check.save()
 
@@ -23,7 +23,7 @@ class SendAlertsTestCase(BaseTestCase):
     @patch("hc.api.management.commands.sendalerts.notify_on_thread")
     def test_it_notifies_when_check_goes_down(self, mock_notify):
         check = Check(user=self.alice, status="up")
-        check.last_ping = timezone.now() - timedelta(days=2)
+        check.last_ping = now() - timedelta(days=2)
         check.alert_after = check.get_alert_after()
         check.save()
 
@@ -42,7 +42,7 @@ class SendAlertsTestCase(BaseTestCase):
     @patch("hc.api.management.commands.sendalerts.notify_on_thread")
     def test_it_notifies_when_check_goes_up(self, mock_notify):
         check = Check(user=self.alice, status="down")
-        check.last_ping = timezone.now()
+        check.last_ping = now()
         check.alert_after = check.get_alert_after()
         check.save()
 
@@ -64,7 +64,7 @@ class SendAlertsTestCase(BaseTestCase):
     @patch("hc.api.management.commands.sendalerts.notify_on_thread")
     def test_it_updates_alert_after(self, mock_notify):
         check = Check(user=self.alice, status="up")
-        check.last_ping = timezone.now() - timedelta(hours=1)
+        check.last_ping = now() - timedelta(hours=1)
         check.alert_after = check.last_ping
         check.save()
 
@@ -85,7 +85,7 @@ class SendAlertsTestCase(BaseTestCase):
     @patch("hc.api.management.commands.sendalerts.notify")
     def test_it_works_synchronously(self, mock_notify):
         check = Check(user=self.alice, status="up")
-        check.last_ping = timezone.now() - timedelta(days=2)
+        check.last_ping = now() - timedelta(days=2)
         check.alert_after = check.get_alert_after()
         check.save()
 
@@ -99,7 +99,7 @@ class SendAlertsTestCase(BaseTestCase):
         self.profile.save()
 
         check = Check(user=self.alice, status="down")
-        check.last_ping = timezone.now() - timedelta(days=2)
+        check.last_ping = now() - timedelta(days=2)
         check.alert_after = check.get_alert_after()
         check.save()
 
@@ -113,7 +113,7 @@ class SendAlertsTestCase(BaseTestCase):
         self.bobs_profile.save()
 
         check = Check(user=self.alice, status="down")
-        check.last_ping = timezone.now() - timedelta(days=2)
+        check.last_ping = now() - timedelta(days=2)
         check.alert_after = check.get_alert_after()
         check.save()
 
@@ -121,3 +121,19 @@ class SendAlertsTestCase(BaseTestCase):
 
         self.bobs_profile.refresh_from_db()
         self.assertIsNotNone(self.bobs_profile.next_nag_date)
+
+    def test_it_does_not_touch_future_next_nag_dates(self):
+        original_nag_date = now() + timedelta(minutes=30)
+        self.profile.nag_period = timedelta(hours=1)
+        self.profile.next_nag_date = original_nag_date
+        self.profile.save()
+
+        check = Check(user=self.alice, status="down")
+        check.last_ping = now() - timedelta(days=2)
+        check.alert_after = check.get_alert_after()
+        check.save()
+
+        notify(check.id, Mock())
+
+        self.profile.refresh_from_db()
+        self.assertEqual(self.profile.next_nag_date, original_nag_date)
\ No newline at end of file