def test_load_response_body(self): mock_request = self._create_mock_request() storage = RequestStorage(base_dir=self.base_dir) request_id = storage.save_request(mock_request, request_body=b'test request body') mock_response = self._create_mock_resonse() storage.save_response(request_id, mock_response, response_body=b'test response body') response_body = storage.load_response_body(request_id) self.assertEqual(response_body, b'test response body')
def test_load_response_body(self): body = b'test response body' request = self._create_request() storage = RequestStorage(base_dir=self.base_dir) storage.save_request(request) mock_response = self._create_response(body=body) storage.save_response(request.id, mock_response) response_body = storage.load_response_body(request.id) self.assertEqual(body, response_body)
def test_load_response_body_encoded_error(self): body = b'test response body' request = self._create_request() storage = RequestStorage(base_dir=self.base_dir) storage.save_request(request) response = self._create_response(body=body) response.headers['Content-Encoding'] = 'gzip' storage.save_response(request.id, response) response_body = storage.load_response_body(request.id) self.assertEqual(body, response_body)
def test_load_response_body_encoded(self): io = BytesIO() with gzip.GzipFile(fileobj=io, mode='wb') as f: f.write(b'test response body') mock_request = self._create_mock_request() storage = RequestStorage(base_dir=self.base_dir) request_id = storage.save_request(mock_request, request_body=b'test request body') mock_response = self._create_mock_resonse() mock_response.headers['Content-Encoding'] = 'gzip' storage.save_response(request_id, mock_response, response_body=io.getvalue()) response_body = storage.load_response_body(request_id) self.assertEqual(response_body, b'test response body')