class TestPaymentModel(VokoTestCase):
def setUp(self):
self.payment = PaymentFactory()
def test_create_and_link_credit(self):
balance = self.payment.create_and_link_credit()
self.assertEqual(balance.user, self.payment.order.user)
self.assertEqual(balance.type, "CR")
self.assertEqual(balance.amount, self.payment.amount)
self.assertEqual(balance.notes, "iDeal betaling voor bestelling #%d" % self.payment.order.id)
self.assertEqual(balance.payment, self.payment)
def test_create_and_link_credit_does_not_overwrite_existing_balance(self):
balance = self.payment.create_and_link_credit()
self.payment.create_and_link_credit()
payment = Payment.objects.get(id=self.payment.id)
self.assertEqual(balance, payment.balance)
