def test_load_response(self):
request = self._create_request()
storage = RequestStorage(base_dir=self.base_dir)
storage.save_request(request)
response = self._create_response()
storage.save_response(request.id, response)
requests = storage.load_requests()
self.assertIsNotNone(requests[0].response)
def test_load_response(self):
mock_request = self._create_mock_request()
storage = RequestStorage(base_dir=self.base_dir)
request_id = storage.save_request(mock_request)
mock_response = self._create_mock_resonse()
storage.save_response(request_id, mock_response)
requests = storage.load_requests()
self.assertIsNotNone(requests[0]['response'])
def test_load_response_encoded_body_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_requests()[0].response.body
self.assertEqual(body, response_body)
def test_clear_requests(self):
mock_request_1 = self._create_mock_request()
mock_request_2 = self._create_mock_request()
storage = RequestStorage(base_dir=self.base_dir)
storage.save_request(mock_request_1)
storage.save_request(mock_request_2)
storage.clear_requests()
requests = storage.load_requests()
self.assertFalse(requests)
self.assertFalse(glob.glob(os.path.join(self.base_dir, '.seleniumwire', 'storage-*', '*')))
def test_load_requests(self):
mock_request_1 = self._create_mock_request()
mock_request_2 = self._create_mock_request()
storage = RequestStorage(base_dir=self.base_dir)
request_id1 = storage.save_request(mock_request_1)
request_id2 = storage.save_request(mock_request_2)
requests = storage.load_requests()
self.assertEqual(len(requests), 2)
self.assertEqual(requests[0]['id'], request_id1)
self.assertEqual(requests[1]['id'], request_id2)
self.assertIsNone(requests[0]['response'])
self.assertIsNone(requests[1]['response'])
def test_load_requests(self):
request_1 = self._create_request()
request_2 = self._create_request()
storage = RequestStorage(base_dir=self.base_dir)
storage.save_request(request_1)
storage.save_request(request_2)
requests = storage.load_requests()
self.assertEqual(2, len(requests))
self.assertEqual(request_1.id, requests[0].id)
self.assertEqual(request_2.id, requests[1].id)
self.assertIsNone(requests[0].response)
self.assertIsNone(requests[1].response)
def test_load_response_encoded_body(self):
body = b'test response body'
io = BytesIO()
with gzip.GzipFile(fileobj=io, mode='wb') as f:
f.write(body)
request = self._create_request()
storage = RequestStorage(base_dir=self.base_dir)
storage.save_request(request)
response = self._create_response(body=io.getvalue())
response.headers['Content-Encoding'] = 'gzip'
storage.save_response(request.id, response)
response_body = storage.load_requests()[0].response.body
self.assertEqual(body, response_body)