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)