Skip to content
Snippets Groups Projects
Commit 9503e08a authored by Dariusz Kędzierski's avatar Dariusz Kędzierski
Browse files

Save request data in proxy file

parent 5878fbd4
No related tags found
No related merge requests found
...@@ -6,6 +6,7 @@ from tavern.util import exceptions ...@@ -6,6 +6,7 @@ from tavern.util import exceptions
from tavern._plugins.rest.request import RestRequest from tavern._plugins.rest.request import RestRequest
from time import perf_counter as perf from time import perf_counter as perf
from uuid import uuid4 from uuid import uuid4
from json import dumps
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
...@@ -16,10 +17,10 @@ class HivemindRequest(RestRequest): ...@@ -16,10 +17,10 @@ class HivemindRequest(RestRequest):
test_id = uuid4() test_id = uuid4()
file_name = os.getenv("HIVEMIND_BENCHMARKS_IDS_FILE", None) file_name = os.getenv("HIVEMIND_BENCHMARKS_IDS_FILE", None)
if file_name is not None: if file_name is not None:
self._request_args['json']['id'] = str(test_id)
with open(file_name, "a") as f: with open(file_name, "a") as f:
writer = csv.writer(f) writer = csv.writer(f)
writer.writerow([str(test_id), start_time]) writer.writerow([str(test_id), start_time, dumps(self._request_args['json'])])
self._request_args['json']['id'] = str(test_id)
return self._prepared() return self._prepared()
except requests.exceptions.RequestException as e: except requests.exceptions.RequestException as e:
logger.exception("Error running prepared request") logger.exception("Error running prepared request")
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment