from datetime import timedelta as td
|
|
|
|
from django.utils.timezone import now
|
|
from hc.test import BaseTestCase
|
|
|
|
|
|
class NotificationsTestCase(BaseTestCase):
|
|
|
|
def test_it_saves_reports_allowed_true(self):
|
|
self.profile.reports_allowed = False
|
|
self.profile.save()
|
|
|
|
self.client.login(username="[email protected]", password="password")
|
|
|
|
form = {"reports_allowed": "on", "nag_period": "0"}
|
|
r = self.client.post("/accounts/profile/notifications/", form)
|
|
self.assertEqual(r.status_code, 200)
|
|
|
|
self.profile.refresh_from_db()
|
|
self.assertTrue(self.profile.reports_allowed)
|
|
self.assertIsNotNone(self.profile.next_report_date)
|
|
|
|
def test_it_saves_reports_allowed_false(self):
|
|
self.profile.reports_allowed = True
|
|
self.profile.next_report_date = now()
|
|
self.profile.save()
|
|
|
|
self.client.login(username="[email protected]", password="password")
|
|
|
|
form = {"nag_period": "0"}
|
|
r = self.client.post("/accounts/profile/notifications/", form)
|
|
self.assertEqual(r.status_code, 200)
|
|
|
|
self.profile.refresh_from_db()
|
|
self.assertFalse(self.profile.reports_allowed)
|
|
self.assertIsNone(self.profile.next_report_date)
|
|
|
|
def test_it_saves_hourly_nag_period(self):
|
|
self.client.login(username="[email protected]", password="password")
|
|
|
|
form = {"nag_period": "3600"}
|
|
r = self.client.post("/accounts/profile/notifications/", form)
|
|
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_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")
|
|
|
|
form = {"nag_period": "1234"}
|
|
r = self.client.post("/accounts/profile/notifications/", form)
|
|
self.assertEqual(r.status_code, 200)
|
|
|
|
self.profile.refresh_from_db()
|
|
self.assertEqual(self.profile.nag_period.total_seconds(), 3600)
|