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

Fix tests

Browse source
This commit is contained in:
Cyber MacGeddon 2025-12-04 21:03:52 +00:00
parent ff9d74ef79
commit 2a77c4eb6f
2 changed files with 15 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

18
tests/unit/test_cli/test_config_commands.py
View file

@ -108,7 +108,8 @@ class TestListConfigItems:
mock_list.assert_called_once_with( mock_list.assert_called_once_with(
url='http://custom.com', url='http://custom.com',
config_type='prompt', config_type='prompt',
format_type='json' format_type='json',
token=None
) )
def test_list_main_uses_defaults(self): def test_list_main_uses_defaults(self):
@ -126,7 +127,8 @@ class TestListConfigItems:
mock_list.assert_called_once_with( mock_list.assert_called_once_with(
url='http://localhost:8088/', url='http://localhost:8088/',
config_type='prompt', config_type='prompt',
format_type='text' format_type='text',
token=None
) )
@ -193,7 +195,8 @@ class TestGetConfigItem:
url='http://custom.com', url='http://custom.com',
config_type='prompt', config_type='prompt',
key='template-1', key='template-1',
format_type='json' format_type='json',
token=None
) )
@ -249,7 +252,8 @@ class TestPutConfigItem:
url='http://custom.com', url='http://custom.com',
config_type='prompt', config_type='prompt',
key='new-template', key='new-template',
value='Custom prompt: {input}' value='Custom prompt: {input}',
token=None
) )
def test_put_main_with_stdin_arg(self): def test_put_main_with_stdin_arg(self):
@ -273,7 +277,8 @@ class TestPutConfigItem:
url='http://localhost:8088/', url='http://localhost:8088/',
config_type='prompt', config_type='prompt',
key='stdin-template', key='stdin-template',
value=stdin_content value=stdin_content,
token=None
) )
def test_put_main_mutually_exclusive_args(self): def test_put_main_mutually_exclusive_args(self):
@ -328,7 +333,8 @@ class TestDeleteConfigItem:
mock_delete.assert_called_once_with( mock_delete.assert_called_once_with(
url='http://custom.com', url='http://custom.com',
config_type='prompt', config_type='prompt',
key='old-template' key='old-template',
token=None
) )

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

@ -135,7 +135,8 @@ class TestSetToolStructuredQuery:
arguments=[], arguments=[],
group=None, group=None,
state=None, state=None,
applicable_states=None applicable_states=None,
token=None
) )
def test_set_main_structured_query_no_arguments_needed(self): def test_set_main_structured_query_no_arguments_needed(self):
@ -313,7 +314,7 @@ class TestShowToolsStructuredQuery:
show_main() show_main()
mock_show.assert_called_once_with(url='http://custom.com') mock_show.assert_called_once_with(url='http://custom.com', token=None)
class TestStructuredQueryToolValidation: class TestStructuredQueryToolValidation: