def test_client_pays_by_parts(self):
credit_transaction = CreditTransaction()
credit_transaction.agent_from = self.client
credit_transaction.agent_to = self.service_provider
credit_transaction.amount = 1000
self.ledger.add_transaction(credit_transaction)
debit_transaction = DebitTransaction()
debit_transaction.transaction = credit_transaction
debit_transaction.amount = 200
debit_transaction.agent_from = self.client
debit_transaction.agent_to = self.service_provider
self.ledger.add_transaction(debit_transaction)
self.assertEqual(200, self.account_manager.get_agent_to_balance(self.service_provider))
self.assertEqual(-800, self.account_manager.get_agent_from_balance(self.client))
debit_transaction = DebitTransaction()
debit_transaction.transaction = credit_transaction
debit_transaction.amount = 800
debit_transaction.agent_from = self.client
debit_transaction.agent_to = self.service_provider
self.ledger.add_transaction(debit_transaction)
self.assertEqual(1000, self.account_manager.get_agent_to_balance(self.service_provider))
self.assertEqual(0, self.account_manager.get_receivable_balance(self.service_provider))
self.assertEqual(0, self.account_manager.get_agent_from_balance(self.client))
self.assertEqual(0, self.account_manager.get_balance(self.client))
self.assertEqual(1000, self.account_manager.get_balance(self.service_provider))
def test_payment_transaction(self):
credit_transaction = CreditTransaction()
credit_transaction.agent_from = self.client
credit_transaction.agent_to = self.service_provider
credit_transaction.amount = 200
debit_transaction = DebitTransaction()
debit_transaction.agent_from = self.client
debit_transaction.agent_to = self.service_provider
debit_transaction.transaction = credit_transaction
debit_transaction.amount = 200
self.assertEqual(credit_transaction.agent_from, debit_transaction.agent_from)
self.assertEqual(credit_transaction.agent_to, debit_transaction.agent_to)
self.assertEqual(200, debit_transaction.amount)
self.assertEqual(TRANSACTION_DEBIT, debit_transaction.transaction_type)
def test_client_is_billed_and_pays_full(self):
credit_transaction = CreditTransaction()
credit_transaction.agent_from = self.client
credit_transaction.agent_to = self.service_provider
credit_transaction.amount = 1000
self.ledger.add_transaction(credit_transaction)
debit_transaction = DebitTransaction()
debit_transaction.transaction = credit_transaction
debit_transaction.amount = 1000
debit_transaction.agent_from = self.client
debit_transaction.agent_to = self.service_provider
self.ledger.add_transaction(debit_transaction)
self.assertEqual(2, len(self.ledger.get_transactions_from(self.client)))
self.assertEqual(0, len(self.ledger.get_transactions_to(self.client)))
self.assertEqual(2, len(self.ledger.get_transactions_to(self.service_provider)))
def test_add_batch_transactions(self):
transactions = list()
credit_transaction = CreditTransaction()
credit_transaction.agent_from = self.client
credit_transaction.agent_to = self.service_provider
credit_transaction.amount = 1000
debit_transaction = DebitTransaction()
debit_transaction.transaction = credit_transaction
debit_transaction.amount = 1000
debit_transaction.agent_from = self.client
debit_transaction.agent_to = self.service_provider
transactions.append(credit_transaction)
transactions.append(debit_transaction)
self.ledger.add_batch(transactions, 'helo')
self.assertEqual(2, len(self.ledger.get_transactions_from(self.client)))
self.assertEqual(0, len(self.ledger.get_transactions_to(self.client)))
self.assertEqual(2, len(self.ledger.get_transactions_to(self.service_provider)))
self.assertEqual('helo', self.ledger.get_transactions_to(self.service_provider)[0].batch_id)
