From cbaeebaecf7ba5fb9b41a2c25497ab4fe7d30d2d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Dariusz=20K=C4=99dzierski?= <dkedzierski@syncad.com>
Date: Mon, 4 Jan 2021 09:50:32 +0100
Subject: [PATCH] Set tests ids file from env variable
---
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 2b837e8..2e73189 100644
--- a/tavern_time_request/request.py
+++ b/tavern_time_request/request.py
@@ -1,6 +1,7 @@
import logging
import requests
import csv
+import os
from tavern.util import exceptions
from tavern._plugins.rest.request import RestRequest
from time import perf_counter as perf
@@ -13,10 +14,12 @@ class HivemindRequest(RestRequest):
try:
start_time = perf()
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)
+ file_name = os.getenv("HIVEMIND_BENCHMARKS_IDS_FILE", None)
+ if file_name is not None:
+ with open(file_name, "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