Python CidMiddleware.process_response示例

示例#1

文件： test_middleware.py 项目： gryphonlabs/cid

 def test_process_response_with_no_cid(self, get_cid):
     get_cid.return_value = None
     request = Mock()
     response = {}
     middleware = CidMiddleware()
     middleware.process_response(request, response)
     self.assertNotIn('X_CORRELATION_ID', response.keys())

示例#2

文件： test_middleware.py 项目： gryphonlabs/cid

 def test_process_response(self, get_cid):
     get_cid.return_value = self.cid
     request = Mock()
     response = {}
     middleware = CidMiddleware()
     middleware.process_response(request, response)
     self.assertEqual(response['X_CORRELATION_ID'], self.cid)

示例#3

 def test_process_response_with_no_cid(self, get_cid):
     get_cid.return_value = None
     request = Mock()
     response = {}
     middleware = CidMiddleware()
     response = middleware.process_response(request, response)
     self.assertNotIn('X_CORRELATION_ID', response.keys())

示例#4

 def test_process_response_custom_header_name(self, get_cid):
     get_cid.return_value = self.cid
     request = Mock()
     response = {}
     middleware = CidMiddleware()
     response = middleware.process_response(request, response)
     self.assertEqual(response['X-Custom-Name'], self.cid)

示例#5

 def test_process_response(self, get_cid):
     get_cid.return_value = self.cid
     request = Mock()
     response = {}
     middleware = CidMiddleware()
     response = middleware.process_response(request, response)
     self.assertEqual(response['X_CORRELATION_ID'], self.cid)