def setUp(self): db_session.query(User).delete() email = '*****@*****.**' password = '******' u = User(email, password, u'firstname', u'lastname') u.is_admin = True db_session.add(u) db_session.commit()
def _fixture_setup(self): """ if has fixture, use pickle install it """ db_session.query(User).delete() db_session.commit() user = User('*****@*****.**', '123456', 'ff', 'tt') user.is_admin = True db_session.add(user) db_session.commit()
def register(email, password, firstname, lastname): """ register new user TODO: 设计输出, try ..except """ user = User(email, password) user.firstname = firstname user.lastname = lastname db_session.add(user).commit() return user