feat(disk_perf): add support of uri for previous csv file

Signed-off-by: Mathieu Labourier <[email protected]>

Author: Millefeuille42

tests/storage/benchmarks/conftest.py

@@ -1,7 +1,11 @@
+import logging
+import os
 import pytest
 import tempfile
-import os
-import logging
+import urllib.request
+from urllib.parse import urlparse
+from uuid import uuid4
+
 
 from lib.commands import SSHCommandFailed
 from .helpers import load_results_from_csv
@@ -92,13 +96,18 @@ def pytest_addoption(parser):
         "--prev-csv",
         action="store",
         default=None,
-        help="Path to previous CSV results file for comparison",
+        help="Path/URI to previous CSV results file for comparison",
     )
 
 @pytest.fixture(scope="session")
-def prev_results(request):
-    csv_path = request.config.getoption("--prev-csv")
-    results = {}
-    if csv_path and os.path.exists(csv_path):
-        load_results_from_csv(csv_path)
-    return results
+def prev_results(pytestconfig):
+    csv_uri = pytestconfig.getoption("--prev-csv")
+    if not csv_uri:
+        return {}
+    csv_path = csv_uri
+    if urlparse(csv_uri).scheme != "":
+        csv_path = f"{uuid4()}.csv"
+        urllib.request.urlretrieve(csv_uri, csv_path)
+    if not os.path.exists(csv_path):
+        raise FileNotFoundError(csv_path)
+    return load_results_from_csv(csv_path)