pytest-neon 3.0.0__py3-none-any.whl → 3.0.1__py3-none-any.whl

pytest_neon/__init__.py

@@ -9,7 +9,7 @@ from pytest_neon.plugin import (
     neon_engine,
 )
 
-__version__ = "3.0.0"
+__version__ = "3.0.1"
 __all__ = [
     "NeonBranch",
     "neon_apply_migrations",

pytest_neon/plugin.py

@@ -577,6 +577,7 @@ class XdistCoordinator:
     def __init__(self, tmp_path_factory: pytest.TempPathFactory):
         self.worker_id = _get_xdist_worker_id()
         self.is_xdist = self.worker_id != "main"
+        self._worker_count: int | None = None
 
         if self.is_xdist:
             root_tmp_dir = tmp_path_factory.getbasetemp().parent
@@ -584,6 +585,21 @@ class XdistCoordinator:
         else:
             self._lock_dir = None
 
+    def _get_worker_count(self) -> int:
+        """Get the total number of xdist workers."""
+        if self._worker_count is not None:
+            return self._worker_count
+
+        # PYTEST_XDIST_WORKER_COUNT is set by xdist
+        count_str = os.environ.get("PYTEST_XDIST_WORKER_COUNT")
+        if count_str:
+            self._worker_count = int(count_str)
+        else:
+            # Fallback: count from worker ID pattern (gw0, gw1, etc.)
+            # This shouldn't happen in normal xdist runs
+            self._worker_count = 1
+        return self._worker_count
+
     def coordinate_resource(
         self,
         resource_name: str,
@@ -642,6 +658,53 @@ class XdistCoordinator:
         signal_file = self._lock_dir / f"neon_{signal_name}"
         signal_file.write_text("done")
 
+    def signal_worker_done(self) -> None:
+        """Signal that this worker has completed all tests."""
+        if not self.is_xdist or self._lock_dir is None:
+            return
+
+        done_file = self._lock_dir / f"neon_worker_done_{self.worker_id}"
+        done_file.write_text("done")
+
+    def wait_for_all_workers_done(self, timeout: float = 300) -> None:
+        """
+        Wait for all xdist workers to signal completion.
+
+        This ensures the branch isn't deleted while other workers are still
+        running tests.
+
+        Args:
+            timeout: Maximum time to wait in seconds (default: 5 minutes)
+        """
+        if not self.is_xdist or self._lock_dir is None:
+            return
+
+        worker_count = self._get_worker_count()
+        waited = 0.0
+        poll_interval = 0.5
+
+        while waited < timeout:
+            done_count = 0
+            for i in range(worker_count):
+                done_file = self._lock_dir / f"neon_worker_done_gw{i}"
+                if done_file.exists():
+                    done_count += 1
+
+            if done_count >= worker_count:
+                return
+
+            time.sleep(poll_interval)
+            waited += poll_interval
+
+        # Timeout - log warning but proceed with cleanup anyway
+        # This prevents infinite hangs if a worker crashes
+        warnings.warn(
+            f"Timeout waiting for all workers to complete after {timeout}s. "
+            f"Only {done_count}/{worker_count} workers signaled completion. "
+            f"Proceeding with branch cleanup.",
+            stacklevel=2,
+        )
+
 
 class EnvironmentManager:
     """Manages DATABASE_URL environment variable lifecycle."""
@@ -870,6 +933,10 @@ def _neon_test_branch(
     This creates a single branch with expiry that all tests share.
     The first worker creates the branch, others reuse it.
 
+    Branch cleanup is coordinated so the creator waits for ALL workers
+    to complete before deleting the branch, preventing connection errors
+    for workers that finish later.
+
     Yields:
         Tuple of (branch, is_creator) where is_creator indicates if this
         worker created the branch (and should run migrations/cleanup).
@@ -893,7 +960,13 @@ def _neon_test_branch(
         yield branch, is_creator
     finally:
         env_manager.restore()
+        # Signal that this worker is done with all tests
+        _neon_xdist_coordinator.signal_worker_done()
+
         if is_creator:
+            # Wait for all other workers to finish before deleting the branch
+            # This prevents "endpoint not found" errors for slower workers
+            _neon_xdist_coordinator.wait_for_all_workers_done()
             _neon_branch_manager.delete_branch(branch.branch_id)
 
 

{pytest_neon-3.0.0.dist-info → pytest_neon-3.0.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytest-neon
-Version: 3.0.0
+Version: 3.0.1
 Summary: Pytest plugin for Neon database branch isolation in tests
 Project-URL: Homepage, https://github.com/ZainRizvi/pytest-neon
 Project-URL: Repository, https://github.com/ZainRizvi/pytest-neon

pytest_neon-3.0.1.dist-info/RECORD

@@ -0,0 +1,8 @@
+pytest_neon/__init__.py,sha256=VNFVMj2RGnWj_CYynDMxIoPZ8XSKcN6v_lT94pAlYgg,404
+pytest_neon/plugin.py,sha256=9wL17R95Wcu2fTti_MGNe4y66BZmcGs86qwgZ4GHD1c,42825
+pytest_neon/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pytest_neon-3.0.1.dist-info/METADATA,sha256=U1j8ZiEFOTSl_h4VNmnf7YKywXZ_mb7l8DEjzeUgE-s,10706
+pytest_neon-3.0.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+pytest_neon-3.0.1.dist-info/entry_points.txt,sha256=5U88Idj_G8-PSDb9VF3OwYFbGLHnGOo_GxgYvi0dtXw,37
+pytest_neon-3.0.1.dist-info/licenses/LICENSE,sha256=aKKp_Ex4WBHTByY4BhXJ181dzB_qYhi2pCUmZ7Spn_0,1067
+pytest_neon-3.0.1.dist-info/RECORD,,

pytest_neon-3.0.0.dist-info/RECORD

@@ -1,8 +0,0 @@
-pytest_neon/__init__.py,sha256=W6OIAvx3sB2jfKCZaUUK1XcQPBxtWCnaYQCRn5NQhuA,404
-pytest_neon/plugin.py,sha256=ADH8on2MMI7jIKdk9mQDKbk59ypMwGNuh4_NsYNTphQ,40114
-pytest_neon/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pytest_neon-3.0.0.dist-info/METADATA,sha256=F68tFRyHHx6C5eKTXjq_bUMRtmR_Q0LcVrlKm8CwLHs,10706
-pytest_neon-3.0.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-pytest_neon-3.0.0.dist-info/entry_points.txt,sha256=5U88Idj_G8-PSDb9VF3OwYFbGLHnGOo_GxgYvi0dtXw,37
-pytest_neon-3.0.0.dist-info/licenses/LICENSE,sha256=aKKp_Ex4WBHTByY4BhXJ181dzB_qYhi2pCUmZ7Spn_0,1067
-pytest_neon-3.0.0.dist-info/RECORD,,