apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-25 05:01:01 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fixed tests

Browse source
This commit is contained in:
Cyber MacGeddon 2025-09-16 23:49:35 +01:00
parent bcee8eb125
commit d2e3e317b7
2 changed files with 7 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/unit/test_retrieval/test_structured_diag/test_schema_selection.py
View file

@ -123,6 +123,7 @@ async def test_schema_selection_success(service, mock_flow):
mock_response = MagicMock() mock_response = MagicMock()
mock_response.error = None mock_response.error = None
mock_response.text = '["products", "orders"]' mock_response.text = '["products", "orders"]'
mock_response.object = None # Explicitly set to None
prompt_request_flow.return_value = mock_response prompt_request_flow.return_value = mock_response
# Create request # Create request
@ -162,6 +163,7 @@ async def test_schema_selection_empty_response(service, mock_flow):
mock_response = MagicMock() mock_response = MagicMock()
mock_response.error = None mock_response.error = None
mock_response.text = "" mock_response.text = ""
mock_response.object = "" # Both fields empty
prompt_request_flow.return_value = mock_response prompt_request_flow.return_value = mock_response
# Create request # Create request
@ -219,6 +221,7 @@ async def test_schema_selection_invalid_json(service, mock_flow):
mock_response = MagicMock() mock_response = MagicMock()
mock_response.error = None mock_response.error = None
mock_response.text = "not valid json" mock_response.text = "not valid json"
mock_response.object = None
prompt_request_flow.return_value = mock_response prompt_request_flow.return_value = mock_response
# Create request # Create request
@ -246,6 +249,7 @@ async def test_schema_selection_non_array_response(service, mock_flow):
mock_response = MagicMock() mock_response = MagicMock()
mock_response.error = None mock_response.error = None
mock_response.text = '{"schema": "products"}' # Object instead of array mock_response.text = '{"schema": "products"}' # Object instead of array
mock_response.object = None
prompt_request_flow.return_value = mock_response prompt_request_flow.return_value = mock_response
# Create request # Create request
@ -273,6 +277,7 @@ async def test_schema_selection_with_options(service, mock_flow):
mock_response = MagicMock() mock_response = MagicMock()
mock_response.error = None mock_response.error = None
mock_response.text = '["products"]' mock_response.text = '["products"]'
mock_response.object = None
prompt_request_flow.return_value = mock_response prompt_request_flow.return_value = mock_response
# Create request with options # Create request with options
@ -333,6 +338,7 @@ async def test_schema_selection_empty_schemas(service, mock_flow):
mock_response = MagicMock() mock_response = MagicMock()
mock_response.error = None mock_response.error = None
mock_response.text = '[]' mock_response.text = '[]'
mock_response.object = None
prompt_request_flow.return_value = mock_response prompt_request_flow.return_value = mock_response
# Create request # Create request

2
trustgraph-base/trustgraph/messaging/translators/diagnosis.py
View file

@ -57,7 +57,7 @@ class StructuredDataDiagnosisResponseTranslator(MessageTranslator):
result["descriptor"] = obj.descriptor result["descriptor"] = obj.descriptor
if obj.metadata: if obj.metadata:
result["metadata"] = obj.metadata result["metadata"] = obj.metadata
if obj.schema_matches: if obj.schema_matches is not None:
result["schema-matches"] = obj.schema_matches result["schema-matches"] = obj.schema_matches
return result return result