def test_001_create_user(self):
admin_user = User.add_user("admin_user", "test", "admin")
staff_user = User.add_user('staff_user', 'test', 'staff')
user = User.objects(name='admin_user').first()
assert user.name == admin_user.name
assert user.group == 'admin'
def user_login(create_new=False):
data = request.get_json()
if create_new is True:
username = data.get('username')
password = data.get('password')
user_group = data.get('user_group')
if user_group not in ['admin', 'staff']:
user_group = 'staff'
user = User.add_user(username, password, user_group)
response = success_response({'user_created': True, '_id': user.id})
return jsonify(response)
else:
username = data.get('username')
password = data.get('password')
user = User.objects(name=username).first()
if user is None:
response = failure_response(400, 'username not found')
return jsonify(response)
else:
if check_password_hash(user.password, password) is True:
login_user(user)
response = success_response({
'login': '******',
'name': user.name,
'_id': user.id
})
return jsonify(response)
