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)