zlwuandCursor
628a20dbbd
Supports 缴费明细 list-mode row delete for draft transfers. Co-authored-by: Cursor <cursoragent@cursor.com>
56 lines
1.9 KiB
Python
56 lines
1.9 KiB
Python
"""软删除待缴费（dfj）转账笔"""
|
|
|
|
import unittest
|
|
from types import SimpleNamespace
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
from app.crud import protocol_sample_payment as psp
|
|
|
|
|
|
class TestSoftDeletePaymentTransfer(unittest.TestCase):
|
|
def test_function_exists(self):
|
|
self.assertTrue(callable(getattr(psp, "soft_delete_payment_transfer", None)))
|
|
|
|
def test_soft_delete_dfj_sets_del_flag_and_commits(self):
|
|
t = SimpleNamespace(
|
|
id="t1",
|
|
pay_status="dfj",
|
|
del_flag=0,
|
|
update_by=None,
|
|
update_time=None,
|
|
)
|
|
db = MagicMock()
|
|
orig = psp._get_transfer_or_raise
|
|
psp._get_transfer_or_raise = lambda *a, **k: t
|
|
try:
|
|
with patch.object(psp, "_now", return_value="2026-10-10T12:00:00"):
|
|
psp.soft_delete_payment_transfer(
|
|
db, transfer_id="t1", tenant_no="zkxw", user_id="u1"
|
|
)
|
|
self.assertEqual(t.del_flag, 1)
|
|
self.assertEqual(t.update_by, "u1")
|
|
self.assertEqual(t.update_time, "2026-10-10T12:00:00")
|
|
db.add.assert_called_once_with(t)
|
|
db.commit.assert_called_once()
|
|
finally:
|
|
psp._get_transfer_or_raise = orig
|
|
|
|
def test_soft_delete_rejects_non_dfj(self):
|
|
t = SimpleNamespace(id="t1", pay_status="jfqrz", del_flag=0)
|
|
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.soft_delete_payment_transfer(
|
|
db, transfer_id="t1", tenant_no="zkxw", user_id="u1"
|
|
)
|
|
self.assertIn("删除", str(ctx.exception))
|
|
db.commit.assert_not_called()
|
|
finally:
|
|
psp._get_transfer_or_raise = orig
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|