WIP: STILL FIXING FUNDAMENTAL TRASH
This commit is contained in:
@@ -1,5 +1,5 @@
|
||||
from unittest.mock import patch
|
||||
import ai_client
|
||||
from src import ai_client
|
||||
|
||||
def test_ai_client_send_gemini_cli() -> None:
|
||||
"""
|
||||
@@ -10,8 +10,8 @@ def test_ai_client_send_gemini_cli() -> None:
|
||||
test_response = "This is a dummy response from the Gemini CLI."
|
||||
# Set provider to gemini_cli
|
||||
ai_client.set_provider("gemini_cli", "gemini-2.5-flash-lite")
|
||||
# 1. Mock 'ai_client.GeminiCliAdapter' (which we will add)
|
||||
with patch('ai_client.GeminiCliAdapter') as MockAdapterClass:
|
||||
# 1. Mock 'src.ai_client.GeminiCliAdapter'
|
||||
with patch('src.ai_client.GeminiCliAdapter') as MockAdapterClass:
|
||||
mock_adapter_instance = MockAdapterClass.return_value
|
||||
mock_adapter_instance.send.return_value = {"text": test_response, "tool_calls": []}
|
||||
mock_adapter_instance.last_usage = {"total_tokens": 100}
|
||||
|
||||
Reference in New Issue
Block a user