def test_get_product(self):
order = OrderModel.get(1)
assert order != None
orders = OrderModel.list()
assert len(orders) == 1
assert type(orders) == type([])
def put(self):
try:
data = request.get_json()
item = OrderModel.get(data['id'])
status = Status().name_to_enum(data['status'])
item.update()
return "success", 201
except:
return "error", 401
def get(self):
current_user = jwt.decode(request.headers['Authorization'],
environ.get('JWT_SECRET_KEY'),
options={'verify_exp': False})
if Roles().enum_to_name(current_user['sub']['role']) == 'admin':
if 'id' in request.args:
item = OrderModel.get(request.args['id'])
item = serialize_model(item)
item['status'] = Status().enum_to_name(item['status'])
return item
elif 'id_user' in request.args:
itens = OrderModel.list_by_user(request.args['id_user'])
itens = serialize_model_list(itens)
for item in itens:
item['status'] = Status().enum_to_name(item['status'])
return itens
list = OrderModel.list()
itens = serialize_model_list(list)
for item in itens:
item['status'] = Status().enum_to_name(item['status'])
return itens
else:
if 'id' in request.args:
item = OrderModel.get(request.args['id'])
if item.id_user == current_user['sub']['id']:
item = serialize_model(item)
item['status'] = Status().enum_to_name(item['status'])
return item
else:
return "You don't have access", 401
else:
itens = OrderModel.list_by_user(current_user['sub']['id'])
itens = serialize_model_list(itens)
for item in itens:
item['status'] = Status().enum_to_name(item['status'])
return itens
def test_update_order(self):
order = OrderModel.get(2)
order.status = 2
order.update()
assert order.status == 2
