1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42
|
from tests.test_helper import *
from datetime import date
class TestDisbursement(unittest.TestCase):
attributes = {
"merchant_account": {
"id": "sub_merchant_account",
"status": "active",
"master_merchant_account": {
"id": "master_merchant_account",
"status": "active"
},
},
"id": "123456",
"exception_message": "invalid_account_number",
"amount": "100.00",
"disbursement_date": date(2013, 4, 10),
"follow_up_action": "update",
"transaction_ids": ["asdf", "qwer"],
"disbursement_type": "credit"
}
def test_constructor(self):
disbursement = Disbursement(None, TestDisbursement.attributes)
self.assertEqual("123456", disbursement.id)
self.assertEqual(Decimal("100.00"), disbursement.amount)
self.assertEqual(["asdf", "qwer"], disbursement.transaction_ids)
self.assertEqual("master_merchant_account", disbursement.merchant_account.master_merchant_account.id)
def test_credit(self):
disbursement = Disbursement(None, TestDisbursement.attributes)
self.assertTrue(disbursement.is_credit())
def test_debit(self):
thing = TestDisbursement.attributes
thing["disbursement_type"] = "debit"
disbursement = Disbursement(None, TestDisbursement.attributes)
self.assertTrue(disbursement.is_debit())
|