class AuthenticatorMiddlewareTest(TestCase): client_class = RequestFactory def setUp(self): self.middleware = AuthenticatorMiddleware() def test_existing_user(self): user = create_user() request = self.client.get("/", HTTP_X_TOKEN=user.token) self.middleware.process_request(request) self.assertEqual(request.user, user) def test_not_existing_user(self): request = self.client.get("/", HTTP_X_TOKEN=uuid4()) self.middleware.process_request(request) self.assertIsNone(request.user) def test_missing_header(self): request = self.client.get("/") self.middleware.process_request(request) self.assertIsNone(request.user)
def setUp(self): self.middleware = AuthenticatorMiddleware()