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)