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