"""趋势回调运行中计划：实际成交价重算补仓表与金额盈亏比。"""
from __future__ import annotations

import json
import sys
import unittest
from pathlib import Path

ROOT = Path(__file__).resolve().parents[1]
sys.path.insert(0, str(ROOT))

from strategy_snapshot_lib import attach_trend_dca_levels  # noqa: E402
from strategy_trend_lib import calc_trend_plan_money_metrics  # noqa: E402


class TestTrendDcaEnrichFills(unittest.TestCase):
    def _base_plan(self, **overrides):
        plan = {
            "direction": "long",
            "stop_loss": 0.329,
            "take_profit": 0.476,
            "first_order_amount": 115,
            "snapshot_available_usdt": 97.98,
            "risk_percent": 5,
            "contract_size": 1.0,
            "grid_prices_json": json.dumps([0.3465, 0.343, 0.3395, 0.336, 0.3325]),
            "leg_amounts_json": json.dumps([23, 23, 23, 23, 23]),
            "dca_legs": 5,
            "first_order_done": 1,
            "legs_done": 0,
            "avg_entry_price": 0.3537,
            "order_amount_open": 115,
            "target_order_amount": 230,
            "leg_fill_prices_json": json.dumps([0.3537]),
        }
        plan.update(overrides)
        return plan

    def test_header_money_rr_not_price_rr(self):
        plan = self._base_plan()
        metrics = calc_trend_plan_money_metrics(plan)
        self.assertAlmostEqual(metrics["risk_amount_u"], 4.899, places=2)
        self.assertIsNotNone(metrics["money_rr"])
        self.assertLess(metrics["money_rr"], 4.0)

    def test_done_dca_uses_actual_fill_price(self):
        plan = self._base_plan(
            legs_done=1,
            avg_entry_price=0.3512,
            order_amount_open=138,
            leg_fill_prices_json=json.dumps([0.3537, 0.3458]),
        )
        enriched = attach_trend_dca_levels(plan)
        levels = enriched["dca_levels"]
        self.assertEqual(len(levels), 6)
        dca1 = levels[1]
        self.assertEqual(dca1["status"], "done")
        self.assertAlmostEqual(dca1["price"], 0.3458, places=4)
        self.assertIsNotNone(dca1["avg_entry"])
        self.assertIsNotNone(dca1["rr"])
        dca2 = levels[2]
        self.assertEqual(dca2["status"], "pending")
        self.assertAlmostEqual(dca2["price"], 0.343, places=4)

    def test_missing_dca_fills_align_last_avg_with_header(self):
        """缺补仓成交价时，末档加仓后均价应对齐计划头部 avg_entry_price。"""
        plan = self._base_plan(
            legs_done=2,
            avg_entry_price=0.3507,
            order_amount_open=161,
            leg_fill_prices_json=json.dumps([0.3436]),
            grid_prices_json=json.dumps([0.343, 0.343, 0.3395, 0.336, 0.3325]),
        )
        enriched = attach_trend_dca_levels(plan)
        levels = enriched["dca_levels"]
        dca2 = levels[2]
        self.assertEqual(dca2["status"], "done")
        self.assertAlmostEqual(dca2["avg_entry"], 0.3507, places=4)
        self.assertGreater(dca2["price"], 0.343)


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