- from datetime import timedelta as td
-
- from django.utils.timezone import now
- from hc.api.models import Check
- from hc.test import BaseTestCase
-
-
- class NotificationsTestCase(BaseTestCase):
- url = "/accounts/profile/notifications/"
-
- def _payload(self, **kwargs):
- result = {"reports": "monthly", "nag_period": "0", "tz": "Europe/Riga"}
- result.update(kwargs)
- return result
-
- def test_it_saves_reports_monthly(self):
- self.profile.reports = "off"
- self.profile.save()
-
- self.client.login(username="[email protected]", password="password")
-
- r = self.client.post(self.url, self._payload())
- self.assertEqual(r.status_code, 200)
-
- self.profile.refresh_from_db()
- self.assertEqual(self.profile.reports, "monthly")
- self.assertEqual(self.profile.next_report_date.day, 1)
-
- def test_it_saves_reports_weekly(self):
- self.profile.reports = "off"
- self.profile.save()
-
- self.client.login(username="[email protected]", password="password")
-
- r = self.client.post(self.url, self._payload(reports="weekly"))
- self.assertEqual(r.status_code, 200)
-
- self.profile.refresh_from_db()
- self.assertEqual(self.profile.reports, "weekly")
- self.assertEqual(self.profile.next_report_date.weekday(), 0)
-
- def test_it_saves_reports_off(self):
- self.profile.reports = "monthly"
- self.profile.next_report_date = now()
- self.profile.save()
-
- self.client.login(username="[email protected]", password="password")
-
- r = self.client.post(self.url, self._payload(reports="off"))
- self.assertEqual(r.status_code, 200)
-
- self.profile.refresh_from_db()
- self.assertEqual(self.profile.reports, "off")
- self.assertIsNone(self.profile.next_report_date)
-
- def test_it_sets_next_nag_date_when_setting_hourly_nag_period(self):
- Check.objects.create(project=self.project, status="down")
-
- self.client.login(username="[email protected]", password="password")
-
- r = self.client.post(self.url, self._payload(nag_period="3600"))
- self.assertEqual(r.status_code, 200)
-
- self.profile.refresh_from_db()
- self.assertEqual(self.profile.nag_period.total_seconds(), 3600)
- self.assertIsNotNone(self.profile.next_nag_date)
-
- def test_it_clears_next_nag_date_when_setting_hourly_nag_period(self):
- self.profile.next_nag_date = now() + td(minutes=30)
- self.profile.save()
-
- self.client.login(username="[email protected]", password="password")
-
- r = self.client.post(self.url, self._payload(nag_period="3600"))
- self.assertEqual(r.status_code, 200)
-
- self.profile.refresh_from_db()
- self.assertEqual(self.profile.nag_period.total_seconds(), 3600)
- self.assertIsNone(self.profile.next_nag_date)
-
- def test_it_does_not_save_nonstandard_nag_period(self):
- self.profile.nag_period = td(seconds=3600)
- self.profile.save()
-
- self.client.login(username="[email protected]", password="password")
-
- r = self.client.post(self.url, self._payload(nag_period="1234"))
- self.assertEqual(r.status_code, 200)
-
- self.profile.refresh_from_db()
- self.assertEqual(self.profile.nag_period.total_seconds(), 3600)
-
- def test_it_saves_tz(self):
- self.client.login(username="[email protected]", password="password")
-
- r = self.client.post(self.url, self._payload())
- self.assertEqual(r.status_code, 200)
-
- self.profile.refresh_from_db()
- self.assertEqual(self.profile.tz, "Europe/Riga")
-
- def test_it_ignores_bad_tz(self):
- self.profile.tz = "Europe/Riga"
- self.profile.save()
-
- self.client.login(username="[email protected]", password="password")
-
- r = self.client.post(self.url, self._payload(reports="weekly", tz="Foo/Bar"))
- self.assertEqual(r.status_code, 200)
-
- self.profile.refresh_from_db()
- self.assertEqual(self.profile.reports, "weekly")
- self.assertEqual(self.profile.tz, "Europe/Riga")
|
