pytest-visionspec 0.2.0__tar.gz → 0.2.1__tar.gz

{pytest_visionspec-0.2.0 → pytest_visionspec-0.2.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytest-visionspec
-Version: 0.2.0
+Version: 0.2.1
 Summary: Pytest plugin that auto-reports test results with screenshots to VisionSpec
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown

{pytest_visionspec-0.2.0 → pytest_visionspec-0.2.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "pytest-visionspec"
-version = "0.2.0"
+version = "0.2.1"
 description = "Pytest plugin that auto-reports test results with screenshots to VisionSpec"
 readme = "README.md"
 requires-python = ">=3.9"

{pytest_visionspec-0.2.0 → pytest_visionspec-0.2.1}/pytest_visionspec/__init__.py

@@ -3,13 +3,13 @@
 Usage:
     pip install pytest-visionspec
     Set env vars: VS_API_KEY, VS_API_URL
-    Tag tests: @pytest.mark.vs("journey-id")
+    Tag tests: @pytest.mark.vs("journey-id", revision="slug")
 
 The plugin:
-1. Detects @pytest.mark.vs("journey-id") tags on tests
+1. Detects @pytest.mark.vs("journey-id", revision="slug") tags on tests
 2. After each test, captures a Playwright screenshot (if a page fixture exists)
 3. Uploads screenshots to VisionSpec storage
-4. Posts the result (pass/fail + screenshot_urls + screenshot_labels) to VisionSpec
+4. Posts the result (pass/fail + screenshot_urls + screenshot_metadata) to VisionSpec
 
 Labeled screenshots:
     Use the vs_screenshot fixture to capture labeled screenshots mid-test:
@@ -20,7 +20,9 @@ Labeled screenshots:
         vs_screenshot("dark mode")    # captures + labels
 """
 
+import hashlib
 import os
+import struct
 import sys
 import glob
 from pathlib import Path
@@ -124,9 +126,10 @@ def pytest_runtest_makereport(item, call):
     # Upload any captured screenshots/videos
     screenshot_urls = []
     screenshot_labels = []
+    screenshot_metadata = []
     video_url = None
     try:
-        screenshot_urls, screenshot_labels, video_url = _collect_and_upload_artifacts(item, api_key, api_url)
+        screenshot_urls, screenshot_labels, screenshot_metadata, video_url = _collect_and_upload_artifacts(item, api_key, api_url)
     except Exception:
         pass
 
@@ -158,6 +161,7 @@ def pytest_runtest_makereport(item, call):
                     "versions": _get_versions(),
                     "screenshot_urls": screenshot_urls,
                     "screenshot_labels": screenshot_labels,
+                    "screenshot_metadata": screenshot_metadata,
                     "video_url": video_url,
                     "revision_slug": revision,
                 })
@@ -184,22 +188,39 @@ def _safe_filename(name: str) -> str:
     return name.replace("[", "-").replace("]", "").replace("/", "-").replace(":", "-")
 
 
+def _png_dimensions(data: bytes) -> tuple:
+    """Extract width and height from PNG header without Pillow."""
+    if len(data) < 24 or data[:8] != b'\x89PNG\r\n\x1a\n':
+        return (0, 0)
+    # IHDR chunk starts at byte 8: 4 bytes length + 4 bytes "IHDR" + 4 bytes width + 4 bytes height
+    width, height = struct.unpack('>II', data[16:24])
+    return (width, height)
+
+
+def _file_metadata(data: bytes, url: str) -> dict:
+    """Build screenshot_metadata entry from file bytes and uploaded URL."""
+    width, height = _png_dimensions(data)
+    return {
+        "url": url,
+        "width": width,
+        "height": height,
+        "size_bytes": len(data),
+        "sha256": hashlib.sha256(data).hexdigest(),
+    }
+
+
 def _collect_and_upload_artifacts(item, api_key, api_url):
     """Find screenshots/videos in test-results/ and upload them.
 
-    Returns (screenshot_urls, screenshot_labels, video_url).
-    Labels are extracted from filenames: 'labeled-N-some-label.png' -> 'some label'.
-    The final end-of-test screenshot gets label 'end'.
+    Returns (screenshot_urls, screenshot_labels, screenshot_metadata, video_url).
     """
     screenshot_urls = []
     screenshot_labels = []
+    screenshot_metadata = []
     video_url = None
 
     safe_name = _safe_filename(item.name)
 
-    # Get labeled captures from the vs_screenshot fixture (if used)
-    vs_labels = getattr(item, "_vs_screenshot_labels", [])
-
     patterns = [
         f"test-results/{safe_name}/*.png",
         f"test-results/{safe_name}/*.jpg",
@@ -228,9 +249,10 @@ def _collect_and_upload_artifacts(item, api_key, api_url):
                 "webm": "video/webm", "mp4": "video/mp4",
             }.get(ext, "application/octet-stream")
 
+            file_bytes = path.read_bytes()
             upload_r = httpx.post(f"{api_url}/api/upload",
                 headers={"X-API-Key": api_key, "Content-Type": mime, "X-Filename": path.name},
-                content=path.read_bytes(),
+                content=file_bytes,
                 timeout=30)
 
             if upload_r.status_code == 200:
@@ -238,21 +260,20 @@ def _collect_and_upload_artifacts(item, api_key, api_url):
                 if mime.startswith("video/"):
                     video_url = data["url"]
                 else:
-                    screenshot_urls.append(data["url"])
+                    url = data["url"]
+                    screenshot_urls.append(url)
+                    screenshot_metadata.append(_file_metadata(file_bytes, url))
                     # Derive label from filename
                     stem = path.stem
                     if stem.startswith("labeled-") and len(stem.split("-", 2)) >= 3:
-                        # labeled-0-some-label -> some label
                         label = stem.split("-", 2)[2].replace("-", " ")
-                    elif stem == "end":
+                    elif stem in ("end", "after"):
                         label = "end"
-                    elif stem == "after":
-                        label = "end"  # legacy
                     else:
                         label = ""
                     screenshot_labels.append(label)
 
-    return screenshot_urls, screenshot_labels, video_url
+    return screenshot_urls, screenshot_labels, screenshot_metadata, video_url
 
 
 def _get_versions():

{pytest_visionspec-0.2.0 → pytest_visionspec-0.2.1}/pytest_visionspec.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytest-visionspec
-Version: 0.2.0
+Version: 0.2.1
 Summary: Pytest plugin that auto-reports test results with screenshots to VisionSpec
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown