def setUp(self): """This optional method is called before every test method """ db.drop_all() users = [] users.append( user.User(username="******", name="name", email="*****@*****.**", password_hash="password_hash", salt="salt", profile_picture=b"profile_picture")) users.append( user.User(username="******", password_hash="iiojfeaioieof", salt="saltySalt")) events = [] events.append( event.Event(name="Tie my shoe", owner_id=1, event_type=event.EventType.NOTE)) events.append( event.Event(name="Raid Area 51", owner_id=1, parent_id=1, event_type=event.EventType.NOTE)) db.create_all() for value in users: db.session.add(value) for value in events: db.session.add(value) db.session.commit()
def setUp(self): """This optional method is called before every test method """ db.drop_all() users = [] users.append( user.User(username="******", name="name", email="*****@*****.**", password_hash="password_hash", salt="salt", profile_picture=b"profile_picture")) users.append( user.User(username="******", password_hash="iiojfeaioieof", salt="saltySalt")) users.append( user.User(username="******", name="jeff bob", password_hash="eeeeeeeeeeeeeee", salt="fffffffffffffff")) users.append( user.User(username="******", email="*****@*****.**", password_hash="asdfafeadf", salt="graefgafae")) db.create_all() for value in users: db.session.add(value) db.session.commit()
def test_getUser(self): person = user.getUser("username") self.assertEqual(person.salt, "salt") #Retreive by username works testPerson = user.User.query.filter_by(username="******").first() person = user.getUser(testPerson.id) self.assertEqual(person.name, "jeff bob") #Retreive by id works self.assertRaises(TypeError, lambda : user.getUser(user.User())) #Does not accept User objects self.assertRaises(TypeError, lambda : user.getUser(3.4)) #Does not accept floats