pintest-cli 0.2.5__tar.gz → 0.2.7__tar.gz

{pintest_cli-0.2.5 → pintest_cli-0.2.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pintest-cli
-Version: 0.2.5
+Version: 0.2.7
 Summary: Run only the tests affected by your code changes.
 Author: Pintest Contributors
 Classifier: Development Status :: 3 - Alpha

{pintest_cli-0.2.5 → pintest_cli-0.2.7}/pintest/build_mapping_iterative.py

@@ -52,6 +52,7 @@ def run_test_chunk_with_mapping(
     # Run chunk of tests with coverage
     cmd = [
         sys.executable, "-m", "pytest",
+        "-p", "pintest.pytest_plugin",
         "--cov",
         "--cov-context=test",
         "--cov-append",

{pintest_cli-0.2.5 → pintest_cli-0.2.7}/pintest/cli.py

@@ -138,7 +138,7 @@ class PintestRunner:
             return 0
         
         # Build pytest command
-        cmd = [sys.executable, "-m", "pytest"]
+        cmd = [sys.executable, "-m", "pytest", "-p", "pintest.pytest_plugin"]
         
         # Automatically generate coverage for local updates (suppress terminal report)
         cmd.extend(["--cov", "--cov-context=test", "--cov-append", "--cov-report="])

{pintest_cli-0.2.5 → pintest_cli-0.2.7}/pintest/cloud_mapping_db.py

@@ -108,6 +108,7 @@ class CloudMappingDB:
             True on success, False on failure
         """
         from .coverage_mapper import CoverageMapper   # existing module
+        from .push_cache import PushCache
 
         if not coverage_file.exists():
             if verbose:
@@ -137,19 +138,36 @@ class CloudMappingDB:
                         test_file_ranges[key] = RangeSet()
                     test_file_ranges[key].add_range(line_num, line_num)
 
+        cache_db_path = coverage_file.parent / ".pintest_push_cache.db"
+        push_cache = PushCache(cache_db_path)
+        cached_state = push_cache.get_cached_state(self._branch)
+
+        import json
+        durations_file = coverage_file.parent / ".pintest_durations.json"
+        try:
+            test_durations = json.loads(durations_file.read_text()) if durations_file.exists() else {}
+        except Exception:
+            test_durations = {}
+
         mappings = []
         for (test_name, file_path), rs in test_file_ranges.items():
-            mappings.append({
-                "test_name": test_name,
-                "file_path": file_path,
-                "ranges": rs.to_compact_string(),
-            })
+            compact_ranges = rs.to_compact_string()
+            key = (test_name, file_path)
+            duration_ms = test_durations.get(test_name, 0)
+            cached_ranges, cached_duration = cached_state.get(key, ("", 0))
+            if cached_ranges != compact_ranges or (duration_ms > 0 and duration_ms != cached_duration):
+                mappings.append({
+                    "test_name": test_name,
+                    "file_path": file_path,
+                    "ranges": compact_ranges,
+                    "duration_ms": duration_ms,
+                })
 
         if not mappings:
-            print("ℹ️  No coverage mappings to push")
+            print("ℹ️  All coverage mappings are up-to-date with Pintest Cloud (0 deltas)")
             return True
 
-        print(f"☁️  Pushing {len(mappings)} mappings to Pintest...", flush=True)
+        print(f"☁️  Pushing {len(mappings)} delta mappings to Pintest...", flush=True)
 
         payload_base = {
             "branch": self._branch,
@@ -179,6 +197,7 @@ class CloudMappingDB:
                 timeout=60,
             )
             resp.raise_for_status()
+            push_cache.batch_upsert(self._branch, chunk_data)
             data = resp.json()
             inserted = data.get('inserted', 0)
             updated = data.get('updated', 0)

{pintest_cli-0.2.5 → pintest_cli-0.2.7}/pintest/pre_commit_hook.py

@@ -426,6 +426,7 @@ def run_test_chunk_with_mapping_update(
     # Run tests with coverage - output flows directly to terminal
     cmd = [
         sys.executable, "-m", "pytest",
+        "-p", "pintest.pytest_plugin",
         "--cov",
         "--cov-context=test",
         "--cov-append",
@@ -558,6 +559,7 @@ def run_single_test_with_mapping_update(
     # Run single test with coverage
     cmd = [
         "python", "-m", "pytest",
+        "-p", "pintest.pytest_plugin",
         "--cov",
         "--cov-context=test",
         "--cov-append",
@@ -1023,6 +1025,7 @@ def run_tests_with_coverage(
     # Build pytest command
     cmd = [
         sys.executable, "-m", "pytest",
+        "-p", "pintest.pytest_plugin",
         "--cov=src",
         "--cov-context=test",
         "--cov-append",  # Append to existing coverage

pintest_cli-0.2.7/pintest/push_cache.py

@@ -0,0 +1,67 @@
+import sqlite3
+from pathlib import Path
+from typing import Dict, List, Tuple
+
+
+class PushCache:
+    """
+    Manages the local SQLite cache for Pintest Cloud delta pushes.
+    Tracks previously synchronized test coverage mappings to filter out unmodified entries.
+    """
+
+    def __init__(self, db_path: Path):
+        self.db_path = db_path
+        self._init_db()
+
+    def _init_db(self):
+        """Initialize the local push cache database schema."""
+        with sqlite3.connect(self.db_path) as conn:
+            conn.execute("PRAGMA journal_mode=WAL")
+            conn.execute("""
+                CREATE TABLE IF NOT EXISTS push_cache (
+                    branch TEXT,
+                    test_name TEXT,
+                    file_path TEXT,
+                    ranges TEXT,
+                    duration_ms INTEGER DEFAULT 0,
+                    last_pushed_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+                    PRIMARY KEY (branch, test_name, file_path)
+                )
+            """)
+            try:
+                conn.execute("ALTER TABLE push_cache ADD COLUMN duration_ms INTEGER DEFAULT 0")
+            except sqlite3.OperationalError:
+                pass
+            conn.execute("CREATE INDEX IF NOT EXISTS idx_push_cache_lookup ON push_cache(branch)")
+
+    def get_cached_state(self, branch: str) -> Dict[Tuple[str, str], Tuple[str, int]]:
+        """
+        Fetch the last successfully pushed state for the active branch.
+
+        Returns:
+            {(test_name, file_path): (ranges_string, duration_ms)}
+        """
+        with sqlite3.connect(self.db_path) as conn:
+            cursor = conn.execute(
+                "SELECT test_name, file_path, ranges, duration_ms FROM push_cache WHERE branch = ?",
+                (branch,)
+            )
+            return {(row[0], row[1]): (row[2], row[3] if row[3] is not None else 0) for row in cursor}
+
+    def batch_upsert(self, branch: str, mappings: List[Dict]):
+        """
+        Atomically update the cache with successfully pushed mappings.
+
+        Args:
+            branch: Active git branch
+            mappings: List of dicts [{"test_name": ..., "file_path": ..., "ranges": ..., "duration_ms": ...}]
+        """
+        with sqlite3.connect(self.db_path) as conn:
+            conn.executemany("""
+                INSERT INTO push_cache (branch, test_name, file_path, ranges, duration_ms, last_pushed_at)
+                VALUES (?, ?, ?, ?, ?, CURRENT_TIMESTAMP)
+                ON CONFLICT(branch, test_name, file_path) DO UPDATE SET
+                    ranges = EXCLUDED.ranges,
+                    duration_ms = CASE WHEN EXCLUDED.duration_ms > 0 THEN EXCLUDED.duration_ms ELSE push_cache.duration_ms END,
+                    last_pushed_at = CURRENT_TIMESTAMP
+            """, [(branch, m["test_name"], m["file_path"], m["ranges"], m.get("duration_ms", 0)) for m in mappings])

pintest_cli-0.2.7/pintest/pytest_plugin.py

@@ -0,0 +1,24 @@
+"""Pytest plugin to measure individual test execution duration for Pintest."""
+import json
+from pathlib import Path
+
+# Stores test_node_id -> duration_ms
+test_durations = {}
+
+
+def pytest_runtest_makereport(item, call):
+    """Record test execution time during the 'call' phase."""
+    if call.when == "call":
+        # call.duration is a float representing exact seconds
+        test_durations[item.nodeid] = int(call.duration * 1000)
+
+
+def pytest_sessionfinish(session, exitstatus):
+    """Dump durations to a temporary file for the CLI push phase."""
+    try:
+        out_file = Path(session.config.rootdir) / ".pintest_durations.json"
+        with open(out_file, "w") as f:
+            json.dump(test_durations, f)
+    except Exception as e:
+        # Silently pass if we cannot write to rootdir
+        pass

{pintest_cli-0.2.5 → pintest_cli-0.2.7}/pintest_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pintest-cli
-Version: 0.2.5
+Version: 0.2.7
 Summary: Run only the tests affected by your code changes.
 Author: Pintest Contributors
 Classifier: Development Status :: 3 - Alpha

{pintest_cli-0.2.5 → pintest_cli-0.2.7}/pintest_cli.egg-info/SOURCES.txt

@@ -9,6 +9,8 @@ pintest/coverage_mapper.py
 pintest/git_diff_parser.py
 pintest/post_commit_hook.py
 pintest/pre_commit_hook.py
+pintest/push_cache.py
+pintest/pytest_plugin.py
 pintest/range_set.py
 pintest/test_mapping_db_v2.py
 pintest/update_mapping.py

{pintest_cli-0.2.5 → pintest_cli-0.2.7}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r", encoding="utf-8") as fh:
 
 setup(
     name="pintest-cli",
-    version="0.2.5",
+    version="0.2.7",
     description="Run only the tests affected by your code changes.",
     long_description=long_description,
     long_description_content_type="text/markdown",