def test_request_history():
class Non_deep_copiable(object):
def __deepcopy__(self, memodict={}):
raise ValueError()
body = Non_deep_copiable()
request = HttpRequest('GET', 'http://localhost/', {'user-agent': 'test_request_history'})
request.body = body
request_history = RequestHistory(request)
assert request_history.http_request.headers == request.headers
assert request_history.http_request.url == request.url
assert request_history.http_request.method == request.method
def test_http_logger_with_generator_body():
class MockHandler(logging.Handler):
def __init__(self):
super(MockHandler, self).__init__()
self.messages = []
def reset(self):
self.messages = []
def emit(self, record):
self.messages.append(record)
mock_handler = MockHandler()
logger = logging.getLogger("testlogger")
logger.addHandler(mock_handler)
logger.setLevel(logging.DEBUG)
policy = HttpLoggingPolicy(logger=logger)
universal_request = HttpRequest('GET', 'http://127.0.0.1/')
mock = Mock()
mock.__class__ = types.GeneratorType
universal_request.body = mock
http_response = HttpResponse(universal_request, None)
http_response.status_code = 202
request = PipelineRequest(universal_request, PipelineContext(None))
policy.on_request(request)
response = PipelineResponse(request, http_response, request.context)
policy.on_response(request, response)
assert all(m.levelname == 'INFO' for m in mock_handler.messages)
assert len(mock_handler.messages) == 6
assert mock_handler.messages[
0].message == "Request URL: 'http://127.0.0.1/'"
assert mock_handler.messages[1].message == "Request method: 'GET'"
assert mock_handler.messages[2].message == 'Request headers:'
assert mock_handler.messages[3].message == 'File upload'
assert mock_handler.messages[4].message == 'Response status: 202'
assert mock_handler.messages[5].message == 'Response headers:'
mock_handler.reset()
