from datetime import date
from decimal import Decimal

from group import (
    draft_from_group,
    group_by_supplier,
    is_claimable,
    missing_mdm_reason,
    money4,
)
from repo import PurchaseContractRow


def _row(**overrides) -> PurchaseContractRow:
    data = dict(
        fid=1,
        oms_id=2026100501,
        supplier_id="SUP001",
        supplier_name="示例供应商",
        contract_sign_date=date(2026, 10, 1),
        contract_start_date=date(2026, 10, 2),
        contract_end_date=date(2027, 9, 30),
        payment_term=90,
        goods_id="G0001",
        goods_name="阿莫西林胶囊",
        reg_no="国药准字H20201234",
        spec="0.25g*24粒",
        unit="盒",
        qty=Decimal("10.0000"),
        unit_price=Decimal("100.0000"),
        purchaser_id="P001",
        purchaser_name="张三",
        planner_id="PL001",
        planner_name="李四",
        contract_entry_status=None,
        contract_no=None,
        process_status=0,
        org_code="901",
        goods_mdm_code="QX0100167583",
        supplier_mdm_code="DW0100046361",
    )
    data.update(overrides)
    return PurchaseContractRow(**data)


def test_protocol_rerun_is_not_claimable():
    row = _row(contract_entry_status=0, contract_no="HT-OLD")
    assert is_claimable(row) is False


def test_blank_contract_no_is_still_claimable():
    assert is_claimable(_row(contract_entry_status=0, contract_no="  ")) is True
    assert is_claimable(_row(contract_entry_status=0, contract_no="")) is True
    assert is_claimable(_row(contract_entry_status=0, contract_no=None)) is True


def test_processing_or_done_is_not_claimable():
    assert is_claimable(_row(process_status=3)) is False
    assert is_claimable(_row(process_status=1)) is False


def test_group_by_supplier_keeps_fid_order_and_first_header():
    rows = [
        _row(fid=2, supplier_id="B", payment_term=30, supplier_mdm_code="MDM-B"),
        _row(fid=1, supplier_id="A", payment_term=90, supplier_mdm_code="MDM-A"),
        _row(fid=3, supplier_id="A", payment_term=60, oms_id=2026100502, goods_id="G0002"),
    ]
    groups = group_by_supplier(rows)
    assert [g[0].supplier_id for g in groups] == ["A", "B"]
    assert [r.fid for r in groups[0]] == [1, 3]
    draft = draft_from_group(groups[0])
    assert draft.customer_code == "MDM-A"
    assert draft.payment_days == "90"
    assert draft.order_date == "2026-10-01"
    assert draft.contract_sign_date == "2026-10-01"
    assert draft.contract_start_date == "2026-10-02"
    assert draft.contract_end_date == "2027-09-30"
    assert draft.amount == "2000.0000"
    assert draft.source_type == ""
    assert draft.is_prepayment == "否"
    assert len(draft.goods_lines) == 2
    assert draft.goods_lines[0].goods_code == "QX0100167583"
    assert draft.goods_lines[0].operation_code == "QX0100167583"
    assert draft.goods_lines[0].mdm_code == "QX0100167583"
    assert draft.goods_lines[0].unit_price == "100.0000"
    assert draft.belong_year == "2026"


def test_missing_mdm_reason():
    rows = [_row(supplier_mdm_code="")]
    assert missing_mdm_reason(rows) == "缺少供应商MDM码"
    assert missing_mdm_reason([_row(goods_mdm_code="")]) == "缺少货品MDM码"
    assert missing_mdm_reason([_row()]) is None


def test_money4():
    assert money4(Decimal("10.5")) == "10.5000"