sdk-python: update v2 client, worker, and tests

Author: durable-workflow-ops

src/durable_workflow/client.py

@@ -107,7 +107,7 @@ async def __aexit__(self, *exc: Any) -> None:
         await self.aclose()
 
     def _headers(self, *, worker: bool = False) -> dict[str, str]:
-        h: dict[str, str] = {"Content-Type": "application/json"}
+        h: dict[str, str] = {"Content-Type": "application/json", "Accept": "application/json"}
         if self.token:
             h["Authorization"] = f"Bearer {self.token}"
         h["X-Namespace"] = self.namespace

src/durable_workflow/worker.py

@@ -33,7 +33,7 @@ def __init__(
         workflows: Iterable[type] = (),
         activities: Iterable[Callable[..., Any]] = (),
         worker_id: str | None = None,
-        poll_timeout: float = 5.0,
+        poll_timeout: float = 35.0,
         max_concurrent_workflow_tasks: int = 10,
         max_concurrent_activity_tasks: int = 10,
         shutdown_timeout: float = 30.0,
@@ -248,6 +248,7 @@ async def _poll_workflow_tasks(self) -> None:
                 continue
             if task is None:
                 self._wf_semaphore.release()
+                await asyncio.sleep(0)
                 continue
             self._track(self._dispatch_workflow_task(task))
 
@@ -278,6 +279,7 @@ async def _poll_activity_tasks(self) -> None:
                 continue
             if task is None:
                 self._act_semaphore.release()
+                await asyncio.sleep(0)
                 continue
             self._track(self._dispatch_activity_task(task))
 

tests/integration/conftest.py

@@ -21,4 +21,4 @@ def server_url() -> str:
 
 @pytest.fixture(scope="session")
 def server_token() -> str:
-    return os.environ.get("DURABLE_WORKFLOW_AUTH_TOKEN", "dev-token-123")
+    return os.environ.get("DURABLE_WORKFLOW_AUTH_TOKEN", "test-token")

tests/test_worker.py

@@ -348,6 +348,10 @@ async def limited_activity() -> str:
             max_concurrent_activity_tasks=2,
         )
 
+        async def _acquire_and_dispatch(t: dict[str, object]) -> None:
+            await worker._act_semaphore.acquire()
+            await worker._dispatch_activity_task(t)
+
         tasks = []
         for i in range(4):
             task = {
@@ -357,7 +361,7 @@ async def limited_activity() -> str:
                 "arguments": "[]",
                 "payload_codec": "json",
             }
-            tasks.append(worker._track(worker._dispatch_activity_task(task)))
+            tasks.append(worker._track(_acquire_and_dispatch(task)))
 
         await asyncio.sleep(0.01)
         assert max_running == 2