def test_log_response_connected(self): expected_code = 200 expected_data = "data" expected_debug_data = "* text\n>> data out\n" response = Response.connection_successful( MockCurlSimple( info={pycurl.RESPONSE_CODE: expected_code}, output=expected_data.encode("utf-8"), debug_output=expected_debug_data.encode("utf-8"), request=fixture_request(), ) ) self.com_logger.log_response(response) self.reporter.assert_reports( fixture_report_item_list_on_success( response.request.url, expected_code, expected_data, expected_debug_data ) ) logger_calls = fixture_logger_calls_on_success( response.request.url, expected_code, expected_data, expected_debug_data ) self.assertEqual(logger_calls, self.logger.mock_calls)
def fixture_response_connected(self, response_code): handle = MockCurl({pycurl.RESPONSE_CODE: response_code}) handle.request_obj = Request(RequestTarget(self.host), RequestData(self.request)) handle.output_buffer = io.BytesIO() handle.output_buffer.write(self.data) return Response.connection_successful(handle)
def fixture_response_connected(self, response_code): handle = MockCurl({pycurl.RESPONSE_CODE: response_code}) handle.request_obj = Request( RequestTarget(self.host), RequestData(self.request) ) handle.output_buffer = io.BytesIO() handle.output_buffer.write(self.data) return Response.connection_successful(handle)