def test_client_model_delete(db_session, conf_logger): clients = Client.get_all() client = util.get_random_client() client.delete() assert client in clients assert client not in Client.get_all() assert Client.get_by_email(client.email) is None
def test_client_model(db_session, conf_logger): users_data = [util.get_random_user_data() for idx in range(8)] for user_data in users_data: user = Client(**user_data) user.set_password(user.password) user.save() for idx, user in enumerate(list(Client.get_all())): assert users_data[idx].get('email') == user.email assert users_data[idx].get('first_name') == user.first_name assert users_data[idx].get('last_name') == user.last_name assert UsersEnum.client == user.user_type
def on_get(self, req, resp): users = Client.get_all() schema = ClientSchema() schema.context['remove_fields'] = ['seed', 'password'] data, errors = schema.dump(users, many=True) if errors: logger.error(errors) raise falcon.HTTPInternalServerError() data = format_response(data) resp.status = falcon.HTTP_200 resp.body = json.dumps(data, ensure_ascii=False)
def get_random_client(): clients = Client.get_all() return random.choice(clients)