prefactor-core 0.2.6__tar.gz → 0.2.7__tar.gz

{prefactor_core-0.2.6 → prefactor_core-0.2.7}/.gitignore

@@ -41,7 +41,6 @@ env/
 
 # uv
 .uv/
-uv.lock
 
 # Type checkers
 .mypy_cache/

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

@@ -1,11 +1,11 @@
 Metadata-Version: 2.4
 Name: prefactor-core
-Version: 0.2.6
+Version: 0.2.7
 Summary: Core Prefactor SDK with async queue-based operations
 Author-email: Prefactor Pty Ltd <josh@prefactor.tech>
 License: MIT
 Requires-Python: <4.0.0,>=3.11.0
-Requires-Dist: prefactor-http>=0.1.4
+Requires-Dist: prefactor-http>=0.1.5
 Requires-Dist: pydantic>=2.0.0
 Description-Content-Type: text/markdown
 

{prefactor_core-0.2.6 → prefactor_core-0.2.7}/pyproject.toml

@@ -9,7 +9,7 @@ authors = [
 ]
 requires-python = ">=3.11.0, <4.0.0"
 dependencies = [
-    "prefactor-http>=0.1.4",
+    "prefactor-http>=0.1.5",
     "pydantic>=2.0.0",
 ]
 

{prefactor_core-0.2.6 → prefactor_core-0.2.7}/src/prefactor_core/_version.py

@@ -3,5 +3,5 @@
 from __future__ import annotations
 
 PACKAGE_NAME = "prefactor-core"
-__version__ = "0.2.6"
+__version__ = "0.2.7"
 PACKAGE_VERSION = __version__

{prefactor_core-0.2.6 → prefactor_core-0.2.7}/src/prefactor_core/client.py

@@ -142,6 +142,14 @@ class PrefactorCoreClient:
         )
         await self._http.__aenter__()
 
+        try:
+            await self._http.validate_token()
+        except BaseException as exc:
+            http = self._http
+            self._http = None
+            await http.__aexit__(type(exc), exc, exc.__traceback__)
+            raise
+
         # Initialize executor
         self._executor = TaskExecutor(
             queue=self._queue,

{prefactor_core-0.2.6 → prefactor_core-0.2.7}/tests/test_agent_instance_finish_status.py

@@ -53,6 +53,9 @@ class _StubHttpClient:
     async def __aexit__(self, exc_type, exc, tb):
         return None
 
+    async def validate_token(self):
+        return {"status": "success"}
+
 
 def _make_client_config() -> PrefactorCoreConfig:
     return PrefactorCoreConfig(

{prefactor_core-0.2.6 → prefactor_core-0.2.7}/tests/test_agent_instance_register.py

@@ -56,6 +56,9 @@ class _StubHttpClient:
     async def __aexit__(self, exc_type, exc, tb):
         return None
 
+    async def validate_token(self):
+        return {"status": "success"}
+
 
 @pytest.mark.asyncio
 async def test_create_agent_instance_passes_environment_id_to_register():

{prefactor_core-0.2.6 → prefactor_core-0.2.7}/tests/test_failure_handling.py

@@ -68,17 +68,28 @@ class _StubHttpClient:
         *_args,
         agent_instances: _StubAgentInstances | None = None,
         agent_spans: _StubAgentSpans | None = None,
+        validate_token_error: PrefactorAuthError | None = None,
         **_kwargs,
     ) -> None:
         self.agent_instances = agent_instances or _StubAgentInstances()
         self.agent_spans = agent_spans or _StubAgentSpans()
+        self._validate_token_error = validate_token_error
+        self.enter_calls = 0
+        self.exit_calls = 0
 
     async def __aenter__(self):
+        self.enter_calls += 1
         return self
 
     async def __aexit__(self, exc_type, exc, tb):
+        self.exit_calls += 1
         return None
 
+    async def validate_token(self):
+        if self._validate_token_error is not None:
+            raise self._validate_token_error
+        return {"status": "success"}
+
 
 def _make_client_config(max_retries: int = 0) -> PrefactorCoreConfig:
     return PrefactorCoreConfig(
@@ -375,3 +386,23 @@ async def test_latched_failure_drops_already_queued_backlog():
             await client.close()
 
         assert exc_info.value.dropped_operations == 1
+
+
+@pytest.mark.asyncio
+async def test_initialize_raises_auth_error_when_token_validation_fails():
+    """Invalid tokens should fail during initialize() before workers start."""
+    stub_http = _StubHttpClient(
+        validate_token_error=PrefactorAuthError("Token expired", "bad_authtoken", 401)
+    )
+
+    with patch("prefactor_core.client.PrefactorHttpClient", return_value=stub_http):
+        client = PrefactorCoreClient(_make_client_config())
+        with pytest.raises(PrefactorAuthError) as exc_info:
+            await client.initialize()
+
+        assert exc_info.value.code == "bad_authtoken"
+        assert stub_http.enter_calls == 1
+        assert stub_http.exit_calls == 1
+        assert client._http is None
+        assert client._executor is None
+        await client.close()

{prefactor_core-0.2.6 → prefactor_core-0.2.7}/tests/test_sdk_header.py

@@ -37,6 +37,10 @@ class TestPrefactorCoreSdkHeader:
                 "prefactor_http.client.PrefactorHttpClient.__aexit__",
                 AsyncMock(return_value=None),
             ),
+            patch(
+                "prefactor_http.client.PrefactorHttpClient.validate_token",
+                AsyncMock(return_value={"status": "success"}),
+            ),
         ):
             await client.initialize()
             assert client._http is not None
@@ -59,6 +63,10 @@ class TestPrefactorCoreSdkHeader:
                 "prefactor_http.client.PrefactorHttpClient.__aexit__",
                 AsyncMock(return_value=None),
             ),
+            patch(
+                "prefactor_http.client.PrefactorHttpClient.validate_token",
+                AsyncMock(return_value={"status": "success"}),
+            ),
         ):
             await client.initialize()
             assert client._http is not None