mirror of
https://github.com/Xe138/AI-Trader.git
synced 2026-04-01 17:17:24 -04:00
Bill
14cf88f642
Major improvements: - Fixed all 42 broken tests (database connection leaks) - Added db_connection() context manager for proper cleanup - Created comprehensive test suites for undertested modules New test coverage: - tools/general_tools.py: 26 tests (97% coverage) - tools/price_tools.py: 11 tests (validates NASDAQ symbols, date handling) - api/price_data_manager.py: 12 tests (85% coverage) - api/routes/results_v2.py: 3 tests (98% coverage) - agent/reasoning_summarizer.py: 2 tests (87% coverage) - api/routes/period_metrics.py: 2 edge case tests (100% coverage) - agent/mock_provider: 1 test (100% coverage) Database fixes: - Added db_connection() context manager to prevent leaks - Updated 16+ test files to use context managers - Fixed drop_all_tables() to match new schema - Added CHECK constraint for action_type - Added ON DELETE CASCADE to trading_days foreign key Test improvements: - Updated SQL INSERT statements with all required fields - Fixed date parameter handling in API integration tests - Added edge case tests for validation functions - Fixed import errors across test suite Results: - Total coverage: 84.81% (was 61%) - Tests passing: 406 (was 364 with 42 failures) - Total lines covered: 6364 of 7504 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
87 lines
2.7 KiB
Python
87 lines
2.7 KiB
Python
import pytest
|
|
import asyncio
|
|
from agent.mock_provider.mock_ai_provider import MockAIProvider
|
|
from agent.mock_provider.mock_langchain_model import MockChatModel
|
|
|
|
|
|
def test_mock_provider_rotates_stocks():
|
|
"""Test that mock provider returns different stocks on different days"""
|
|
provider = MockAIProvider()
|
|
|
|
# Day 1 should recommend AAPL
|
|
response1 = provider.generate_response("2025-01-01", step=0)
|
|
assert "AAPL" in response1
|
|
assert "<FINISH_SIGNAL>" in response1
|
|
|
|
# Day 2 should recommend MSFT
|
|
response2 = provider.generate_response("2025-01-02", step=0)
|
|
assert "MSFT" in response2
|
|
assert "<FINISH_SIGNAL>" in response2
|
|
|
|
# Responses should be different
|
|
assert response1 != response2
|
|
|
|
|
|
def test_mock_provider_finish_signal():
|
|
"""Test that all responses include finish signal"""
|
|
provider = MockAIProvider()
|
|
response = provider.generate_response("2025-01-01", step=0)
|
|
assert "<FINISH_SIGNAL>" in response
|
|
|
|
|
|
def test_mock_provider_valid_json_tool_calls():
|
|
"""Test that responses contain valid tool call syntax"""
|
|
provider = MockAIProvider()
|
|
response = provider.generate_response("2025-01-01", step=0)
|
|
assert "[calls tool_get_price" in response or "get_price" in response.lower()
|
|
|
|
|
|
def test_mock_chat_model_invoke():
|
|
"""Test synchronous invoke returns proper message format"""
|
|
model = MockChatModel(date="2025-01-01")
|
|
|
|
messages = [{"role": "user", "content": "Analyze the market"}]
|
|
response = model.invoke(messages)
|
|
|
|
assert hasattr(response, "content")
|
|
assert "AAPL" in response.content
|
|
assert "<FINISH_SIGNAL>" in response.content
|
|
|
|
|
|
def test_mock_chat_model_ainvoke():
|
|
"""Test asynchronous invoke returns proper message format"""
|
|
async def run_test():
|
|
model = MockChatModel(date="2025-01-02")
|
|
messages = [{"role": "user", "content": "Analyze the market"}]
|
|
response = await model.ainvoke(messages)
|
|
|
|
assert hasattr(response, "content")
|
|
assert "MSFT" in response.content
|
|
assert "<FINISH_SIGNAL>" in response.content
|
|
|
|
asyncio.run(run_test())
|
|
|
|
|
|
def test_mock_chat_model_different_dates():
|
|
"""Test that different dates produce different responses"""
|
|
model1 = MockChatModel(date="2025-01-01")
|
|
model2 = MockChatModel(date="2025-01-02")
|
|
|
|
msg = [{"role": "user", "content": "Trade"}]
|
|
response1 = model1.invoke(msg)
|
|
response2 = model2.invoke(msg)
|
|
|
|
assert response1.content != response2.content
|
|
|
|
|
|
def test_mock_provider_string_representation():
|
|
"""Test __str__ and __repr__ methods"""
|
|
provider = MockAIProvider()
|
|
|
|
str_repr = str(provider)
|
|
repr_repr = repr(provider)
|
|
|
|
assert "MockAIProvider" in str_repr
|
|
assert "development" in str_repr
|
|
assert str_repr == repr_repr
|