"""A 股涨跌停纯函数测试（板块表 + 封板判断）。全用 raw 价格。"""
import pandas as pd

from sanguo_trader.limit import (
    get_board,
    limit_ratio,
    limit_up_price,
    limit_down_price,
    is_one_word_lock,
    is_t_lock,
    is_locked_for_buy_symbol,
    is_locked_for_sell_symbol,
)
from sanguo_trader.models import AccountConfig

CFG = AccountConfig(initial_capital=1_000_000)


def bar(open, high, low, close):
    return pd.Series({"open": open, "high": high, "low": low, "close": close})


# ---- 板块分类 ----
def test_board_classification():
    assert get_board("600000") == "main"     # 沪市主板
    assert get_board("000001") == "main"     # 深市主板
    assert get_board("300750") == "gem"      # 创业板
    assert get_board("301128") == "gem"      # 创业板（301）
    assert get_board("688981") == "star"     # 科创板
    assert get_board("689009") == "star"     # 科创板（689）
    assert get_board("830799") == "bse"      # 北交所
    assert get_board("920002") == "bse"      # 北交所（920）


# ---- 幅度查表 ----
def test_limit_ratio():
    assert limit_ratio("main", is_st=False) == 0.10
    assert limit_ratio("gem", is_st=False) == 0.20
    assert limit_ratio("star", is_st=False) == 0.20
    assert limit_ratio("bse", is_st=False) == 0.30
    # ST 统一 5%
    assert limit_ratio("main", is_st=True) == 0.05
    assert limit_ratio("gem", is_st=True) == 0.05
    assert limit_ratio("star", is_st=True) == 0.05
    assert limit_ratio("bse", is_st=True) == 0.30  # 北交所 ST 同 30%


# ---- 涨跌停价（按 pricetick 取整）----
def test_limit_up_price_rounds_to_pricetick():
    # 10.00 * 1.10 = 11.00
    assert limit_up_price(10.0, 0.10, 0.01) == 11.0
    # 9.99 * 1.20 = 11.988 → 11.99
    assert limit_up_price(9.99, 0.20, 0.01) == 11.99
    # 5.00 * 1.30 = 6.5
    assert limit_up_price(5.0, 0.30, 0.01) == 6.5


def test_limit_down_price_rounds_to_pricetick():
    # 10.00 * 0.90 = 9.00
    assert limit_down_price(10.0, 0.10, 0.01) == 9.0
    # 10.00 * 0.80 = 8.00
    assert limit_down_price(10.0, 0.20, 0.01) == 8.0


# ---- 一字板 / T 字板 ----
def test_one_word_lock_detected():
    up = limit_up_price(10.0, 0.10, 0.01)
    assert is_one_word_lock(bar(11.0, 11.0, 11.0, 11.0), up) is True
    # 非一字
    assert is_one_word_lock(bar(11.0, 11.5, 10.8, 11.0), up) is False
    assert is_one_word_lock(bar(11.0, 11.0, 10.5, 11.0), up) is False  # T 字板非一字


def test_t_lock_detected():
    up = limit_up_price(10.0, 0.10, 0.01)
    # T 字板：开=涨停 收=涨停 low<open
    assert is_t_lock(bar(11.0, 11.0, 10.5, 11.0), up) is True
    # 一字板不是 T 字板
    assert is_t_lock(bar(11.0, 11.0, 11.0, 11.0), up) is False
    # 开≠涨停 不是 T 字板
    assert is_t_lock(bar(10.5, 11.0, 10.5, 11.0), up) is False


# ---- 买单封板拒单（按 symbol 自动判板块）----
def test_buy_locked_one_word_main_board():
    # 主板 600000，一字涨停 → 买不进
    assert is_locked_for_buy_symbol(
        bar(11.0, 11.0, 11.0, 11.0), "600000", 10.0, CFG, is_st=False
    ) is True


def test_buy_locked_t_lock_conservative():
    # T 字板 → 保守拒买
    assert is_locked_for_buy_symbol(
        bar(11.0, 11.0, 10.5, 11.0), "600000", 10.0, CFG, is_st=False
    ) is True


def test_buy_not_locked_when_opened():
    # 开板（low 远低于涨停）→ 可买
    assert is_locked_for_buy_symbol(
        bar(10.5, 10.8, 10.2, 10.6), "600000", 10.0, CFG, is_st=False
    ) is False


def test_buy_locked_gem_board_20pct():
    # 创业板 300750，10.00 → 涨停 12.00
    assert is_locked_for_buy_symbol(
        bar(12.0, 12.0, 12.0, 12.0), "300750", 10.0, CFG, is_st=False
    ) is True


def test_buy_locked_star_board_20pct():
    # 科创板 688981
    assert is_locked_for_buy_symbol(
        bar(12.0, 12.0, 12.0, 12.0), "688981", 10.0, CFG, is_st=False
    ) is True


def test_buy_locked_bse_board_30pct():
    # 北交所 830799，5.00 → 涨停 6.5
    assert is_locked_for_buy_symbol(
        bar(6.5, 6.5, 6.5, 6.5), "830799", 5.0, CFG, is_st=False
    ) is True


def test_buy_locked_st_5pct():
    # ST 股 5% 涨停，10.00 → 10.50
    assert is_locked_for_buy_symbol(
        bar(10.5, 10.5, 10.5, 10.5), "600000", 10.0, CFG, is_st=True
    ) is True


# ---- 卖单封板（跌停对称）----
def test_sell_locked_one_word_limit_down():
    # 主板一字跌停：10 → 9.0
    assert is_locked_for_sell_symbol(
        bar(9.0, 9.0, 9.0, 9.0), "600000", 10.0, CFG, is_st=False
    ) is True


def test_sell_locked_t_lock_limit_down():
    # 跌停 T 字板：开=跌停 收=跌停 high>open（盘中反弹过）
    assert is_locked_for_sell_symbol(
        bar(9.0, 9.5, 9.0, 9.0), "600000", 10.0, CFG, is_st=False
    ) is True


def test_sell_not_locked_when_opened():
    # 非跌停形态 → 可卖
    assert is_locked_for_sell_symbol(
        bar(9.5, 9.8, 9.2, 9.6), "600000", 10.0, CFG, is_st=False
    ) is False