def test_update_payment(self):
self.payment1.save()
payment1 = Payment.get(id=1)
payment1.update({'balance': 1000.0, 'required': 5000.0})
payment1 = Payment.get(id=1)
self.assertEqual(1000.0, payment1.balance)
self.assertEqual(5000.0, payment1.required)
self.assertTrue(isinstance(
payment1.update({'random': 'bad field'}), dict))
def test_add_and_remove_payments_from_wallet(self):
self.wallet1.save()
self.payment1.save()
self.payment2.save()
wallet1 = Wallet.get(id=1)
self.assertEqual(0, len(wallet1.payments))
wallet1.insert('payments', [Payment.get(id=1), Payment.get(id=2)])
wallet1 = Wallet.get(id=1)
self.assertEqual(True, isinstance(wallet1.payments[0], Payment))
self.assertEqual(True, isinstance(wallet1.payments[1], Payment))
self.assertEqual(2, len(wallet1.payments))
wallet1.remove('payments', id=1)
self.assertEqual(1, len(wallet1.payments))
def test_add_and_remove_deposit_estate_unit_wallet(self):
self.deposit1.save()
self.deposit2.save()
self.estate1.save()
self.unit1.save()
self.wallet1.save()
self.payment1.save()
payment1 = Payment.get(id=1)
payment1.estate = Estate.get(id=1)
payment1.unit = Unit.get(id=1)
payment1.wallet = Wallet.get(id=1)
payment1.insert('deposits', [Deposit.get(id=1), Deposit.get(id=2)])
self.assertEqual(True, isinstance(payment1.estate, Estate))
self.assertEqual(True, isinstance(payment1.unit, Unit))
self.assertEqual(True, isinstance(payment1.wallet, Wallet))
self.assertEqual(True, isinstance(payment1.deposits[0], Deposit))
self.assertEqual(True, isinstance(payment1.deposits[1], Deposit))
self.assertEqual(2, len(payment1.deposits))
payment1.remove('estate')
payment1.remove('unit')
payment1.remove('wallet')
self.assertEqual(None, payment1.estate)
self.assertEqual(None, payment1.unit)
self.assertEqual(None, payment1.wallet)
payment1.remove('deposits', id=1)
self.assertEqual(1, len(payment1.deposits))
def test_get_many_payments(self):
self.assertEqual(True, isinstance(Payment.get(id=1), dict))
self.payment1.save()
self.payment2.save()
self.assertEqual(True, isinstance(Payment.get_all()[0], Payment))
self.assertEqual(True, isinstance(Payment.get_all()[1], Payment))
self.assertEqual(2, len(Payment.get_all()))
def test_add_and_remove_payment_to_deposit(self):
self.deposit1.save()
self.payment1.save()
deposit1 = Deposit.get(id=1)
self.assertEqual(None, deposit1.payment)
deposit1.insert('payment', Payment.get(id=1))
self.assertEqual(True, isinstance(deposit1.payment, Payment))
deposit1.remove('payment')
self.assertEqual(None, deposit1.payment)
def test_add_and_remove_payment_from_unit(self):
self.unit1.save()
self.payment1.save()
unit1 = Unit.get(id=1)
self.assertEqual(None, unit1.payment)
payment1 = Payment.get(id=1)
unit1.insert('payment', payment1)
unit1 = Unit.get(id=1)
self.assertEqual(True, isinstance(unit1.payment, Payment))
unit1.remove('payment')
self.assertEqual(None, unit1.payment)
def test_add_and_remove_payment_to_estate(self):
self.estate1.save()
self.payment1.save()
estate1 = Estate.get(id=1)
self.assertEqual(None, estate1.payment)
payment1 = Payment.get(id=1)
estate1.insert('payment', payment1)
estate1 = Estate.get(id=1)
self.assertEqual(True, isinstance(estate1.payment, Payment))
estate1.remove('payment')
self.assertEqual(None, estate1.payment)
def test_delete_payment(self):
self.payment1.save()
self.assertTrue(Payment.check_exists(id=1))
payment1 = Payment.get(id=1)
self.assertTrue(payment1.delete())
self.assertFalse(Payment.check_exists(id=1))
def test_get_payment(self):
self.assertEqual(True, isinstance(Payment.get(id=1), dict))
self.payment1.save()
self.assertEqual(True, isinstance(Payment.get(id=1), Payment))