def test_add_user(self):
name = 'user_name'
password = '******'
level = 'normal'
user = add_user(name, password, level)
assert add_user(name, password, level) is None
assert len(get_user(key=user.key)) == 1
user = get_user()[0]
assert user.name == 'user_name'
assert user.password == _encrypt('user_name', 'user_password')
assert check_user_password(user, 'user_password')
assert user.level == 'normal'
def test_update_user(self):
name = 'user_name'
password = '******'
level = 'normal'
add_user(name, password, level)
user = get_user()[0]
update_user(user, password='******', level='disable')
user = get_user()[0]
assert user.name == 'user_name'
assert user.password == _encrypt(user.name, 'new_password')
assert check_user_password(user, 'new_password')
assert user.level == 'disable'
try:
update_user(user, name='some_name')
except AttributeError:
pass
try:
update_user(user, level='some_level')
except AssertionError:
pass
