From e50c6f8ddaf1ae9c9dcbcde76466533956ee8fe7 Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Thu, 4 Sep 2025 15:44:34 +0100 Subject: [PATCH] Update tests --- .../contract/test_structured_data_contracts.py | 18 +++++++++++++++++- 1 file changed, 17 insertions(+), 1 deletion(-) diff --git a/tests/contract/test_structured_data_contracts.py b/tests/contract/test_structured_data_contracts.py index d3034ded..af8e70df 100644 --- a/tests/contract/test_structured_data_contracts.py +++ b/tests/contract/test_structured_data_contracts.py @@ -298,4 +298,20 @@ class TestStructuredDataSerializationContracts: "detected_schemas": ["test"], "confidence": 0.9 } - assert serialize_deserialize_test(QuestionToStructuredQueryResponse, response_data) \ No newline at end of file + assert serialize_deserialize_test(QuestionToStructuredQueryResponse, response_data) + + def test_structured_query_serialization(self): + """Test structured query request/response serialization contract""" + # Test request + request_data = { + "question": "Show me all customers" + } + assert serialize_deserialize_test(StructuredQueryRequest, request_data) + + # Test response + response_data = { + "error": None, + "data": '{"customers": [{"id": "1", "name": "John"}]}', + "errors": [] + } + assert serialize_deserialize_test(StructuredQueryResponse, response_data) \ No newline at end of file