You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
45 lines
1.6 KiB
45 lines
1.6 KiB
from datetime import timedelta as td
|
|
|
|
from django.core import signing
|
|
from django.utils.timezone import now
|
|
from hc.test import BaseTestCase
|
|
|
|
|
|
class UnsubscribeReportsTestCase(BaseTestCase):
|
|
|
|
def test_token_works(self):
|
|
self.profile.next_report_date = now()
|
|
self.profile.nag_period = td(hours=1)
|
|
self.profile.next_nag_date = now()
|
|
self.profile.save()
|
|
|
|
token = signing.Signer().sign("foo")
|
|
url = "/accounts/unsubscribe_reports/alice/?token=%s" % token
|
|
r = self.client.get(url)
|
|
self.assertContains(r, "You have been unsubscribed")
|
|
|
|
self.profile.refresh_from_db()
|
|
self.assertFalse(self.profile.reports_allowed)
|
|
self.assertIsNone(self.profile.next_report_date)
|
|
|
|
self.assertEqual(self.profile.nag_period.total_seconds(), 0)
|
|
self.assertIsNone(self.profile.next_nag_date)
|
|
|
|
def test_bad_token_gets_rejected(self):
|
|
url = "/accounts/unsubscribe_reports/alice/?token=invalid"
|
|
r = self.client.get(url)
|
|
self.assertContains(r, "Incorrect Link")
|
|
|
|
def test_signed_username_works(self):
|
|
sig = signing.TimestampSigner(salt="reports").sign("alice")
|
|
url = "/accounts/unsubscribe_reports/%s/" % sig
|
|
r = self.client.get(url)
|
|
self.assertContains(r, "You have been unsubscribed")
|
|
|
|
self.profile.refresh_from_db()
|
|
self.assertFalse(self.profile.reports_allowed)
|
|
|
|
def test_bad_signature_gets_rejected(self):
|
|
url = "/accounts/unsubscribe_reports/invalid/"
|
|
r = self.client.get(url)
|
|
self.assertContains(r, "Incorrect Link")
|