zkxw/backend/tests/test_protocol_refund_record.py

"""协议级退费登记校验单测"""

import unittest

from app.services.protocol_refund_record import validate_refund_amount


class TestValidateRefundAmount(unittest.TestCase):
    def test_refund_amount_must_be_positive(self):
        with self.assertRaises(ValueError):
            validate_refund_amount(0)
        with self.assertRaises(ValueError):
            validate_refund_amount(-1)
        with self.assertRaises(ValueError):
            validate_refund_amount(None)
        self.assertEqual(validate_refund_amount(0.01), validate_refund_amount("0.01"))


if __name__ == "__main__":
    unittest.main()