"""KIS poll_fill external-cancel detection (dcy-013 / REQ-012, issue #9).

inquire-daily-ccld marks a cancelled order with cncl_yn="Y". These tests stub
the HTTP layer (token pre-seeded, httpx.get faked) and assert the FillStatus
mapping for full cancel / partial-fill-then-cancel / not-cancelled.
"""
from __future__ import annotations

from datetime import datetime, timedelta, timezone

from trex_core.adapters.broker import kis as kis_mod
from trex_core.adapters.broker.kis import KISBroker

ODNO = "0000123456"


def _broker():
    b = KISBroker(
        "12345678-01",
        app_key="k", app_secret="s",
        base_url="https://openapivts.koreainvestment.com:29443",  # mock endpoint
    )
    b._token = "tok"
    b._token_exp = datetime.now(tz=timezone.utc) + timedelta(hours=1)
    return b


class _Resp:
    def __init__(self, payload):
        self._payload = payload

    def json(self):
        return self._payload


def _fake_get(payload):
    def _get(*a, **k):
        return _Resp(payload)
    return _get


def _row(**kw):
    base = {"odno": ODNO, "ord_qty": "10", "tot_ccld_qty": "0",
            "avg_prvs": "0", "cncl_yn": "N", "ord_dt": "20260608", "ord_gno_brno": "06010"}
    base.update(kw)
    return {"output1": [base]}


def test_full_external_cancel_maps_to_cancelled(monkeypatch):
    monkeypatch.setattr(kis_mod.httpx, "get", _fake_get(_row(tot_ccld_qty="0", cncl_yn="Y")))
    fs = _broker().poll_fill(f"06010:{ODNO}", side="sell", symbol="005930")
    assert fs.state == "cancelled" and fs.is_terminal_reject and not fs.is_partial_cancel


def test_zero_fill_not_cancelled_stays_accepted(monkeypatch):
    monkeypatch.setattr(kis_mod.httpx, "get", _fake_get(_row(tot_ccld_qty="0", cncl_yn="N")))
    fs = _broker().poll_fill(f"06010:{ODNO}", side="sell", symbol="005930")
    assert fs.state == "accepted" and not fs.is_terminal_reject


def test_partial_fill_then_cancel_flags_partial_cancel(monkeypatch):
    monkeypatch.setattr(
        kis_mod.httpx, "get",
        _fake_get(_row(ord_qty="10", tot_ccld_qty="4", avg_prvs="70000", cncl_yn="Y")),
    )
    fs = _broker().poll_fill(f"06010:{ODNO}", side="sell", symbol="005930")
    assert fs.state == "partially_filled"
    assert fs.cancelled and fs.is_partial_cancel
    assert fs.filled_qty == 4 and fs.avg_price == 70000.0
    assert fs.execution_id == "kis-daily-ccld:20260608:06010:0000123456:4"


def test_full_fill_unaffected_by_cancel_flag(monkeypatch):
    monkeypatch.setattr(
        kis_mod.httpx, "get",
        _fake_get(_row(ord_qty="10", tot_ccld_qty="10", avg_prvs="70000", cncl_yn="N")),
    )
    fs = _broker().poll_fill(f"06010:{ODNO}", side="sell", symbol="005930")
    assert fs.is_filled and fs.filled_qty == 10 and not fs.is_partial_cancel