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

Fixing tests

Browse source
This commit is contained in:
Cyber MacGeddon 2025-09-05 08:35:06 +01:00
parent 49417eccab
commit 8bdd787c6a
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
tests/integration/test_structured_query_integration.py
View file

@ -594,10 +594,11 @@ class TestStructuredQueryServiceIntegration:
def create_flow_router(nlp_client, objects_client, response_producer): def create_flow_router(nlp_client, objects_client, response_producer):
def flow_router(service_name): def flow_router(service_name):
nonlocal service_call_count nonlocal service_call_count
service_call_count += 1
if service_name == "nlp-query-request": if service_name == "nlp-query-request":
service_call_count += 1
return nlp_client return nlp_client
elif service_name == "objects-query-request": elif service_name == "objects-query-request":
service_call_count += 1
return objects_client return objects_client
elif service_name == "response": elif service_name == "response":
return response_producer return response_producer
@ -619,7 +620,7 @@ class TestStructuredQueryServiceIntegration:
await asyncio.gather(*tasks) await asyncio.gather(*tasks)
# Assert - All requests should be processed # Assert - All requests should be processed
assert call_count == 6 # 2 calls per request (NLP + Objects) assert service_call_count == 6 # 2 calls per request (NLP + Objects)
for flow in flows: for flow in flows:
flow.return_value.send.assert_called_once() flow.return_value.send.assert_called_once()