Assert Python workflow task parity fixtures

Assert Python workflow task parity fixtures

Author: durable-workflow-ops

tests/fixtures/control-plane/workflow-task-complete-parity.json

@@ -0,0 +1,66 @@
+{
+  "schema": "durable-workflow.polyglot.control-plane-request-fixture",
+  "version": 1,
+  "operation": "workflow-task.complete",
+  "request": {
+    "method": "POST",
+    "path": "/worker/workflow-tasks/workflow-task-231/complete",
+    "body": {
+      "lease_owner": "polyglot-worker-231",
+      "workflow_task_attempt": 2,
+      "commands": [
+        {
+          "type": "schedule_activity",
+          "activity_type": "inventory.sync-page",
+          "task_queue": "external-activities",
+          "input": {
+            "order_id": "order-231",
+            "page": 7
+          }
+        }
+      ]
+    }
+  },
+  "semantic_body": {
+    "task_id": "workflow-task-231",
+    "lease_owner": "polyglot-worker-231",
+    "workflow_task_attempt": 2,
+    "outcome": "completed"
+  },
+  "response_body": {
+    "task_id": "workflow-task-231",
+    "workflow_task_attempt": 2,
+    "outcome": "completed",
+    "run_status": "running"
+  },
+  "cli": {
+    "argv": {
+      "task-id": "workflow-task-231",
+      "attempt": "2",
+      "--lease-owner": "polyglot-worker-231",
+      "--command": [
+        "{\"type\":\"schedule_activity\",\"activity_type\":\"inventory.sync-page\",\"task_queue\":\"external-activities\",\"input\":{\"order_id\":\"order-231\",\"page\":7}}"
+      ],
+      "--json": true
+    }
+  },
+  "sdk_python": {
+    "method": "complete_workflow_task",
+    "kwargs": {
+      "task_id": "workflow-task-231",
+      "lease_owner": "polyglot-worker-231",
+      "workflow_task_attempt": 2,
+      "commands": [
+        {
+          "type": "schedule_activity",
+          "activity_type": "inventory.sync-page",
+          "task_queue": "external-activities",
+          "input": {
+            "order_id": "order-231",
+            "page": 7
+          }
+        }
+      ]
+    }
+  }
+}

tests/fixtures/control-plane/workflow-task-poll-parity.json

@@ -0,0 +1,53 @@
+{
+  "schema": "durable-workflow.polyglot.control-plane-request-fixture",
+  "version": 1,
+  "operation": "workflow-task.poll",
+  "request": {
+    "method": "POST",
+    "path": "/worker/workflow-tasks/poll",
+    "body": {
+      "worker_id": "polyglot-worker-231",
+      "task_queue": "external-workflows"
+    }
+  },
+  "semantic_body": {
+    "worker_id": "polyglot-worker-231",
+    "task_queue": "external-workflows",
+    "task_id": "workflow-task-231",
+    "workflow_id": "order-231",
+    "run_id": "run-231",
+    "workflow_task_attempt": 2
+  },
+  "response_body": {
+    "task": {
+      "task_id": "workflow-task-231",
+      "workflow_id": "order-231",
+      "run_id": "run-231",
+      "workflow_task_attempt": 2,
+      "lease_owner": "polyglot-worker-231",
+      "history_events": [
+        {
+          "event_id": 1,
+          "event_type": "WorkflowStarted",
+          "payload": {
+            "workflow_type": "orders.fulfillment"
+          }
+        }
+      ]
+    }
+  },
+  "cli": {
+    "argv": {
+      "worker-id": "polyglot-worker-231",
+      "--task-queue": "external-workflows",
+      "--json": true
+    }
+  },
+  "sdk_python": {
+    "method": "poll_workflow_task",
+    "kwargs": {
+      "worker_id": "polyglot-worker-231",
+      "task_queue": "external-workflows"
+    }
+  }
+}

tests/test_client.py

@@ -1433,6 +1433,34 @@ async def test_500_server_error(self, client: Client) -> None:
 
 
 class TestFailWorkflowTask:
+    @pytest.mark.asyncio
+    async def test_poll_workflow_task_matches_polyglot_fixture(self, client: Client) -> None:
+        fixture_path = Path(__file__).parent / "fixtures" / "control-plane" / "workflow-task-poll-parity.json"
+        fixture = json.loads(fixture_path.read_text())
+        resp = _mock_response(200, fixture["response_body"])
+
+        with patch.object(client._http, "request", new_callable=AsyncMock, return_value=resp) as mock:
+            task = await client.poll_workflow_task(**fixture["sdk_python"]["kwargs"])
+
+        assert task == fixture["response_body"]["task"]
+        assert task["task_id"] == fixture["semantic_body"]["task_id"]
+        assert mock.call_args.args[:2] == (fixture["request"]["method"], f"/api{fixture['request']['path']}")
+        assert mock.call_args.kwargs["json"] == fixture["request"]["body"]
+
+    @pytest.mark.asyncio
+    async def test_complete_workflow_task_matches_polyglot_fixture(self, client: Client) -> None:
+        fixture_path = Path(__file__).parent / "fixtures" / "control-plane" / "workflow-task-complete-parity.json"
+        fixture = json.loads(fixture_path.read_text())
+        resp = _mock_response(200, fixture["response_body"])
+
+        with patch.object(client._http, "request", new_callable=AsyncMock, return_value=resp) as mock:
+            result = await client.complete_workflow_task(**fixture["sdk_python"]["kwargs"])
+
+        assert result == fixture["response_body"]
+        assert result["outcome"] == fixture["semantic_body"]["outcome"]
+        assert mock.call_args.args[:2] == (fixture["request"]["method"], f"/api{fixture['request']['path']}")
+        assert mock.call_args.kwargs["json"] == fixture["request"]["body"]
+
     @pytest.mark.asyncio
     async def test_body_shape(self, client: Client) -> None:
         resp = _mock_response(200, {"task_id": "t1", "outcome": "failed"})