def setUp(self): from sqlalchemy import MetaData meta = MetaData(bind=dao.engine) meta.reflect() for table in reversed(meta.sorted_tables): dao.engine.execute(table.delete()) self.expected_username = "******" self.expected_hash = "somehash" user = User(username=self.expected_username, hash=self.expected_hash) dao.save_user(user)
def test_save_user_returns_false_when_the_hash_is_missing(self): expected_username = '******' user = User(username=expected_username) success = dao.save_user(user) self.assertFalse(success, "Persisting a user with a missing hash" "should not succeed.")
def test_save_user(self): expected_username = '******' expected_hash = 'some_hash' user = User(username=expected_username, hash=expected_hash) success = dao.save_user(user) self.assertTrue(success, "The user was not successfully persisted" "to the database.") session = dao.DBSession() users = session.query(User).all() self.assertEqual(len(users), 1, "Expected only 1 user to be returned!") persisted_user = users[0] self.assertEqual(expected_username, persisted_user.username, "The persisted username did not match the " "expected username.") self.assertEqual(expected_hash, persisted_user.hash, "The persisted hash did not match the expected hash.")