def test_save_product(self): products = len(ProductModel.list()) new_product = ProductModel() new_product.name = "Produto 02" new_product.provider = ProviderModel.get_by_cnpj("04790618000153") new_product.quantity = 100 new_product.price = 10.0 new_product.available = True new_product.save() assert products + 1 == len(ProductModel.list())
def test_get_product(self): product = ProductModel.get(1) assert product.name == 'Produto 01' products = ProductModel.list() assert len(products) == 1 assert type(products) == type([])
def get(self): if 'id' in request.args: item = ProductModel.get(request.args['id']) item = serialize_model(item) return item elif 'id_provider' in request.args: itens = ProductModel.list_by_provider(request.args['id_provider']) itens = serialize_model_list(itens) return itens list = ProductModel.list() return serialize_model_list(list)
def first_order(): if UserModel.get_by_cpf("23666513840") == None: first_user() return if len(ProductModel.list()) == 0: first_product() return new_order = OrderModel() new_order.user = UserModel.get_by_cpf("23666513840") new_order.products = str({"id": 1, "quantity": 10}) new_order.save() return
def test_delete_product(self): products = len(ProductModel.list()) product = ProductModel.delete(2) assert products - 1 == len(ProductModel.list())