def test_response_header_set(self):
with self.settings(LOG_REQUEST_ID_HEADER='REQUEST_ID_HEADER', REQUEST_ID_RESPONSE_HEADER='REQUEST_ID'):
request = self.factory.get('/')
request.META['REQUEST_ID_HEADER'] = 'some_request_id'
middleware = RequestIDMiddleware()
middleware.process_request(request)
response = test_view(request)
middleware.process_response(request, response)
self.assertTrue(response.has_header('REQUEST_ID'))
def test_response_header_set(self):
with self.settings(LOG_REQUEST_ID_HEADER='REQUEST_ID_HEADER', REQUEST_ID_RESPONSE_HEADER='REQUEST_ID'):
request = self.factory.get('/')
request.META['REQUEST_ID_HEADER'] = 'some_request_id'
middleware = RequestIDMiddleware()
middleware.process_request(request)
response = test_view(request)
middleware.process_response(request, response)
self.assertTrue(response.has_header('REQUEST_ID'))
def test_response_header_set(self):
with self.settings(LOG_REQUEST_ID_HEADER="REQUEST_ID_HEADER", REQUEST_ID_RESPONSE_HEADER="REQUEST_ID"):
request = self.factory.get("/")
request.META["REQUEST_ID_HEADER"] = "some_request_id"
middleware = RequestIDMiddleware()
middleware.process_request(request)
response = test_view(request)
middleware.process_response(request, response)
self.assertTrue(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])
