"""针对本轮 A 股增强功能的单元测试。"""

import pytest
import struct
from unittest.mock import patch, MagicMock, AsyncMock
from xmtdx import TdxClient, Market
from xmtdx.models.security import SecurityInfo
from xmtdx.models.timeseries import TransactionRecord
from xmtdx.models.quote import SecurityQuote
from xmtdx.models.stats import FundFlow, HistoricalFundFlow, MarketStat

@patch("xmtdx.client.TdxConnection")
def test_get_fund_flow_logic(mock_conn_cls):
    """测试资金流分类计算逻辑。"""
    mock_conn = mock_conn_cls.return_value
    client = TdxClient("127.0.0.1")
    
    # 构造模拟 Tick 数据
    mock_recs = [
        TransactionRecord(10, 0, 100.0, 100, 0, 0), # super_in (100*100*100 = 100w)
        TransactionRecord(10, 1, 10.0, 250, 1, 0),  # large_out (10*250*100 = 25w)
        TransactionRecord(10, 2, 10.0, 10, 0, 0),   # small_in (10*10*100 = 1w)
    ]
    
    with patch.object(TdxClient, "get_transaction_data", return_value=mock_recs):
        flow = client.get_fund_flow(Market.SH, "600000")
        
        assert flow.super_in == 1000000.0
        assert flow.large_out == 250000.0
        assert flow.small_in == 10000.0
        assert flow.main_net_inflow == 1000000.0 - 250000.0

@patch("xmtdx.client.TdxConnection")
def test_get_security_list_all_filtering(mock_conn_cls):
    """测试三市 A 股过滤与行业挂载逻辑。"""
    client = TdxClient("127.0.0.1")
    
    # 模拟行业配置 tdxhy.cfg
    industry_cfg = b"1|600000|T01|||X01\n0|000001|T02|||X02\n2|830000|T03|||X03"
    
    # 模拟各市场返回
    def mock_get_list(market, start):
        if market == Market.SH:
            return [
                SecurityInfo(Market.SH, "600000", "SH_A", 100, 2, 10.0),
                SecurityInfo(Market.SH, "999999", "INDEX", 100, 2, 3000.0), # 应被过滤
            ]
        if market == Market.SZ:
            return [SecurityInfo(Market.SZ, "000001", "SZ_A", 100, 2, 10.0)]
        if market == Market.BJ:
            return [SecurityInfo(Market.BJ, "830000", "BJ_A", 100, 2, 10.0)]
        return []

    with patch.object(TdxClient, "get_report_file", return_value=industry_cfg), \
         patch.object(TdxClient, "get_security_count", return_value=1), \
         patch.object(TdxClient, "get_security_list", side_effect=mock_get_list):
        
        all_stocks = client.get_security_list_all()
        
        assert len(all_stocks) == 3
        codes = [s.code for s in all_stocks]
        assert "600000" in codes
        assert "000001" in codes
        assert "830000" in codes
        
        s0 = next(s for s in all_stocks if s.code == "600000")
        assert s0.industry_tdx == "T01"

@patch("xmtdx.client.TdxConnection")
def test_get_market_stat_mapping(mock_conn_cls):
    """测试市场统计字段映射。"""
    client = TdxClient("127.0.0.1")
    
    mock_quote = SecurityQuote(
        Market.SH, "880005",
        price=3000.0,      # up
        pre_close=2000.0,  # down
        open=500.0,        # neutral
        high=5500.0,       # total
        low=100.0, vol=1000000.0, cur_vol=0, amount=50000000.0,
        s_vol=0, b_vol=0, active1=0, active2=0,
        bid1=0, bid_vol1=0, bid2=0, bid_vol2=0, bid3=0, bid_vol3=0, bid4=0, bid_vol4=0, bid5=0, bid_vol5=0,
        ask1=0, ask_vol1=0, ask2=0, ask_vol2=0, ask3=0, ask_vol3=0, ask4=0, ask_vol4=0, ask5=0, ask_vol5=0,
        rise_speed=0, limit_up=0, limit_down=0
    )
    
    with patch.object(TdxClient, "get_security_quotes", return_value=[mock_quote]):
        stat = client.get_market_stat()
        assert stat.up_count == 3000
        assert stat.total_count == 5500

def test_get_history_fund_flow_parsing():
    """测试历史资金流序列解析逻辑。"""
    from xmtdx.commands.fund_flow import GetHistoryFundFlowCmd
    
    # 模拟 Category 22 响应 (Header 9 + Count 2 + Body 36)
    body = bytearray(9)
    body.extend(struct.pack("<H", 1)) # 1 record
    
    # Record: Date(I) + 8 * custom_float(i)
    # 2025-01-08
    date = 20250108
    # 模拟 8 个流向金额
    record = struct.pack("<Iiiiiiiii", date, 100, 200, 300, 400, 500, 600, 700, 800)
    body.extend(record)
    
    cmd = GetHistoryFundFlowCmd(Market.SH, "600000", 0, 1)
    res = cmd.parse_response(bytes(body))
    
    assert len(res) == 1
    assert res[0].year == 2025
    assert res[0].month == 1
    assert res[0].day == 8