mirror of
https://github.com/Xe138/AI-Trader.git
synced 2026-04-02 01:27:24 -04:00
Bill
b09e1b0b11
Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
70 lines
2.0 KiB
Python
70 lines
2.0 KiB
Python
import os
|
|
import pytest
|
|
import asyncio
|
|
from unittest.mock import AsyncMock, MagicMock, patch
|
|
from agent.base_agent.base_agent import BaseAgent
|
|
|
|
|
|
def test_base_agent_uses_mock_in_dev_mode():
|
|
"""Test BaseAgent uses mock model when DEPLOYMENT_MODE=DEV"""
|
|
os.environ["DEPLOYMENT_MODE"] = "DEV"
|
|
|
|
agent = BaseAgent(
|
|
signature="test-agent",
|
|
basemodel="mock/test-trader",
|
|
log_path="./data/dev_agent_data"
|
|
)
|
|
|
|
# Mock MCP client to avoid needing running services
|
|
async def mock_initialize():
|
|
# Mock the MCP client
|
|
agent.client = MagicMock()
|
|
agent.tools = []
|
|
|
|
# Create mock model based on deployment mode
|
|
from tools.deployment_config import is_dev_mode
|
|
if is_dev_mode():
|
|
from agent.mock_provider import MockChatModel
|
|
agent.model = MockChatModel(date="2025-01-01")
|
|
|
|
# Run mock initialization
|
|
asyncio.run(mock_initialize())
|
|
|
|
assert agent.model is not None
|
|
assert "Mock" in str(type(agent.model))
|
|
|
|
os.environ["DEPLOYMENT_MODE"] = "PROD"
|
|
|
|
|
|
def test_base_agent_warns_about_api_keys_in_dev(capsys):
|
|
"""Test BaseAgent logs warning about API keys in DEV mode"""
|
|
os.environ["DEPLOYMENT_MODE"] = "DEV"
|
|
os.environ["OPENAI_API_KEY"] = "sk-test123"
|
|
|
|
# Test the warning function directly
|
|
from tools.deployment_config import log_api_key_warning
|
|
log_api_key_warning()
|
|
|
|
captured = capsys.readouterr()
|
|
assert "WARNING" in captured.out
|
|
assert "OPENAI_API_KEY" in captured.out
|
|
|
|
os.environ.pop("OPENAI_API_KEY")
|
|
os.environ["DEPLOYMENT_MODE"] = "PROD"
|
|
|
|
|
|
def test_base_agent_uses_dev_data_path():
|
|
"""Test BaseAgent uses dev data paths in DEV mode"""
|
|
os.environ["DEPLOYMENT_MODE"] = "DEV"
|
|
|
|
agent = BaseAgent(
|
|
signature="test-agent",
|
|
basemodel="mock/test-trader",
|
|
log_path="./data/agent_data" # Original path
|
|
)
|
|
|
|
# Should be converted to dev path
|
|
assert "dev_agent_data" in agent.base_log_path
|
|
|
|
os.environ["DEPLOYMENT_MODE"] = "PROD"
|