pytest-testinel 0.2.1__py3-none-any.whl → 0.3.1__py3-none-any.whl

pytest_testinel/results_reporter.py

@@ -1,4 +1,4 @@
-import abc, datetime, json, os, uuid
+import abc, datetime, json, os, queue, threading, uuid
 from urllib.parse import unquote, urlparse
 
 import requests
@@ -14,6 +14,18 @@ class ReportingBackend(abc.ABC):
     def on_end(self) -> None:
         return
 
+    def request_upload_link(self, filename: str) -> dict | None:
+        return None
+
+    def upload_screenshot(
+        self,
+        upload_url: str,
+        method: str,
+        headers: dict,
+        filename: str,
+    ) -> None:
+        return
+
 
 class NoopReportingBackend(ReportingBackend):
     def record_event(self, event: dict) -> None:
@@ -47,6 +59,22 @@ class HttpReportingBackend(ReportingBackend):
     def record_event(self, event: dict) -> None:
         requests.post(self.url, json=event, verify=False)
 
+    def request_upload_link(self, filename: str) -> dict | None:
+        upload_url = f"{self.url.rstrip('/')}/screenshots/upload-link/"
+        response = requests.post(upload_url, json={"filename": filename}, verify=False)
+        response.raise_for_status()
+        return response.json()
+
+    def upload_screenshot(
+        self,
+        upload_url: str,
+        method: str,
+        headers: dict,
+        filename: str,
+    ) -> None:
+        with open(filename, "rb") as f:
+            requests.request(method, upload_url, data=f, headers=headers)
+
 
 class ResultsReporter:
     run_id: str
@@ -58,6 +86,9 @@ class ResultsReporter:
         self.dsn = dsn
         self.run_id = str(uuid.uuid4())
         self.tests = []
+        self.screenshots: list[str] = []
+        self._upload_queue: queue.Queue | None = None
+        self._uploader: threading.Thread | None = None
         if backend:
             self.backend = backend
         else:
@@ -94,6 +125,10 @@ class ResultsReporter:
         )
 
     def report_end(self) -> None:
+        if self._upload_queue is not None:
+            self._upload_queue.put(None)
+        if self._uploader is not None:
+            self._uploader.join(timeout=10)
         self.backend.record_event(
             {
                 "run_id": self.run_id,
@@ -110,5 +145,62 @@ class ResultsReporter:
                 "event": event,
                 "timestamp": datetime.datetime.now(datetime.UTC).isoformat(),
                 "payload": payload,
+                "screenshots": self.screenshots
             }
         )
+        self.screenshots = []
+
+    def report_screenshot(self, filename: str) -> None:
+        upload_info = None
+        try:
+            upload_info = self.backend.request_upload_link(filename)
+        except Exception:
+            upload_info = None
+
+        if not upload_info:
+            self.screenshots.append(filename)
+            return
+
+        object_key = upload_info.get("object_key")
+        if object_key:
+            self.screenshots.append(object_key)
+        else:
+            self.screenshots.append(filename)
+
+        self._ensure_uploader()
+        upload_url = upload_info.get("upload_url")
+        method = upload_info.get("method", "PUT")
+        headers = upload_info.get("headers", {})
+        if self._upload_queue is not None and upload_url:
+            self._upload_queue.put((upload_url, method, headers, filename))
+
+    def _ensure_uploader(self) -> None:
+        if self._uploader is not None:
+            return
+        self._upload_queue = queue.Queue()
+        self._uploader = threading.Thread(
+            target=self._upload_loop,
+            name="testinel-screenshot-uploader",
+            daemon=True,
+        )
+        self._uploader.start()
+
+    def _upload_loop(self) -> None:
+        if self._upload_queue is None:
+            return
+        while True:
+            item = self._upload_queue.get()
+            if item is None:
+                break
+            upload_url, method, headers, filename = item
+            try:
+                self.backend.upload_screenshot(
+                    upload_url=upload_url,
+                    method=method,
+                    headers=headers,
+                    filename=filename,
+                )
+            except Exception:
+                pass
+            finally:
+                self._upload_queue.task_done()

pytest_testinel/testinel.py

@@ -50,6 +50,42 @@ def _get_test_reporter() -> ResultsReporter:
     return _test_reporter
 
 
+def _safe_path(value: object) -> str:
+    try:
+        path = os.fspath(value)
+    except TypeError:
+        return str(value)
+    if isinstance(path, bytes):
+        try:
+            return path.decode()
+        except Exception:
+            return str(path)
+    return path
+
+
+def _patch_selenium_save_screenshot() -> None:
+    try:
+        from selenium.webdriver.remote.webdriver import WebDriver
+    except Exception:
+        return
+
+    original = getattr(WebDriver, "save_screenshot", None)
+    if original is None or getattr(original, "_testinel_patched", False):
+        return
+
+    def patched(self, filename, *args, **kwargs):
+        result = original(self, filename, *args, **kwargs)
+        try:
+            _get_test_reporter().report_screenshot(_safe_path(filename))
+        except Exception:
+            return result
+        return result
+
+    patched._testinel_patched = True
+    patched._testinel_original = original
+    WebDriver.save_screenshot = patched
+
+
 def serialize_repr(long_repr: ExceptionChainRepr) -> dict:
     return asdict(long_repr)
 
@@ -143,3 +179,6 @@ def reporter(request):
 def pytest_collection_finish(session):
     tests = [to_test_dict(item) for item in session.items]
     _get_test_reporter().tests = tests
+
+
+_patch_selenium_save_screenshot()

{pytest_testinel-0.2.1.dist-info → pytest_testinel-0.3.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytest-testinel
-Version: 0.2.1
+Version: 0.3.1
 Summary: Testinel’s pytest plugin captures structured test execution data directly from pytest and sends it to Testinel, where your test results become searchable, comparable, and actually useful.
 Author: Volodymyr Obrizan
 Author-email: Volodymyr Obrizan <obrizan@first.institute>

pytest_testinel-0.3.1.dist-info/RECORD

@@ -0,0 +1,8 @@
+pytest_testinel/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pytest_testinel/results_reporter.py,sha256=vxifbwbc9NEslwjRqJPxp7C1LFCqV6dw0YUf1UC5aQQ,6280
+pytest_testinel/testinel.py,sha256=NXwUWpE_zSQ_c8BKr64WdZV0vsnnveOF1-vsHEO6Sqs,5166
+pytest_testinel-0.3.1.dist-info/licenses/LICENSE,sha256=g-9SNaCTpjdaj63-SMa03OHuKdcZXsXk4cRKZcLfD1c,1065
+pytest_testinel-0.3.1.dist-info/WHEEL,sha256=iHtWm8nRfs0VRdCYVXocAWFW8ppjHL-uTJkAdZJKOBM,80
+pytest_testinel-0.3.1.dist-info/entry_points.txt,sha256=-vjKXhLEcZLenCKBgXoFNmlNc5DM7qI04yqsDUWH-Ok,49
+pytest_testinel-0.3.1.dist-info/METADATA,sha256=Kzmt48bwY3QHwKqyFrc_9ZUBXg2KKwepqPfgtfZi4yg,1903
+pytest_testinel-0.3.1.dist-info/RECORD,,

{pytest_testinel-0.2.1.dist-info → pytest_testinel-0.3.1.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: uv 0.9.26
+Generator: uv 0.9.30
 Root-Is-Purelib: true
 Tag: py3-none-any

pytest_testinel-0.2.1.dist-info/RECORD

@@ -1,8 +0,0 @@
-pytest_testinel/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pytest_testinel/results_reporter.py,sha256=HSLdSumSjHswJ2oVbVLVMKVk1dY8sxaGKGfVtp977ic,3283
-pytest_testinel/testinel.py,sha256=pIA05QYyqAex5Vn9NEsApBRi7rDIbYtnsKTfX4l3rT0,4152
-pytest_testinel-0.2.1.dist-info/licenses/LICENSE,sha256=g-9SNaCTpjdaj63-SMa03OHuKdcZXsXk4cRKZcLfD1c,1065
-pytest_testinel-0.2.1.dist-info/WHEEL,sha256=XV0cjMrO7zXhVAIyyc8aFf1VjZ33Fen4IiJk5zFlC3g,80
-pytest_testinel-0.2.1.dist-info/entry_points.txt,sha256=-vjKXhLEcZLenCKBgXoFNmlNc5DM7qI04yqsDUWH-Ok,49
-pytest_testinel-0.2.1.dist-info/METADATA,sha256=4TrN9WmDuDh3j31aN9XBjY1CymVqk4-LvX08yrs_QPA,1903
-pytest_testinel-0.2.1.dist-info/RECORD,,