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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pintest-cli
-Version: 0.2.6
+Version: 0.2.8
 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.8}/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",
@@ -298,7 +299,7 @@ def main():
     parser.add_argument(
         "--mapping-db",
         type=Path,
-        help="Path to mapping database (default: <repo>/.test_mapping.db)"
+        help="Path to mapping database (default: <repo>/.pintest/test_mapping.db)"
     )
     parser.add_argument(
         "--test-dir",
@@ -315,7 +316,19 @@ def main():
     args = parser.parse_args()
     
     repo_root = args.repo_root.resolve()
-    mapping_db = args.mapping_db or (repo_root / ".test_mapping.db")
+    if args.mapping_db:
+        mapping_db = args.mapping_db
+    else:
+        mapping_db = repo_root / ".pintest" / "test_mapping.db"
+        legacy_db = repo_root / ".test_mapping.db"
+        if not mapping_db.exists() and legacy_db.exists():
+            mapping_db.parent.mkdir(parents=True, exist_ok=True)
+            try:
+                legacy_db.rename(mapping_db)
+            except Exception:
+                mapping_db = legacy_db
+        else:
+            mapping_db.parent.mkdir(parents=True, exist_ok=True)
     
     return build_mapping_iteratively(
         repo_root,

{pintest_cli-0.2.6 → pintest_cli-0.2.8}/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="])
@@ -188,7 +188,14 @@ def cmd_run(args):
         # 2. Try Local V2 Mapping DB
         mapping_db = args.mapping_db if hasattr(args, 'mapping_db') else None
         if not mapping_db:
-            mapping_db = repo_root / ".test_mapping.db"
+            mapping_db = repo_root / ".pintest" / "test_mapping.db"
+            legacy_db = repo_root / ".test_mapping.db"
+            if not mapping_db.exists() and legacy_db.exists():
+                mapping_db.parent.mkdir(parents=True, exist_ok=True)
+                try:
+                    legacy_db.rename(mapping_db)
+                except Exception:
+                    mapping_db = legacy_db
             
         if not mapping_db.exists():
             print("🏗️  No local mapping DB found. Initializing build...", file=sys.stderr)
@@ -303,7 +310,17 @@ def cmd_build_mapping(args):
     from .build_mapping_iterative import build_mapping_iteratively
     
     repo_root = args.repo_root.resolve()
-    mapping_db = args.mapping_db or (repo_root / ".test_mapping.db")
+    if args.mapping_db:
+        mapping_db = args.mapping_db
+    else:
+        mapping_db = repo_root / ".pintest" / "test_mapping.db"
+        legacy_db = repo_root / ".test_mapping.db"
+        if not mapping_db.exists() and legacy_db.exists():
+            mapping_db.parent.mkdir(parents=True, exist_ok=True)
+            try:
+                legacy_db.rename(mapping_db)
+            except Exception:
+                mapping_db = legacy_db
     
     # Load config to get default test_dir if it was saved during track
     from .config import Config
@@ -618,7 +635,7 @@ Examples:
     update_parser.add_argument(
         "--mapping-db",
         type=Path,
-        help="Path to mapping database (default: <repo>/.test_mapping.db)"
+        help="Path to mapping database (default: <repo>/.pintest/test_mapping.db)"
     )
     update_parser.add_argument(
         "-v", "--verbose",
@@ -648,7 +665,7 @@ Examples:
     build_parser.add_argument(
         "--mapping-db",
         type=Path,
-        help="Path to mapping database (default: <repo>/.test_mapping.db)"
+        help="Path to mapping database (default: <repo>/.pintest/test_mapping.db)"
     )
     build_parser.add_argument(
         "--test-dir",

{pintest_cli-0.2.6 → pintest_cli-0.2.8}/pintest/cloud_mapping_db.py

@@ -138,19 +138,46 @@ 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"
+        pintest_dir = coverage_file.parent / ".pintest"
+        pintest_dir.mkdir(parents=True, exist_ok=True)
+        
+        cache_db_path = pintest_dir / "push_cache.db"
+        legacy_cache = coverage_file.parent / ".pintest_push_cache.db"
+        if not cache_db_path.exists() and legacy_cache.exists():
+            try:
+                legacy_cache.rename(cache_db_path)
+            except Exception:
+                cache_db_path = legacy_cache
+
         push_cache = PushCache(cache_db_path)
         cached_state = push_cache.get_cached_state(self._branch)
 
+        import json
+        durations_file = pintest_dir / "durations.json"
+        legacy_durations = coverage_file.parent / ".pintest_durations.json"
+        if not durations_file.exists() and legacy_durations.exists():
+            try:
+                legacy_durations.rename(durations_file)
+            except Exception:
+                durations_file = legacy_durations
+
+        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.8}/pintest/pre_commit_hook.py

@@ -71,7 +71,7 @@ def ensure_git_lfs(repo_root: Path, verbose: bool = False) -> bool:
     """
     Ensure Git LFS is initialized in the repository.
     
-    This is needed if the repo uses Git LFS for storing large files like .test_mapping.db.
+    This is needed if the repo uses Git LFS for storing large files like .pintest/test_mapping.db.
     Safe to run multiple times.
     
     Args:
@@ -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
@@ -1076,7 +1079,7 @@ def main():
         "--mapping-db",
         type=Path,
         default=None,
-        help="Path to test mapping database (default: <repo>/.test_mapping.db)"
+        help="Path to test mapping database (default: <repo>/.pintest/test_mapping.db)"
     )
     parser.add_argument(
         "--test-dir",
@@ -1115,13 +1118,21 @@ def main():
     repo_root = args.repo_root.resolve()
     
     # Set up log file for tracking all output
-    log_file = repo_root / ".pre-commit.log"
+    pintest_dir = repo_root / ".pintest"
+    pintest_dir.mkdir(parents=True, exist_ok=True)
+    log_file = pintest_dir / "pre-commit.log"
     
     # Use TeeOutput to write to both stdout and log file
     with TeeOutput(log_file):
         # Default mapping database location
         if args.mapping_db is None:
-            mapping_db = repo_root / ".test_mapping.db"
+            mapping_db = pintest_dir / "test_mapping.db"
+            legacy_db = repo_root / ".test_mapping.db"
+            if not mapping_db.exists() and legacy_db.exists():
+                try:
+                    legacy_db.rename(mapping_db)
+                except Exception:
+                    mapping_db = legacy_db
         else:
             mapping_db = args.mapping_db
         

{pintest_cli-0.2.6 → pintest_cli-0.2.8}/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.8/pintest/pytest_plugin.py

@@ -0,0 +1,26 @@
+"""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:
+        pintest_dir = Path(session.config.rootdir) / ".pintest"
+        pintest_dir.mkdir(parents=True, exist_ok=True)
+        out_file = pintest_dir / "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.8}/pintest/update_mapping.py

@@ -42,7 +42,16 @@ def update_mapping(
             coverage_file = repo_root / "coverage" / ".coverage"
     
     if mapping_db is None:
-        mapping_db = repo_root / ".test_mapping.db"
+        mapping_db = repo_root / ".pintest" / "test_mapping.db"
+        legacy_db = repo_root / ".test_mapping.db"
+        if not mapping_db.exists() and legacy_db.exists():
+            mapping_db.parent.mkdir(parents=True, exist_ok=True)
+            try:
+                legacy_db.rename(mapping_db)
+            except Exception:
+                mapping_db = legacy_db
+        else:
+            mapping_db.parent.mkdir(parents=True, exist_ok=True)
     
     if verbose:
         pass
@@ -101,7 +110,7 @@ def main():
     parser.add_argument(
         "--mapping-db",
         type=Path,
-        help="Path to mapping database (default: <repo>/.test_mapping.db)"
+        help="Path to mapping database (default: <repo>/.pintest/test_mapping.db)"
     )
     parser.add_argument(
         "-v", "--verbose",

{pintest_cli-0.2.6 → pintest_cli-0.2.8}/pintest_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pintest-cli
-Version: 0.2.6
+Version: 0.2.8
 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.8}/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.8}/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.8",
     description="Run only the tests affected by your code changes.",
     long_description=long_description,
     long_description_content_type="text/markdown",