From b2bac3d0a315cfdfdf09f6a6504461cd7e0e22a3 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Dariusz=20K=C4=99dzierski?= <dkedzierski@syncad.com>
Date: Mon, 21 Dec 2020 14:56:03 +0100
Subject: [PATCH] Pass time info trough tmp file
---
tavern_time_request/request.py | 11 +++++++----
1 file changed, 7 insertions(+), 4 deletions(-)
diff --git a/tavern_time_request/request.py b/tavern_time_request/request.py
index bd270af..2b837e8 100644
--- a/tavern_time_request/request.py
+++ b/tavern_time_request/request.py
@@ -1,19 +1,22 @@
import logging
import requests
+import csv
from tavern.util import exceptions
from tavern._plugins.rest.request import RestRequest
from time import perf_counter as perf
+from uuid import uuid4
logger = logging.getLogger(__name__)
class HivemindRequest(RestRequest):
def run(self):
try:
- # add sent timestamp to each request in perf_counter format
start_time = perf()
- headers = self._request_args.get('headers', {})
- headers['Sent-At'] = str(start_time)
- self._request_args['headers'] = headers
+ test_id = uuid4()
+ with open("/tmp/test_ids.csv", "a") as f:
+ writer = csv.writer(f)
+ writer.writerow([str(test_id), start_time])
+ self._request_args['json']['id'] = str(test_id)
return self._prepared()
except requests.exceptions.RequestException as e:
logger.exception("Error running prepared request")
--
GitLab