1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
"""Tests for strategy CLI commands."""
from unittest.mock import patch, MagicMock
from click.testing import CliRunner
from trading_cli.main import cli
def _make_mock_strategy(name, warmup_period=14):
s = MagicMock()
s.name = name
s.warmup_period = warmup_period
type(s).__name__ = "MockStrategy"
return s
def test_strategy_list():
mock_strategies = [_make_mock_strategy("rsi"), _make_mock_strategy("macd", 26)]
with patch("trading_cli.commands.strategy._load_all_strategies", return_value=mock_strategies):
runner = CliRunner()
result = runner.invoke(cli, ["strategy", "list"])
assert result.exit_code == 0
assert "rsi" in result.output.lower() or "Strategy" in result.output
def test_strategy_list_empty():
with patch("trading_cli.commands.strategy._load_all_strategies", return_value=[]):
runner = CliRunner()
result = runner.invoke(cli, ["strategy", "list"])
assert result.exit_code == 0
assert "No strategies found" in result.output
def test_strategy_info_rsi():
mock_strategies = [_make_mock_strategy("rsi")]
with patch("trading_cli.commands.strategy._load_all_strategies", return_value=mock_strategies):
runner = CliRunner()
result = runner.invoke(cli, ["strategy", "info", "--name", "rsi"])
assert result.exit_code == 0
assert "rsi" in result.output.lower()
def test_strategy_info_unknown():
mock_strategies = [_make_mock_strategy("rsi")]
with patch("trading_cli.commands.strategy._load_all_strategies", return_value=mock_strategies):
runner = CliRunner()
result = runner.invoke(cli, ["strategy", "info", "--name", "nonexistent"])
# Should handle gracefully (exit 0 or 1 with message)
assert "not found" in result.output.lower() or result.exit_code != 0 or "nonexistent" in result.output
|
