Assert Python workflow query control-plane parity

Assert Python workflow query parity fixture

Author: durable-workflow-ops

tests/fixtures/control-plane/workflow-query-parity.json

@@ -0,0 +1,57 @@
+{
+  "schema": "durable-workflow.polyglot.control-plane-request-fixture",
+  "version": 1,
+  "operation": "workflow.query",
+  "request": {
+    "method": "POST",
+    "path": "/workflows/wf-polyglot-231/query/order.status"
+  },
+  "semantic_body": {
+    "workflow_id": "wf-polyglot-231",
+    "query_name": "order.status",
+    "input": [
+      {
+        "include_line_items": true,
+        "currency": "USD"
+      }
+    ]
+  },
+  "cli": {
+    "argv": {
+      "workflow-id": "wf-polyglot-231",
+      "query-name": "order.status",
+      "--input": "[{\"include_line_items\":true,\"currency\":\"USD\"}]"
+    },
+    "expected_body": {
+      "input": [
+        {
+          "include_line_items": true,
+          "currency": "USD"
+        }
+      ]
+    }
+  },
+  "sdk_python": {
+    "args": {
+      "workflow_id": "wf-polyglot-231",
+      "query_name": "order.status",
+      "args": [
+        {
+          "include_line_items": true,
+          "currency": "USD"
+        }
+      ]
+    },
+    "expected_body": {},
+    "payload_envelope": {
+      "field": "input",
+      "codec": "avro",
+      "decoded": [
+        {
+          "include_line_items": true,
+          "currency": "USD"
+        }
+      ]
+    }
+  }
+}

tests/test_client.py

@@ -365,6 +365,37 @@ async def test_query(self, client: Client) -> None:
             result = await client.query_workflow("wf-1", "status")
             assert result == {"result": "active"}
 
+    @pytest.mark.asyncio
+    async def test_query_request_matches_polyglot_fixture(self, client: Client) -> None:
+        fixture_path = Path(__file__).parent / "fixtures" / "control-plane" / "workflow-query-parity.json"
+        fixture = json.loads(fixture_path.read_text())
+        sdk = fixture["sdk_python"]
+        expected = sdk["expected_body"]
+        envelope_contract = sdk["payload_envelope"]
+
+        resp = _mock_response(200, {"result": {"status": "processing", "line_items": 3, "currency": "USD"}})
+
+        with patch.object(client._http, "request", new_callable=AsyncMock, return_value=resp) as mock:
+            result = await client.query_workflow(**sdk["args"])
+
+        assert result["result"]["status"] == "processing"
+
+        call_args = mock.call_args
+        assert call_args.args[0] == fixture["request"]["method"]
+        assert call_args.args[1] == f"/api{fixture['request']['path']}"
+        body = call_args.kwargs.get("json") or call_args[1].get("json")
+
+        for field, value in expected.items():
+            assert body[field] == value
+
+        envelope = body[envelope_contract["field"]]
+        assert envelope["codec"] == envelope_contract["codec"]
+        assert serializer.decode(envelope["blob"], codec=envelope["codec"]) == envelope_contract["decoded"]
+
+        semantic = fixture["semantic_body"]
+        assert sdk["args"]["workflow_id"] == semantic["workflow_id"]
+        assert sdk["args"]["query_name"] == semantic["query_name"]
+
     @pytest.mark.asyncio
     async def test_query_not_found(self, client: Client) -> None:
         resp = _mock_response(404, {"reason": "query_not_found", "message": "query [status] not declared"})