49 lines
1.6 KiB
Python
49 lines
1.6 KiB
Python
import unittest
|
|
from unittest.mock import patch, MagicMock
|
|
import io
|
|
import json
|
|
import sys
|
|
import os
|
|
|
|
# Add project root to sys.path
|
|
sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
|
|
|
|
# Import after path fix
|
|
from scripts.cli_tool_bridge import main
|
|
|
|
class TestCliToolBridgeMapping(unittest.TestCase):
|
|
def setUp(self) -> None:
|
|
os.environ['GEMINI_CLI_HOOK_CONTEXT'] = 'manual_slop'
|
|
|
|
@patch('sys.stdin', new_callable=io.StringIO)
|
|
@patch('sys.stdout', new_callable=io.StringIO)
|
|
@patch('api_hook_client.ApiHookClient.request_confirmation')
|
|
def test_mapping_from_api_format(self, mock_request: MagicMock, mock_stdout: MagicMock, mock_stdin: MagicMock) -> None:
|
|
"""
|
|
Verify that bridge correctly maps 'id', 'name', 'input' (Gemini API format)
|
|
into tool_name and tool_input for the hook client.
|
|
"""
|
|
api_tool_call = {
|
|
'id': 'call123',
|
|
'name': 'read_file',
|
|
'input': {'path': 'test.txt'}
|
|
}
|
|
# 1. Mock stdin with the API format JSON
|
|
mock_stdin.write(json.dumps(api_tool_call))
|
|
mock_stdin.seek(0)
|
|
# 2. Mock ApiHookClient to return approved
|
|
mock_request.return_value = {'approved': True}
|
|
# Run main
|
|
main()
|
|
# 3. Verify that request_confirmation was called with mapped values
|
|
# If it's not mapped, it will likely be called with None or fail
|
|
mock_request.assert_called_once_with('read_file', {'path': 'test.txt'})
|
|
# 4. Capture stdout and assert allow
|
|
output_str = mock_stdout.getvalue().strip()
|
|
self.assertTrue(output_str, "Stdout should not be empty")
|
|
output = json.loads(output_str)
|
|
self.assertEqual(output.get('decision'), 'allow')
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|