def testSaveUpdate(self): caroline = Account(username="******", password_hash="password", balance=10000, first_name="Caroline", last_name="Grabowski", email="*****@*****.**") caroline.save() caroline_id = caroline.id caroline2 = Account.from_id(caroline_id) caroline2.username = "******" caroline2.balance = 20000 caroline2.first_name = "Caro" caroline2.last_name = "Grabo" caroline2.save() self.assertEqual(caroline2.id, caroline_id, "update should not change ID number") caroline3 = Account.from_id(caroline_id) self.assertEqual(caroline3.username, "cgrabow16", "update should update username") self.assertEqual(caroline3.balance, 20000, "update should update balance") self.assertEqual(caroline3.first_name, "Caro", "update should update name") self.assertEqual(caroline3.last_name, "Grabo", "update should update name")
def testFromId(self): caroline = Account(username="******", password_hash="password", balance=10000, first_name="Caroline", last_name="Grabowski", email="*****@*****.**") caroline.save() caroline_id = caroline.id caroline2 = Account.from_id(caroline_id) self.assertEqual(caroline2.first_name, "Caroline") alex = Account.from_id(10340923950399) self.assertIsNone(alex, "from_id returns None for nonexistent row")