pytest-neon 2.1.1__tar.gz → 2.1.3__tar.gz

{pytest_neon-2.1.1 → pytest_neon-2.1.3}/CLAUDE.md

@@ -81,15 +81,9 @@ Tests in `tests/` use `pytester` for testing pytest plugins. The plugin itself c
 
 ## Publishing
 
-Use the GitHub Actions release workflow:
+**Always use the GitHub Actions release workflow** - do not manually bump versions:
 1. Go to Actions → Release → Run workflow
 2. Choose patch/minor/major
 3. Workflow bumps version, commits, tags, and publishes to PyPI
 
-Or manually:
-```bash
-uv build
-uv publish --token $PYPI_TOKEN
-```
-
 Package name on PyPI: `pytest-neon`

{pytest_neon-2.1.1 → pytest_neon-2.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pytest-neon
-Version: 2.1.1
+Version: 2.1.3
 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-2.1.1 → pytest_neon-2.1.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "pytest-neon"
-version = "2.1.1"
+version = "2.1.3"
 description = "Pytest plugin for Neon database branch isolation in tests"
 readme = "README.md"
 license = "MIT"

{pytest_neon-2.1.1 → pytest_neon-2.1.3}/src/pytest_neon/__init__.py

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

{pytest_neon-2.1.1 → pytest_neon-2.1.3}/src/pytest_neon/plugin.py

@@ -480,9 +480,17 @@ def _wait_for_operations(
         return  # All operations already complete
 
     waited = 0.0
+    first_poll = True
     while pending_op_ids and waited < max_wait_seconds:
-        time.sleep(poll_interval)
-        waited += poll_interval
+        # Poll immediately first time (operation usually completes instantly),
+        # then wait between subsequent polls
+        if first_poll:
+            time.sleep(0.1)  # Tiny delay to let operation start
+            waited += 0.1
+            first_poll = False
+        else:
+            time.sleep(poll_interval)
+            waited += poll_interval
 
         # Check status of each pending operation
         still_pending = []
@@ -494,10 +502,10 @@ def _wait_for_operations(
                 op_data = response.json().get("operation", {})
                 status = op_data.get("status")
 
-                if status == "error":
+                if status == "failed":
                     err = op_data.get("error", "unknown error")
                     raise RuntimeError(f"Operation {op_id} failed: {err}")
-                if status not in ("finished", "skipped"):
+                if status not in ("finished", "skipped", "cancelled"):
                     still_pending.append(op_id)
             except requests.RequestException:
                 # On network error, assume still pending and retry

{pytest_neon-2.1.1 → pytest_neon-2.1.3}/tests/test_reset_behavior.py

@@ -162,6 +162,43 @@ class TestResetRetryBehavior:
         # Should have polled until finished
         assert get_call_count[0] == 3
 
+    def test_reset_raises_on_failed_operation(self, mocker):
+        """Verify reset raises RuntimeError when operation fails."""
+        branch = NeonBranch(
+            branch_id="br-test",
+            project_id="proj-test",
+            connection_string="postgresql://test",
+            host="test.neon.tech",
+            parent_id="br-parent",
+        )
+
+        # Mock POST response with pending operation
+        mock_post_response = mocker.Mock()
+        mock_post_response.raise_for_status = mocker.Mock()
+        mock_post_response.json.return_value = {
+            "operations": [{"id": "op-123", "status": "running"}]
+        }
+
+        # Mock GET response: operation failed
+        mock_get_response = mocker.Mock()
+        mock_get_response.raise_for_status = mocker.Mock()
+        mock_get_response.json.return_value = {
+            "operation": {
+                "id": "op-123",
+                "status": "failed",
+                "error": "Something went wrong",
+            }
+        }
+
+        mocker.patch(
+            "pytest_neon.plugin.requests.post", return_value=mock_post_response
+        )
+        mocker.patch("pytest_neon.plugin.requests.get", return_value=mock_get_response)
+        mocker.patch("pytest_neon.plugin.time.sleep")
+
+        with pytest.raises(RuntimeError, match="Operation op-123 failed"):
+            _reset_branch_to_parent(branch, "fake-api-key")
+
 
 class TestResetBehavior:
     """Test that branch reset happens between tests."""

{pytest_neon-2.1.1 → pytest_neon-2.1.3}/uv.lock

@@ -1200,7 +1200,7 @@ wheels = [
 
 [[package]]
 name = "pytest-neon"
-version = "2.0.0"
+version = "2.1.1"
 source = { editable = "." }
 dependencies = [
     { name = "neon-api" },