def test_get_estate_board(self):
self.estate1.save()
self.board1.save()
board1 = Board.get(id=1)
board1.insert('estates_owned', [Estate.get(id=1)])
estate1 = Estate.get(id=1)
self.assertEqual(1, estate1.board_id)
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_add_and_remove_estates_owned_to_board(self):
self.board1.save()
self.estate1.save()
self.estate2.save()
board1 = Board.get(id=1)
board1.insert('estates_owned', [Estate.get(id=1), Estate.get(id=2)])
estates = Board.get(id=1).estates_owned
self.assertEqual("Random Address 1", estates[0].address)
self.assertEqual("Random Address 2", estates[1].address)
self.assertEqual(2, len(estates))
self.assertEqual(True, isinstance(estates[0], Estate))
self.assertEqual(True, isinstance(estates[1], Estate))
board1.remove('estates_owned', id=1)
self.assertEqual(1, len(Board.get(id=1).estates_owned))
def test_get_many_estates(self):
self.assertEqual(True, isinstance(Estate.get(id=1), dict))
self.estate1.save()
self.estate2.save()
self.assertEqual(True, isinstance(Estate.get_all()[0], Estate))
self.assertEqual(True, isinstance(Estate.get_all()[1], Estate))
self.assertEqual(2, len(Estate.get_all()))
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_delete_board(self):
self.board1.save()
self.estate1.save()
self.unit1.save()
board1 = Board.get(id=1)
board1.insert('estates_owned', [Estate.get(id=1)])
board1.units_owned.append(Unit.get(id=1))
self.assertEqual(True, Board.get(id=1).delete())
self.assertEqual(True, isinstance(Board.get(id=1), dict))
def test_add_and_remove_units_from_estate(self):
self.estate1.save()
self.unit1.save()
self.unit2.save()
estate1 = Estate.get(id=1)
self.assertEqual(0, len(estate1.estate_units))
estate1.insert('estate_units', [Unit.get(id=1), Unit.get(id=2)])
self.assertEqual(2, len(estate1.estate_units))
self.assertEqual(True, isinstance(estate1.estate_units[0], Unit))
self.assertEqual(True, isinstance(estate1.estate_units[1], Unit))
estate1.remove('estate_units', id=1)
self.assertEqual(1, len(estate1.estate_units))
def test_add_and_remove_board_estate_resident(self):
self.board1.save()
self.estate1.save()
self.user1.save()
self.unit1.save()
unit1 = Unit.get(id=1)
unit1.insert('resident', User.get(id=1))
unit1.insert('board', Board.get(id=1))
unit1.insert('estate', Estate.get(id=1))
self.assertEqual(True, isinstance(unit1.board, Board))
self.assertEqual(True, isinstance(unit1.resident, User))
self.assertEqual(True, isinstance(unit1.estate, Estate))
unit1.remove('resident')
unit1.remove('board')
unit1.remove('estate')
self.assertEqual(None, unit1.board)
self.assertEqual(None, unit1.estate)
self.assertEqual(None, unit1.resident)
def test_get_estate(self):
self.assertEqual(True, isinstance(Estate.get(id=1), dict))
self.estate1.save()
self.assertEqual(True, isinstance(Estate.get(id=1), Estate))