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

Update tests

Browse source
This commit is contained in:
Cyber MacGeddon 2026-02-23 22:01:20 +00:00
parent 6bccfd3d55
commit d8dea2e8a8
1 changed files with 19 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

28
tests/unit/test_cli/test_tool_commands.py
View file

@ -406,9 +406,9 @@ class TestShowToolsStructuredQuery:
@patch('trustgraph.cli.show_tools.Api')
def test_show_mixed_tool_types(self, mock_api_class, mock_api, capsys):
"""Test displaying multiple tool types including structured-query."""
"""Test displaying multiple tool types including structured-query and row-embeddings-query."""
mock_api_class.return_value, mock_config = mock_api
tools = [
{
"name": "ask_knowledge",
@ -417,37 +417,47 @@ class TestShowToolsStructuredQuery:
"collection": "docs"
},
{
"name": "query_data",
"name": "query_data",
"description": "Query structured data",
"type": "structured-query",
"collection": "sales"
},
{
"name": "find_customer",
"description": "Find customers by semantic search",
"type": "row-embeddings-query",
"schema-name": "customers",
"collection": "crm"
},
{
"name": "complete_text",
"description": "Generate text",
"type": "text-completion"
}
]
config_values = [
ConfigValue(type="tool", key=f"tool_{i}", value=json.dumps(tool))
for i, tool in enumerate(tools)
]
mock_config.get_values.return_value = config_values
show_config("http://test.com")
captured = capsys.readouterr()
output = captured.out
# All tool types should be displayed
assert "knowledge-query" in output
assert "structured-query" in output
assert "structured-query" in output
assert "row-embeddings-query" in output
assert "text-completion" in output
# Collections should be shown for appropriate tools
assert "docs" in output # knowledge-query collection
assert "sales" in output # structured-query collection
assert "crm" in output # row-embeddings-query collection
assert "customers" in output # row-embeddings-query schema-name
def test_show_main_parses_args_correctly(self):
"""Test that show main() parses arguments correctly."""