101 lines
3.5 KiB
Python
101 lines
3.5 KiB
Python
"""按样本缴费：转账状态机与批次已缴/欠缴计算"""
|
|
|
|
from decimal import Decimal
|
|
import unittest
|
|
|
|
from app.services.protocol_transfer_status import (
|
|
can_confirm_receipt,
|
|
can_edit_transfer,
|
|
can_submit_receipt,
|
|
schedule_paid_and_arrears,
|
|
)
|
|
|
|
|
|
class TestTransferStatusRules(unittest.TestCase):
|
|
def test_submit_only_dfj(self):
|
|
self.assertTrue(can_submit_receipt("dfj"))
|
|
self.assertFalse(can_submit_receipt("jfqrz"))
|
|
self.assertFalse(can_submit_receipt("yjf"))
|
|
self.assertFalse(can_submit_receipt(""))
|
|
self.assertFalse(can_submit_receipt(None))
|
|
|
|
def test_confirm_only_jfqrz(self):
|
|
self.assertTrue(can_confirm_receipt("jfqrz"))
|
|
self.assertFalse(can_confirm_receipt("dfj"))
|
|
self.assertFalse(can_confirm_receipt("yjf"))
|
|
|
|
def test_edit_before_finance_confirm(self):
|
|
"""已付款、付款确认中可编辑；付款完成不可编辑。"""
|
|
self.assertTrue(can_edit_transfer("dfj"))
|
|
self.assertTrue(can_edit_transfer("jfqrz"))
|
|
self.assertFalse(can_edit_transfer("yjf"))
|
|
self.assertFalse(can_edit_transfer(""))
|
|
self.assertFalse(can_edit_transfer(None))
|
|
|
|
def test_schedule_arrears(self):
|
|
paid, arrears = schedule_paid_and_arrears(
|
|
due=Decimal("1000"),
|
|
confirmed_transfers=[Decimal("300"), Decimal("200")],
|
|
)
|
|
self.assertEqual(paid, Decimal("500"))
|
|
self.assertEqual(arrears, Decimal("500"))
|
|
|
|
def test_schedule_arrears_overpay(self):
|
|
paid, arrears = schedule_paid_and_arrears(
|
|
due=Decimal("100"),
|
|
confirmed_transfers=[Decimal("60"), Decimal("50")],
|
|
)
|
|
self.assertEqual(paid, Decimal("110"))
|
|
self.assertEqual(arrears, Decimal("-10"))
|
|
|
|
|
|
class TestProtocolWithdrawReceipt(unittest.TestCase):
|
|
def test_withdraw_sets_dfj_keeps_finance_remark(self):
|
|
from types import SimpleNamespace
|
|
from unittest.mock import MagicMock
|
|
from app.crud import protocol_sample_payment as psp
|
|
|
|
t = SimpleNamespace(
|
|
id="t1",
|
|
pay_status="jfqrz",
|
|
finance_remark="keep-me",
|
|
finance_confirm_attachments="http://x",
|
|
update_by=None,
|
|
update_time=None,
|
|
)
|
|
db = MagicMock()
|
|
orig = psp._get_transfer_or_raise
|
|
psp._get_transfer_or_raise = lambda *a, **k: t
|
|
try:
|
|
out = psp.withdraw_receipt(
|
|
db, transfer_id="t1", tenant_no="zkxw", user_id="u1"
|
|
)
|
|
self.assertEqual(out.pay_status, "dfj")
|
|
self.assertEqual(out.finance_remark, "keep-me")
|
|
self.assertEqual(out.finance_confirm_attachments, "http://x")
|
|
db.commit.assert_called()
|
|
finally:
|
|
psp._get_transfer_or_raise = orig
|
|
|
|
def test_withdraw_rejects_non_jfqrz(self):
|
|
from types import SimpleNamespace
|
|
from unittest.mock import MagicMock
|
|
from app.crud import protocol_sample_payment as psp
|
|
|
|
t = SimpleNamespace(id="t1", pay_status="dfj", finance_remark="x")
|
|
db = MagicMock()
|
|
orig = psp._get_transfer_or_raise
|
|
psp._get_transfer_or_raise = lambda *a, **k: t
|
|
try:
|
|
with self.assertRaises(ValueError) as ctx:
|
|
psp.withdraw_receipt(
|
|
db, transfer_id="t1", tenant_no="zkxw", user_id="u1"
|
|
)
|
|
self.assertIn("撤回", str(ctx.exception))
|
|
finally:
|
|
psp._get_transfer_or_raise = orig
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|