"""instance_dashboard_lib 单元测试."""
from __future__ import annotations

import sqlite3
import unittest

from lib.instance.instance_dashboard_lib import build_instance_dashboard_payload


def _mem_conn() -> sqlite3.Connection:
    conn = sqlite3.connect(":memory:")
    conn.row_factory = sqlite3.Row
    conn.executescript(
        """
        CREATE TABLE order_monitors (
            id INTEGER PRIMARY KEY,
            symbol TEXT,
            exchange_symbol TEXT,
            direction TEXT,
            status TEXT,
            monitor_type TEXT,
            key_signal_type TEXT,
            trigger_price REAL,
            stop_loss REAL,
            take_profit REAL
        );
        CREATE TABLE key_monitors (
            id INTEGER PRIMARY KEY,
            symbol TEXT,
            exchange_symbol TEXT,
            direction TEXT,
            signal_type TEXT,
            upper REAL,
            lower REAL,
            status TEXT
        );
        CREATE TABLE trend_pullback_plans (
            id INTEGER PRIMARY KEY,
            symbol TEXT,
            exchange_symbol TEXT,
            direction TEXT,
            status TEXT,
            entry_price REAL
        );
        CREATE TABLE roll_groups (
            id INTEGER PRIMARY KEY,
            order_monitor_id INTEGER,
            symbol TEXT,
            exchange_symbol TEXT,
            direction TEXT,
            status TEXT
        );
        """
    )
    return conn


class TestInstanceDashboardLib(unittest.TestCase):
    def test_empty_sections_and_conditional_hidden(self):
        conn = _mem_conn()
        payload = build_instance_dashboard_payload(conn, hedge_enabled=True)
        self.assertTrue(payload["ok"])
        self.assertEqual(payload["orders"]["count"], 0)
        self.assertEqual(payload["keys"]["count"], 0)
        self.assertEqual(payload["strategy"]["count"], 0)
        self.assertFalse(payload["options"]["visible"])
        self.assertFalse(payload["hedge_plan"]["visible"])
        conn.close()

    def test_orders_keys_strategy_and_options_visible(self):
        conn = _mem_conn()
        conn.execute(
            "INSERT INTO order_monitors (symbol, exchange_symbol, direction, status, monitor_type) "
            "VALUES ('BTC/USDT', 'BTC/USDT:USDT', 'long', 'active', 'manual')"
        )
        conn.execute(
            "INSERT INTO key_monitors (symbol, direction, signal_type, upper, lower, status) "
            "VALUES ('ETH/USDT', 'short', '箱体突破', 3000, 2800, 'active')"
        )
        conn.execute(
            "INSERT INTO trend_pullback_plans (symbol, direction, status, entry_price) "
            "VALUES ('SOL/USDT', 'long', 'active', 100)"
        )
        conn.execute(
            "INSERT INTO order_monitors (id, symbol, direction, status) VALUES (9, 'XRP/USDT', 'short', 'active')"
        )
        conn.execute(
            "INSERT INTO roll_groups (order_monitor_id, symbol, direction, status) "
            "VALUES (9, 'XRP/USDT', 'short', 'active')"
        )
        conn.commit()

        def fetch_opts():
            return [{"inst_id": "ETH-USD-260731-3000-C", "opt_type": "C", "pos": 1, "upl": 1.5}]

        payload = build_instance_dashboard_payload(
            conn,
            fetch_options_positions=fetch_opts,
            hedge_enabled=False,
        )
        self.assertEqual(payload["orders"]["count"], 2)
        self.assertEqual(payload["keys"]["count"], 1)
        self.assertEqual(payload["strategy"]["count"], 2)
        self.assertTrue(payload["options"]["visible"])
        self.assertEqual(payload["options"]["count"], 1)
        self.assertFalse(payload["hedge_plan"]["visible"])
        conn.close()


if __name__ == "__main__":
    unittest.main()