WIP: STILL FIXING FUNDAMENTAL TRASH

This commit is contained in:
2026-03-05 14:04:17 -05:00
parent 70d18347d7
commit a13a6c5cd0
11 changed files with 113 additions and 162 deletions

View File

@@ -1,8 +1,8 @@
from typing import Any
from unittest.mock import patch
import ai_client
from src import ai_client
@patch('ai_client.GeminiCliAdapter')
@patch('src.ai_client.GeminiCliAdapter')
def test_send_invokes_adapter_send(mock_adapter_class: Any) -> None:
mock_instance = mock_adapter_class.return_value
mock_instance.send.return_value = {"text": "Hello from mock adapter", "tool_calls": []}
@@ -11,13 +11,13 @@ def test_send_invokes_adapter_send(mock_adapter_class: Any) -> None:
mock_instance.session_id = None
# Force reset to ensure our mock is used
with patch('ai_client._gemini_cli_adapter', mock_instance):
with patch('src.ai_client._gemini_cli_adapter', mock_instance):
ai_client.set_provider("gemini_cli", "gemini-2.0-flash")
res = ai_client.send("context", "msg")
assert res == "Hello from mock adapter"
mock_instance.send.assert_called()
@patch('ai_client.GeminiCliAdapter')
@patch('src.ai_client.GeminiCliAdapter')
def test_get_history_bleed_stats(mock_adapter_class: Any) -> None:
mock_instance = mock_adapter_class.return_value
mock_instance.send.return_value = {"text": "txt", "tool_calls": []}
@@ -25,7 +25,7 @@ def test_get_history_bleed_stats(mock_adapter_class: Any) -> None:
mock_instance.last_latency = 0.5
mock_instance.session_id = "sess"
with patch('ai_client._gemini_cli_adapter', mock_instance):
with patch('src.ai_client._gemini_cli_adapter', mock_instance):
ai_client.set_provider("gemini_cli", "gemini-2.0-flash")
# Initialize by sending a message
ai_client.send("context", "msg")