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

Author: durable-workflow-ops

.github/workflows/ci.yml

@@ -0,0 +1,66 @@
+name: CI
+
+on:
+  push:
+    branches: [main]
+  pull_request:
+    branches: [main]
+
+permissions:
+  contents: read
+
+jobs:
+  lint:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.12"
+      - run: pip install -e '.[dev]'
+      - run: ruff check src/ tests/
+      - run: mypy src/durable_workflow/
+
+  test:
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ["3.10", "3.11", "3.12"]
+    steps:
+      - uses: actions/checkout@v4
+      - uses: actions/setup-python@v5
+        with:
+          python-version: ${{ matrix.python-version }}
+      - run: pip install -e '.[dev]'
+      - run: pytest tests/ -m "not integration" -q
+
+  integration:
+    runs-on: ubuntu-latest
+    needs: [lint, test]
+    steps:
+      - uses: actions/checkout@v4
+        with:
+          path: sdk-python
+      - uses: actions/checkout@v4
+        with:
+          repository: durable-workflow/server
+          path: server
+      - uses: actions/setup-python@v5
+        with:
+          python-version: "3.12"
+      - run: pip install -e '.[dev]'
+        working-directory: sdk-python
+      - name: Start server stack
+        working-directory: sdk-python
+        run: |
+          docker compose -f docker-compose.test.yml up -d --wait --timeout 120
+      - name: Run integration tests
+        working-directory: sdk-python
+        env:
+          DURABLE_WORKFLOW_SERVER_URL: http://localhost:8080
+          DURABLE_WORKFLOW_AUTH_TOKEN: test-token
+        run: pytest tests/integration/ -v
+      - name: Teardown
+        if: always()
+        working-directory: sdk-python
+        run: docker compose -f docker-compose.test.yml down -v

src/durable_workflow/worker.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import asyncio
+import contextlib
 import logging
 import traceback
 import uuid
@@ -295,10 +296,8 @@ async def run(self) -> None:
         await self._register()
         wf_loop = asyncio.create_task(self._poll_workflow_tasks())
         act_loop = asyncio.create_task(self._poll_activity_tasks())
-        try:
+        with contextlib.suppress(asyncio.CancelledError):
             await asyncio.gather(wf_loop, act_loop)
-        except asyncio.CancelledError:
-            pass
 
     async def stop(self) -> None:
         self._stop.set()

tests/test_worker.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import asyncio
+import contextlib
 from unittest.mock import AsyncMock
 
 import pytest
@@ -384,10 +385,8 @@ async def test_run_starts_both_loops(self, mock_client: AsyncMock) -> None:
         await asyncio.sleep(0.05)
         await worker.stop()
         run_task.cancel()
-        try:
+        with contextlib.suppress(asyncio.CancelledError):
             await run_task
-        except asyncio.CancelledError:
-            pass
         assert mock_client.register_worker.call_count == 1
         assert mock_client.poll_workflow_task.call_count >= 1
         assert mock_client.poll_activity_task.call_count >= 1