def useradd(self, username, password): ''' 添加一个用户,如果用户存在则更新用户名和密码。 :username 要添加或更新的用户名称 :password 用户要设置或者更新的密码 ''' user = User.select().where(User.username == username).first() if user: user.password = password else: user = User() user.username = username user.password = password user.role = 'ADMIN' user.locked = False user.save()
def test_new_user(test_client): db.create_all() user = User(first_name='Paul', last_name='Asalu', email='*****@*****.**', tel='09045444444', country='Nigeria', state='Lagos', address='66 Baruwa Street, Agboju', age=22, user_id='1234', sign_up_date=d.datetime.utcnow(), sign_up_method='Firebase', days_left=1) Role.insert_roles() role = Role.query.filter_by(id=1).first() user.role = role Guides.insert_guides() guide = Guides.query.filter_by(id=1).first() user.guides.append(guide) db.session.add(user) db.session.commit() assert user.first_name == 'Paul' assert user.last_name == 'Asalu' assert user.email == '*****@*****.**' assert user.tel == '09045444444' assert user.country == 'Nigeria' assert user.state == 'Lagos' assert user.address == '66 Baruwa Street, Agboju' assert user.travel_history == 0 assert user.age == 22 assert user.user_id == '1234' assert user.sign_up_date == user.sign_up_date assert user.sign_up_method == 'Firebase' assert user.med_state == 'Mild' assert user.days_left == 1 assert user.role.name == 'USER' db.drop_all()