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())
