class TestEnforceTimezoneMiddleware(TestCase):
def setUp(self):
self.user = User.objects.create(username='******')
self.middleware = EnforceTimezoneMiddleware()
def test_anonymous_user(self):
request = HttpRequest()
request.user = AnonymousUser()
self.assertIsNone(self.middleware.process_request(request))
def test_user_with_no_timezone_set(self):
request = HttpRequest()
request.user = self.user
response = self.middleware.process_request(request)
self.assertIsInstance(response, HttpResponseRedirect)
self.assertTrue(
response['Location'].startswith(reverse('selecttimezone')))
self.assertIn(
'?next=', response['Location'])
def test_no_redirect_loop(self):
request = HttpRequest()
request.user = self.user
request.path = reverse('selecttimezone')
self.assertIsNone(self.middleware.process_request(request))
def test_no_redirect_if_user_has_timezone(self):
self.user.timezone = 'GMT'
request = HttpRequest()
request.user = self.user
self.assertIsNone(self.middleware.process_request(request))
def setUp(self):
self.user = User.objects.create(username='******')
self.middleware = EnforceTimezoneMiddleware()
def setUp(self):
self.user = User.objects.create_user('testuser', '*****@*****.**')
self.response_mock = Mock()
self.middleware = EnforceTimezoneMiddleware(self.get_response)
