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

Pass time info trough tmp file

parent 77b37f26
No related branches found
No related tags found
No related merge requests found
import logging import logging
import requests import requests
import csv
from tavern.util import exceptions 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
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
class HivemindRequest(RestRequest): class HivemindRequest(RestRequest):
def run(self): def run(self):
try: try:
# add sent timestamp to each request in perf_counter format
start_time = perf() start_time = perf()
headers = self._request_args.get('headers', {}) test_id = uuid4()
headers['Sent-At'] = str(start_time) with open("/tmp/test_ids.csv", "a") as f:
self._request_args['headers'] = headers writer = csv.writer(f)
writer.writerow([str(test_id), start_time])
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