|
@@ -1,3 +1,4 @@
|
|
|
+import datetime
|
|
|
import json
|
|
|
import warnings
|
|
|
|
|
@@ -5,6 +6,8 @@ from django.core import mail
|
|
|
from django.core.signing import BadSignature
|
|
|
from django.contrib.auth.models import User
|
|
|
from django.test import TestCase, Client, override_settings
|
|
|
+from freezegun import freeze_time
|
|
|
+import pytz
|
|
|
|
|
|
from contribmap.models import Contrib
|
|
|
from contribmap.forms import PublicContribForm
|
|
@@ -149,6 +152,78 @@ class TestViews(APITestCase):
|
|
|
self.assertIn('JohnCleese', mail.outbox[0].subject)
|
|
|
self.assertIn('JohnCleese', mail.outbox[0].body)
|
|
|
|
|
|
+
|
|
|
+class TestManageView(APITestCase):
|
|
|
+ def setUp(self):
|
|
|
+ self.contrib = Contrib.objects.create(
|
|
|
+ name='John',
|
|
|
+ phone='010101010101',
|
|
|
+ contrib_type=Contrib.CONTRIB_CONNECT,
|
|
|
+ privacy_coordinates=True,
|
|
|
+ latitude=0.5,
|
|
|
+ longitude=0.5,
|
|
|
+ )
|
|
|
+ self.token = ContribTokenManager().mk_token(self.contrib)
|
|
|
+
|
|
|
+ def test_manage_with_token(self):
|
|
|
+ # No token
|
|
|
+ response = self.client.get('/map/manage/{}'.format(self.contrib.pk))
|
|
|
+ self.assertEqual(response.status_code, 403)
|
|
|
+
|
|
|
+ # Garbage token
|
|
|
+ response = self.client.get(
|
|
|
+ '/map/manage/{}?token=burp'.format(self.contrib.pk))
|
|
|
+ self.assertEqual(response.status_code, 403)
|
|
|
+
|
|
|
+ # Valid token, but for another contrib
|
|
|
+ contrib2 = Contrib.objects.create(
|
|
|
+ name='John2',
|
|
|
+ phone='010101010101',
|
|
|
+ contrib_type=Contrib.CONTRIB_CONNECT,
|
|
|
+ privacy_coordinates=True,
|
|
|
+ latitude=0.5,
|
|
|
+ longitude=0.5,
|
|
|
+ )
|
|
|
+ token2 = ContribTokenManager().mk_token(contrib2)
|
|
|
+
|
|
|
+ response = self.client.get('/map/manage/{}?token={}'.format(
|
|
|
+ self.contrib.pk, token2))
|
|
|
+ self.assertEqual(response.status_code, 403)
|
|
|
+
|
|
|
+ # Normal legitimate access case
|
|
|
+ response = self.client.get(
|
|
|
+ '/map/manage/{}?token={}'.format(self.contrib.pk, self.token))
|
|
|
+ self.assertEqual(response.status_code, 200)
|
|
|
+
|
|
|
+ # Deleted contrib
|
|
|
+ Contrib.objects.all().delete()
|
|
|
+ response = self.client.get(
|
|
|
+ '/map/manage/{}?token={}'.format(self.contrib.pk, self.token))
|
|
|
+ self.assertEqual(response.status_code, 404)
|
|
|
+
|
|
|
+ def test_delete(self):
|
|
|
+ response = self.client.post(
|
|
|
+ '/map/manage/{}?token={}'.format(self.contrib.pk, self.token),
|
|
|
+ {'action': 'delete'})
|
|
|
+ self.assertEqual(response.status_code, 302)
|
|
|
+ self.assertFalse(Contrib.objects.filter(pk=self.contrib.pk).exists())
|
|
|
+
|
|
|
+ def test_renew(self):
|
|
|
+ self.contrib.date = datetime.datetime(2009, 10, 10, tzinfo=pytz.utc)
|
|
|
+ self.contrib.expiration_date = datetime.datetime(2010, 10, 10, tzinfo=pytz.utc)
|
|
|
+ self.contrib.save()
|
|
|
+
|
|
|
+ with freeze_time('12-12-2100', tz_offset=0):
|
|
|
+ response = self.client.post(
|
|
|
+ '/map/manage/{}?token={}'.format(self.contrib.pk, self.token),
|
|
|
+ {'action': 'renew'})
|
|
|
+ self.assertEqual(response.status_code, 200)
|
|
|
+ self.contrib = Contrib.objects.get(pk=self.contrib.pk) # refresh
|
|
|
+ self.assertEqual(
|
|
|
+ self.contrib.expiration_date.date(),
|
|
|
+ datetime.date(2101, 12, 12))
|
|
|
+
|
|
|
+
|
|
|
class TestForms(TestCase):
|
|
|
valid_data = {
|
|
|
'roof': True,
|