pytest-neon 0.2.1__py3-none-any.whl → 0.3.0__py3-none-any.whl

pytest_neon/__init__.py

@@ -9,7 +9,7 @@ from pytest_neon.plugin import (
     neon_engine,
 )
 
-__version__ = "0.2.1"
+__version__ = "0.3.0"
 __all__ = [
     "NeonBranch",
     "neon_branch",

pytest_neon/plugin.py

@@ -7,14 +7,12 @@ import time
 from collections.abc import Generator
 from dataclasses import dataclass
 from datetime import datetime, timedelta, timezone
-from typing import TYPE_CHECKING, Any
+from typing import Any
 
 import pytest
 import requests
 from neon_api import NeonAPI
-
-if TYPE_CHECKING:
-    pass
+from neon_api.schema import EndpointState
 
 # Default branch expiry in seconds (10 minutes)
 DEFAULT_BRANCH_EXPIRY_SECONDS = 600
@@ -168,9 +166,10 @@ def _create_neon_branch(
         raise RuntimeError(f"No endpoint created for branch {branch.id}")
 
     # Wait for endpoint to be ready (it starts in "init" state)
-    # Endpoints typically become active in 1-2 seconds
+    # Endpoints typically become active in 1-2 seconds, but we allow up to 60s
+    # to handle occasional Neon API slowness or high load scenarios
     max_wait_seconds = 60
-    poll_interval = 0.5
+    poll_interval = 0.5  # Poll every 500ms for responsive feedback
     waited = 0.0
 
     while True:
@@ -180,7 +179,7 @@ def _create_neon_branch(
         endpoint = endpoint_response.endpoint
         state = endpoint.current_state
 
-        if state == "active" or str(state) == "EndpointState.active":
+        if state == EndpointState.active:
             break
 
         if waited >= max_wait_seconds:
@@ -254,7 +253,7 @@ def _reset_branch_to_parent(branch: NeonBranch, api_key: str) -> None:
         "Content-Type": "application/json",
     }
     response = requests.post(
-        url, headers=headers, json={"source_branch_id": branch.parent_id}
+        url, headers=headers, json={"source_branch_id": branch.parent_id}, timeout=30
     )
     response.raise_for_status()
 
@@ -304,6 +303,15 @@ def neon_branch(
     config = request.config
     api_key = _get_config_value(config, "neon_api_key", "NEON_API_KEY")
 
+    # Validate that branch has a parent for reset functionality
+    if not _neon_branch_for_reset.parent_id:
+        pytest.fail(
+            f"\n\nBranch {_neon_branch_for_reset.branch_id} has no parent. "
+            f"The neon_branch fixture requires a parent branch for reset.\n\n"
+            f"Use neon_branch_shared if you don't need reset, or specify "
+            f"a parent branch with --neon-parent-branch or NEON_PARENT_BRANCH_ID."
+        )
+
     yield _neon_branch_for_reset
 
     # Reset branch to parent state after each test
@@ -311,11 +319,11 @@ def neon_branch(
         try:
             _reset_branch_to_parent(branch=_neon_branch_for_reset, api_key=api_key)
         except Exception as e:
-            import warnings
-
-            warnings.warn(
-                f"Failed to reset branch {_neon_branch_for_reset.branch_id}: {e}",
-                stacklevel=2,
+            pytest.fail(
+                f"\n\nFailed to reset branch {_neon_branch_for_reset.branch_id} "
+                f"after test. Subsequent tests in this module may see dirty "
+                f"database state.\n\nError: {e}\n\n"
+                f"To keep the branch for debugging, use --neon-keep-branches"
             )
 
 

{pytest_neon-0.2.1.dist-info → pytest_neon-0.3.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytest-neon
-Version: 0.2.1
+Version: 0.3.0
 Summary: Pytest plugin for Neon database branch isolation in tests
 Project-URL: Homepage, https://github.com/zain/pytest-neon
 Project-URL: Repository, https://github.com/zain/pytest-neon
@@ -276,6 +276,28 @@ jobs:
 
 Branches use copy-on-write storage, so you only pay for data that differs from the parent branch.
 
+### What Reset Does
+
+The `neon_branch` fixture uses Neon's branch restore API to reset database state after each test:
+
+- **Data changes are reverted**: All INSERT, UPDATE, DELETE operations are undone
+- **Schema changes are reverted**: CREATE TABLE, ALTER TABLE, DROP TABLE, etc. are undone
+- **Sequences are reset**: Auto-increment counters return to parent state
+- **Complete rollback**: The branch is restored to the exact state of the parent at the time the child branch was created
+
+This is similar to database transactions but at the branch level.
+
+## Limitations
+
+### Parallel Test Execution
+
+This plugin sets the `DATABASE_URL` environment variable, which is process-global. This means it is **not compatible with pytest-xdist** or other parallel test runners that run tests in the same process.
+
+If you need parallel execution, you can:
+- Use `neon_branch.connection_string` directly instead of relying on `DATABASE_URL`
+- Run with `pytest-xdist --dist=loadfile` to keep modules in separate processes
+- Run tests serially (default pytest behavior)
+
 ## Troubleshooting
 
 ### "psycopg not installed" or "psycopg2 not installed"

pytest_neon-0.3.0.dist-info/RECORD

@@ -0,0 +1,8 @@
+pytest_neon/__init__.py,sha256=RftEKwHM37DDwxA_oVN00DMXTh_3tlpaz6LbbN_IOYA,398
+pytest_neon/plugin.py,sha256=A4eTJIV6XKQv4AIBizhg5qRDrvB_Hfx4ysTerjROHUg,17938
+pytest_neon/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pytest_neon-0.3.0.dist-info/METADATA,sha256=JmC3wuJSupLxVqeDlgzpUyYYpawRArzGj96N2CtcTE0,10555
+pytest_neon-0.3.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+pytest_neon-0.3.0.dist-info/entry_points.txt,sha256=5U88Idj_G8-PSDb9VF3OwYFbGLHnGOo_GxgYvi0dtXw,37
+pytest_neon-0.3.0.dist-info/licenses/LICENSE,sha256=aKKp_Ex4WBHTByY4BhXJ181dzB_qYhi2pCUmZ7Spn_0,1067
+pytest_neon-0.3.0.dist-info/RECORD,,

pytest_neon-0.2.1.dist-info/RECORD

@@ -1,8 +0,0 @@
-pytest_neon/__init__.py,sha256=tFnPqmxYhQPW1IlP53MTxeCumOOvS-OCuuMUn0hW7Zg,398
-pytest_neon/plugin.py,sha256=-7NHdQ-nqfR-ZQXFmLBu_vV445LTLJ44UXnkQPtCgOQ,17217
-pytest_neon/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pytest_neon-0.2.1.dist-info/METADATA,sha256=sNq5SF3HGzgCzh9Y1o9cGLSbPTcF8BV1kegvEdMeVt0,9496
-pytest_neon-0.2.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-pytest_neon-0.2.1.dist-info/entry_points.txt,sha256=5U88Idj_G8-PSDb9VF3OwYFbGLHnGOo_GxgYvi0dtXw,37
-pytest_neon-0.2.1.dist-info/licenses/LICENSE,sha256=aKKp_Ex4WBHTByY4BhXJ181dzB_qYhi2pCUmZ7Spn_0,1067
-pytest_neon-0.2.1.dist-info/RECORD,,