from decimal import Decimal

from job import run_yaoex_pay
from settle import MallLine


class MemoryPayRepo:
    def __init__(self, existing=None):
        self.existing = existing or set()
        self.saved: list[dict] = []

    def load_existing_pay_order_nos(self):
        return set(self.existing)

    def save_pay_rows(self, rows):
        self.saved.extend(rows)


def _line(**overrides) -> MallLine:
    data = dict(
        order_no="ZQD1",
        parent_order_no="",
        supplier_name="广东壹号药业有限公司",
        status="待收货",
        goods_name="阿莫西林",
        product_code="2918",
        bar_code="6921665004933",
        unit_price=Decimal("10.00"),
        pay_price=Decimal("9.85"),
        qty=10,
        goods_amount=Decimal("100.00"),
        order_paid=Decimal("98.50"),
    )
    data.update(overrides)
    return MallLine(**data)


def test_job_inserts_generated_dtl_without_details():
    repo = MemoryPayRepo()
    summary = run_yaoex_pay(repo, [_line()])
    assert summary.inserted == 1
    assert summary.unmatched == 0
    assert repo.saved[0]["oms_id"] == 0
    assert repo.saved[0]["pay_order_dtl_id"]
    assert len(repo.saved[0]["pay_order_dtl_id"]) <= 40
    assert repo.saved[0]["purchase_total"] == Decimal("98.5000")


def test_job_dry_run_does_not_insert():
    repo = MemoryPayRepo()
    summary = run_yaoex_pay(repo, [_line()], dry_run=True)
    assert summary.inserted == 1
    assert repo.saved == []


def test_job_skips_existing_order_no():
    repo = MemoryPayRepo(existing={"ZQD1"})
    summary = run_yaoex_pay(repo, [_line()])
    assert summary.inserted == 0
    assert summary.skipped_dup == 1
    assert repo.saved == []