edit tests for settlements

Author: rserna2010

balanced/resources.py

@@ -602,10 +602,10 @@ class Account(FundingInstrument):
 
     uri_gen = wac.URIGen('/accounts', '{account}')
 
-    def settle(self, destination, **kwargs):
+    def settle(self, funding_instrument, **kwargs):
         return Settlement(
             href=self.settlements.href,
-            destination=destination,
+            funding_instrument=funding_instrument,
             **kwargs
         )
 

tests/test_suite.py

@@ -512,6 +512,7 @@ def test_accounts_transfer(self):
 
         order.debit_from(source=card, amount=1234)
         sweep_account = merchant.account
+        self.assertEqual(sweep_account.balance, 0)
         account_credit = sweep_account.credit(amount=1234, order=order.href,
                                               appears_on_statement_as='Payout')
         self.assertEqual(account_credit.status, 'succeeded')
@@ -522,12 +523,14 @@ def test_accounts_transfer_from_multiple_orders(self):
         merchant = balanced.Customer().save()
         card = balanced.Card(**INTERNATIONAL_CARD).save()
         sweep_account = merchant.account
+        self.assertEqual(sweep_account.balance, 0)
         amount = 1234
 
         order_one = merchant.create_order()
         order_one.debit_from(source=card, amount=amount)
         account_credit_one = sweep_account.credit(amount=amount,
                                                   order=order_one.href)
+        self.assertEqual(sweep_account.balance, amount)
         order_two = merchant.create_order()
         order_two.debit_from(source=card, amount=amount)
         account_credit_two = sweep_account.credit(amount=amount,
@@ -543,6 +546,7 @@ def test_settlement(self):
         sweep_account = merchant.account
         account_credit = sweep_account.credit(amount=1234, order=order.href,
                                               appears_on_statement_as='Payout')
+        self.assertEqual(sweep_account.balance, 1234)
         bank_account = balanced.BankAccount(
             account_number='1234567890',
             routing_number='321174851',
@@ -551,13 +555,75 @@ def test_settlement(self):
         bank_account.associate_to_customer(merchant)
 
         settlement = sweep_account.settle(
-            destination=bank_account.href,
+            funding_instrument=bank_account.href,
             appears_on_statement_as="Settlement Oct",
             description="Settlement for payouts from October")
         self.assertEqual(settlement.amount, 1234)
         self.assertEqual(settlement.appears_on_statement_as, "Settlement Oct")
         self.assertEqual(settlement.description,
                          "Settlement for payouts from October")
+        self.assertEqual(sweep_account.balance, 0)
+
+    def test_reverse_settlement(self):
+        merchant = balanced.Customer().save()
+        order = merchant.create_order()
+        card = balanced.Card(**INTERNATIONAL_CARD).save()
+
+        order.debit_from(source=card, amount=1234)
+        sweep_account = merchant.account
+        account_credit = sweep_account.credit(amount=1234, order=order.href,
+                                              appears_on_statement_as='Payout')
+        self.assertEqual(sweep_account.balance, 1234)
+
+        bank_account = balanced.BankAccount(
+            account_number='1234567890',
+            routing_number='321174851',
+            name='Someone',
+            ).save()
+        bank_account.associate_to_customer(merchant)
+
+        settlement = sweep_account.settle(
+            funding_instrument=bank_account.href,
+            appears_on_statement_as="Settlement Oct",
+            description="Settlement for payouts from October")
+        self.assertEqual(sweep_account.balance, 0)
+
+        credit_from_escrow = sweep_account.credit(amount=1234)
+        self.assertEqual(sweep_account.balance, 1234)
+
+        account_credit.reverse(amount=1234)
+        self.assertEqual(sweep_account.balance, 0)
+
+    def test_reverse_settlement_with_negative_account_balance(self):
+        merchant = balanced.Customer().save()
+        order = merchant.create_order()
+        card = balanced.Card(**INTERNATIONAL_CARD).save()
+
+        order.debit_from(source=card, amount=1234)
+        sweep_account = merchant.account
+        account_credit = sweep_account.credit(amount=1234, order=order.href,
+                                              appears_on_statement_as='Payout')
+        bank_account = balanced.BankAccount(
+            account_number='1234567890',
+            routing_number='321174851',
+            name='Someone',
+            ).save()
+        bank_account.associate_to_customer(merchant)
+
+        settlement = sweep_account.settle(
+            funding_instrument=bank_account.href,
+            appears_on_statement_as="Settlement Oct",
+            description="Settlement for payouts from October")
+        self.assertEqual(sweep_account.balance, 0)
+
+        account_credit.reverse(amount=1234)
+        self.assertEqual(sweep_account.balance, -1234)
+
+        settlement = sweep_account.settle(
+            funding_instrument=bank_account.href,
+            appears_on_statement_as="Settlement Oct",
+            description="Settlement for payouts from October")
+        self.assertEqual(sweep_account.balance, 0)
 
 
 class Rev0URIBasicUseCases(unittest.TestCase):