mirror of
https://github.com/TauricResearch/TradingAgents.git
synced 2026-09-27 06:56:39 +03:00
Yijia-Xiao
1f7a2aa865
- each such setting from the shell or .env is blanked before the package loads
89 lines
2.7 KiB
Python
89 lines
2.7 KiB
Python
"""Shared pytest fixtures that prevent CI hangs when API keys are absent."""
|
|
|
|
import os
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
|
|
def _blank_settings_overlay():
|
|
"""Blank every TRADINGAGENTS_* setting before the package is imported.
|
|
|
|
The package loads .env on import and folds these variables into
|
|
DEFAULT_CONFIG, so a contributor's own settings would become the defaults
|
|
the suite asserts on. A blank value is still present, so load_dotenv leaves
|
|
it alone, and the overlay reads it as unset. Tests of the overlay set their own.
|
|
"""
|
|
from dotenv import dotenv_values, find_dotenv
|
|
|
|
names = set(os.environ)
|
|
for filename in (".env", ".env.enterprise"):
|
|
names |= set(dotenv_values(find_dotenv(filename, usecwd=True)))
|
|
for name in names:
|
|
if name.startswith("TRADINGAGENTS_"):
|
|
os.environ[name] = ""
|
|
|
|
|
|
_blank_settings_overlay()
|
|
|
|
|
|
def pytest_configure(config):
|
|
for marker in ("unit", "integration", "smoke"):
|
|
config.addinivalue_line("markers", f"{marker}: {marker}-level tests")
|
|
|
|
|
|
_API_KEY_ENV_VARS = (
|
|
"OPENAI_API_KEY",
|
|
"GOOGLE_API_KEY",
|
|
"ANTHROPIC_API_KEY",
|
|
"XAI_API_KEY",
|
|
"DEEPSEEK_API_KEY",
|
|
"DASHSCOPE_API_KEY",
|
|
"DASHSCOPE_CN_API_KEY",
|
|
"ZHIPU_API_KEY",
|
|
"ZHIPU_CN_API_KEY",
|
|
"MINIMAX_API_KEY",
|
|
"MINIMAX_CN_API_KEY",
|
|
"OPENROUTER_API_KEY",
|
|
"AZURE_OPENAI_API_KEY",
|
|
"ALPHA_VANTAGE_API_KEY",
|
|
)
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def _dummy_api_keys(monkeypatch):
|
|
for env_var in _API_KEY_ENV_VARS:
|
|
# `or` not a .get default: an env var present but empty (e.g. a key left
|
|
# blank in a .env copied from .env.example) must still get the placeholder.
|
|
monkeypatch.setenv(env_var, os.environ.get(env_var) or "placeholder")
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def _isolate_config():
|
|
"""Reset the global dataflows config before and after each test.
|
|
|
|
``set_config`` merges (it never clears keys absent from the override), so a
|
|
test that sets e.g. ``tool_vendors`` would otherwise leak into later tests
|
|
and make routing behavior order-dependent. Replace the global outright so
|
|
every test starts from a clean DEFAULT_CONFIG.
|
|
"""
|
|
import copy
|
|
|
|
import tradingagents.dataflows.config as config_module
|
|
import tradingagents.default_config as default_config
|
|
|
|
config_module._config = copy.deepcopy(default_config.DEFAULT_CONFIG)
|
|
yield
|
|
config_module._config = copy.deepcopy(default_config.DEFAULT_CONFIG)
|
|
|
|
|
|
@pytest.fixture()
|
|
def mock_llm_client():
|
|
client = MagicMock()
|
|
client.get_llm.return_value = MagicMock()
|
|
with patch(
|
|
"tradingagents.llm_clients.factory.create_llm_client",
|
|
return_value=client,
|
|
):
|
|
yield client
|