105 lines
2.8 KiB
Python
105 lines
2.8 KiB
Python
from decimal import Decimal
|
|
|
|
from job import run_yaoex_pay
|
|
from settle import CartCandidate, MallLine
|
|
|
|
|
|
class MemoryPayRepo:
|
|
def __init__(self, carts: list[CartCandidate], existing=None):
|
|
self.carts = carts
|
|
self.existing = existing or set()
|
|
self.saved: list[dict] = []
|
|
|
|
def load_cart_candidates(self):
|
|
return list(self.carts)
|
|
|
|
def load_existing_pay_keys(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"),
|
|
pay_order_dtl_id="",
|
|
)
|
|
data.update(overrides)
|
|
return MallLine(**data)
|
|
|
|
|
|
def test_job_attaches_dtl_matches_cart_and_inserts():
|
|
repo = MemoryPayRepo(
|
|
[
|
|
CartCandidate(
|
|
fid=1,
|
|
oms_id=2026100201,
|
|
supplier_name="广东壹号药业有限公司",
|
|
bar_code="6921665004933",
|
|
purchase_qty=10,
|
|
org_code="901",
|
|
multi_platform_flag=0,
|
|
)
|
|
]
|
|
)
|
|
details = {
|
|
"ZQD1": {
|
|
"productList": [
|
|
{
|
|
"orderDetailId": 881,
|
|
"productName": "阿莫西林",
|
|
"quantity": 10,
|
|
"payPrice4": 9.85,
|
|
}
|
|
]
|
|
}
|
|
}
|
|
summary = run_yaoex_pay(repo, [_line()], details)
|
|
assert summary.inserted == 1
|
|
assert repo.saved[0]["oms_id"] == 2026100201
|
|
assert repo.saved[0]["pay_order_dtl_id"] == "881"
|
|
assert repo.saved[0]["purchase_total"] == Decimal("98.5000")
|
|
|
|
|
|
def test_job_dry_run_does_not_insert():
|
|
repo = MemoryPayRepo(
|
|
[
|
|
CartCandidate(
|
|
fid=1,
|
|
oms_id=1,
|
|
supplier_name="广东壹号药业有限公司",
|
|
bar_code="6921665004933",
|
|
purchase_qty=10,
|
|
org_code="901",
|
|
multi_platform_flag=0,
|
|
)
|
|
]
|
|
)
|
|
details = {
|
|
"ZQD1": {
|
|
"productList": [
|
|
{
|
|
"orderDetailId": 881,
|
|
"productName": "阿莫西林",
|
|
"quantity": 10,
|
|
"payPrice4": 9.85,
|
|
}
|
|
]
|
|
}
|
|
}
|
|
summary = run_yaoex_pay(repo, [_line()], details, dry_run=True)
|
|
assert summary.inserted == 1
|
|
assert repo.saved == []
|