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()
