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))