ed
cfeb3cb3e0
Changes: - Rename ai_client.send(...) to ai_client.send_result(...) (2 sites) - Add assert result.ok (1 site; the second test only checks result is not None) - Add "from src.result_types import Result" import 2 tests pass.
33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
from unittest.mock import MagicMock
|
|
from src import ai_client
|
|
from src.result_types import Result
|
|
|
|
|
|
def test_gemini_cli_full_integration() -> None:
|
|
ai_client.reset_session()
|
|
ai_client.set_provider("gemini_cli", "gemini-2.0-flash")
|
|
mock_adapter = MagicMock()
|
|
mock_adapter.send.return_value = {
|
|
"text": "Final integrated answer",
|
|
"tool_calls": [],
|
|
}
|
|
mock_adapter.last_usage = {"total_tokens": 10}
|
|
ai_client._gemini_cli_adapter = mock_adapter
|
|
result = ai_client.send_result("context", "integrated test")
|
|
assert result.ok
|
|
assert "Final integrated answer" in result.data
|
|
|
|
|
|
def test_gemini_cli_rejection_and_history() -> None:
|
|
ai_client.reset_session()
|
|
ai_client.set_provider("gemini_cli", "gemini-2.0-flash")
|
|
mock_adapter = MagicMock()
|
|
mock_adapter.send.return_value = {
|
|
"text": "",
|
|
"tool_calls": [{"name": "run_powershell", "args": {"script": "dir"}}],
|
|
}
|
|
mock_adapter.last_usage = {}
|
|
ai_client._gemini_cli_adapter = mock_adapter
|
|
result = ai_client.send_result("ctx", "msg", pre_tool_callback=lambda *a, **kw: None)
|
|
assert result is not None
|