def test_id_generation(self):
request = self.factory.get('/')
middleware = RequestIDMiddleware()
middleware.process_request(request)
self.assertTrue(hasattr(request, 'id'))
test_view(request)
self.assertTrue(request.id in self.handler.messages[0])
def test_id_generation(self):
request = self.factory.get('/')
middleware = RequestIDMiddleware()
middleware.process_request(request)
self.assertTrue(hasattr(request, 'id'))
test_view(request)
self.assertTrue(request.id in self.handler.messages[0])
def test_external_id_missing_in_http_header_should_fallback_to_generated_id(self):
with self.settings(LOG_REQUEST_ID_HEADER='REQUEST_ID_HEADER', GENERATE_REQUEST_ID_IF_NOT_IN_HEADER=True):
request = self.factory.get('/')
middleware = RequestIDMiddleware()
middleware.process_request(request)
self.assertTrue(hasattr(request, 'id'))
test_view(request)
self.assertTrue(request.id in self.handler.messages[0])
def test_external_id_missing_in_http_header_should_fallback_to_generated_id(self):
with self.settings(LOG_REQUEST_ID_HEADER='REQUEST_ID_HEADER', GENERATE_REQUEST_ID_IF_NOT_IN_HEADER=True):
request = self.factory.get('/')
middleware = RequestIDMiddleware()
middleware.process_request(request)
self.assertTrue(hasattr(request, 'id'))
test_view(request)
self.assertTrue(request.id in self.handler.messages[0])
def test_external_id_in_http_header(self):
with self.settings(LOG_REQUEST_ID_HEADER='REQUEST_ID_HEADER'):
request = self.factory.get('/')
request.META['REQUEST_ID_HEADER'] = 'some_request_id'
middleware = RequestIDMiddleware()
middleware.process_request(request)
self.assertEqual(request.id, 'some_request_id')
test_view(request)
self.assertTrue('some_request_id' in self.handler.messages[0])
def test_external_id_in_http_header(self):
with self.settings(LOG_REQUEST_ID_HEADER='REQUEST_ID_HEADER'):
request = self.factory.get('/')
request.META['REQUEST_ID_HEADER'] = 'some_request_id'
middleware = RequestIDMiddleware()
middleware.process_request(request)
self.assertEqual(request.id, 'some_request_id')
test_view(request)
self.assertTrue('some_request_id' in self.handler.messages[0])
def test_external_id_in_http_header(self):
with self.settings(LOG_REQUEST_ID_HEADER="REQUEST_ID_HEADER"):
request = self.factory.get("/")
request.META["REQUEST_ID_HEADER"] = "some_request_id"
middleware = RequestIDMiddleware()
middleware.process_request(request)
self.assertEqual(request.id, "some_request_id")
test_view(request)
self.assertTrue("some_request_id" in self.handler.messages[0])
def test_response_header_unset(self):
with self.settings(LOG_REQUEST_ID_HEADER='REQUEST_ID_HEADER'):
request = self.factory.get('/')
request.META['REQUEST_ID_HEADER'] = 'some_request_id'
middleware = RequestIDMiddleware()
middleware.process_request(request)
response = test_view(request)
self.assertFalse(response.has_header('REQUEST_ID'))
def test_response_header_unset(self):
with self.settings(LOG_REQUEST_ID_HEADER='REQUEST_ID_HEADER'):
request = self.factory.get('/')
request.META['REQUEST_ID_HEADER'] = 'some_request_id'
middleware = RequestIDMiddleware()
middleware.process_request(request)
response = test_view(request)
self.assertFalse(response.has_header('REQUEST_ID'))
def test_response_header_unset(self):
with self.settings(LOG_REQUEST_ID_HEADER="REQUEST_ID_HEADER"):
request = self.factory.get("/")
request.META["REQUEST_ID_HEADER"] = "some_request_id"
middleware = RequestIDMiddleware()
middleware.process_request(request)
response = test_view(request)
self.assertFalse(response.has_header("REQUEST_ID"))
def test_log_requests(self):
class DummyUser(object):
pk = 'fake_pk'
with self.settings(LOG_REQUESTS=True):
request = self.factory.get('/')
request.user = DummyUser()
middleware = RequestIDMiddleware()
middleware.process_request(request)
response = test_view(request)
middleware.process_response(request, response)
self.assertEqual(len(self.handler.messages), 2)
self.assertTrue('fake_pk' in self.handler.messages[1])
def test_log_requests(self):
class DummyUser(object):
pk = "fake_pk"
with self.settings(LOG_REQUESTS=True):
request = self.factory.get("/")
request.user = DummyUser()
middleware = RequestIDMiddleware()
middleware.process_request(request)
response = test_view(request)
middleware.process_response(request, response)
self.assertEqual(len(self.handler.messages), 2)
self.assertTrue("fake_pk" in self.handler.messages[1])
def test_log_user_attribute(self):
class DummyUser(object):
pk = 'fake_pk'
username = '******'
with self.settings(LOG_REQUESTS=True, LOG_USER_ATTRIBUTE='username'):
request = self.factory.get('/')
request.user = DummyUser()
middleware = RequestIDMiddleware()
middleware.process_request(request)
response = test_view(request)
middleware.process_response(request, response)
self.assertEqual(len(self.handler.messages), 2)
self.assertTrue('fake_username' in self.handler.messages[1])
def test_custom_request_id_is_used(self):
request = self.factory.get('/')
test_view(request)
self.assertTrue('[-]' in self.handler.messages[0])
def test_default_no_request_id_is_used(self):
request = self.factory.get('/')
test_view(request)
self.assertTrue(DEFAULT_NO_REQUEST_ID in self.handler.messages[0])
