Ed_
d1ce0eaaeb
- Add cost tracking with new cost_tracker.py module - Enhance Track Proposal modal with editable titles and goals - Add Conductor Setup summary and New Track creation form to MMA Dashboard - Implement Task DAG editing (add/delete tickets) and track-scoped discussion - Add visual polish: color-coded statuses, tinted progress bars, and node indicators - Support live worker streaming from AI providers to GUI panels - Fix numerous integration test regressions and stabilize headless service
36 lines
1.5 KiB
Python
36 lines
1.5 KiB
Python
from typing import Any
|
|
import pytest
|
|
from unittest.mock import patch, MagicMock
|
|
import ai_client
|
|
|
|
@patch('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": []}
|
|
mock_instance.last_usage = {"total_tokens": 100}
|
|
mock_instance.last_latency = 0.5
|
|
mock_instance.session_id = None
|
|
|
|
# Force reset to ensure our mock is used
|
|
with patch('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')
|
|
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": []}
|
|
mock_instance.last_usage = {"input_tokens": 1500}
|
|
mock_instance.last_latency = 0.5
|
|
mock_instance.session_id = "sess"
|
|
|
|
with patch('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")
|
|
stats = ai_client.get_history_bleed_stats()
|
|
assert stats["provider"] == "gemini_cli"
|
|
assert stats["current"] == 1500
|