young-stock-cli/tests/test_cli.py at 69021b0e27c44b76788d1f3dfecde229bcf7be7f · AdvancingTitans/young-stock-cli · GitHub

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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
import json
import sys
from types import SimpleNamespace

import young_stock.cli as cli_module
from young_stock import __version__
from young_stock.cli import cli


def test_version_string():
    assert isinstance(__version__, str)
    assert __version__.count(".") >= 2


def test_cli_help(capsys):
    from click.testing import CliRunner
    runner = CliRunner()
    result = runner.invoke(cli, ["--help"])
    assert result.exit_code == 0
    assert "A-share" in result.output


def test_cli_version():
    from click.testing import CliRunner
    runner = CliRunner()
    result = runner.invoke(cli, ["--version"])
    assert result.exit_code == 0
    assert __version__ in result.output


def test_cli_subcommands_registered():
    from click.testing import CliRunner
    runner = CliRunner()
    result = runner.invoke(cli, ["--help"])
    for sub in ["a", "hk", "us", "global", "indices", "zt-pool", "flow", "cache-clear", "update"]:
        assert sub in result.output, f"subcommand `{sub}` missing from help"


def test_indices_json_outputs_parseable_payload(monkeypatch):
    from click.testing import CliRunner

    monkeypatch.setattr(cli_module._core, "nearest_trade_date", lambda: "20260529")
    monkeypatch.setattr(
        cli_module._core,
        "get_index",
        lambda date: [{"f12": "000001", "f14": "上证指数", "f2": 3000.12}],
    )

    result = CliRunner().invoke(cli, ["indices", "--json"])

    assert result.exit_code == 0
    payload = json.loads(result.output)
    assert payload["date"] == "20260529"
    assert payload["indices"][0]["f12"] == "000001"


def test_a_zt_json_outputs_pool_payload(monkeypatch):
    from click.testing import CliRunner

    monkeypatch.setattr(cli_module._core, "nearest_trade_date", lambda: "20260529")
    monkeypatch.setattr(cli_module._core, "get_zt_pool", lambda date: {"data": {"tc": 2}})
    monkeypatch.setattr(cli_module._core, "get_dt_pool", lambda date: {"data": {"tc": 1}})
    monkeypatch.setattr(cli_module._core, "get_zb_pool", lambda date: {"data": {"tc": 3}})

    result = CliRunner().invoke(cli, ["a", "--zt", "--json"])

    assert result.exit_code == 0
    payload = json.loads(result.output)
    assert payload["zt_pool"]["data"]["tc"] == 2
    assert payload["dt_pool"]["data"]["tc"] == 1
    assert payload["zb_pool"]["data"]["tc"] == 3


def test_flow_json_outputs_parseable_payload(monkeypatch):
    from click.testing import CliRunner

    monkeypatch.setattr(cli_module._core, "nearest_trade_date", lambda: "20260529")
    monkeypatch.setattr(
        cli_module._core,
        "get_fund_flow",
        lambda date: {"主力净流入": "-42899849216.0", "_source": "东财历史日线资金流"},
    )

    result = CliRunner().invoke(cli, ["flow", "--json"])

    assert result.exit_code == 0
    payload = json.loads(result.output)
    assert payload["flow"]["主力净流入"] == "-42899849216.0"


def test_global_json_serializes_quote_data(monkeypatch):
    from click.testing import CliRunner

    quote = cli_module._core.QuoteData(
        symbol="^HSI",
        name="恒生指数",
        market="hk_market",
        date="20260529",
        price=18000.0,
    )

    monkeypatch.setattr(cli_module._core, "nearest_trade_date", lambda: "20260529")
    monkeypatch.setattr(cli_module._core, "get_index", lambda date: [{"f12": "000001"}])
    monkeypatch.setattr(cli_module._core, "fetch_hk_indices_tencent", lambda symbols, date: [quote])
    monkeypatch.setattr(cli_module._core, "fetch_us_indices_sina", lambda symbols, date: [])

    result = CliRunner().invoke(cli, ["global", "--json"])

    assert result.exit_code == 0
    payload = json.loads(result.output)
    assert payload["a"][0]["f12"] == "000001"
    assert payload["hk"][0]["symbol"] == "^HSI"


def test_cli_update_runs_pip_upgrade(monkeypatch):
    from click.testing import CliRunner

    calls = []

    def fake_run(cmd, check):
        calls.append((cmd, check))
        return SimpleNamespace(returncode=0)

    monkeypatch.setattr(cli_module.subprocess, "run", fake_run)

    runner = CliRunner()
    result = runner.invoke(cli, ["update", "--pre", "--user"])

    assert result.exit_code == 0
    assert calls == [(
        [sys.executable, "-m", "pip", "install", "--upgrade", "young-stock-cli", "--pre", "--user"],
        False,
    )]