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()
