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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pintest-cli
-Version: 0.2.6
+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.6 → 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.6 → 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.6 → pintest_cli-0.2.7}/pintest/cloud_mapping_db.py

@@ -142,15 +142,25 @@ class CloudMappingDB:
         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():
             compact_ranges = rs.to_compact_string()
             key = (test_name, file_path)
-            if key not in cached_state or cached_state[key] != compact_ranges:
+            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:

{pintest_cli-0.2.6 → 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.6 → pintest_cli-0.2.7}/pintest/push_cache.py

@@ -23,39 +23,45 @@ class PushCache:
                     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], str]:
+    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}
+            {(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 FROM push_cache WHERE branch = ?",
+                "SELECT test_name, file_path, ranges, duration_ms FROM push_cache WHERE branch = ?",
                 (branch,)
             )
-            return {(row[0], row[1]): row[2] for row in cursor}
+            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[str, str]]):
+    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": ...}]
+            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, last_pushed_at)
-                VALUES (?, ?, ?, ?, CURRENT_TIMESTAMP)
+                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"]) for m in mappings])
+            """, [(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.6 → 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.6
+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.6 → pintest_cli-0.2.7}/pintest_cli.egg-info/SOURCES.txt

@@ -10,6 +10,7 @@ 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.6 → 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.6",
+    version="0.2.7",
     description="Run only the tests affected by your code changes.",
     long_description=long_description,
     long_description_content_type="text/markdown",