def test_get_beverage_update(self):
Beverage.update(1, "Suco de limão", "1 litro", 5)
beverage = Beverage.get_by_id(1)
self.assertEqual(beverage.name, "Suco de limão")
self.assertEqual(beverage.size, "1 litro")
self.assertEqual(beverage.price, 5)
def create_submit(request):
if request.POST:
name = request.POST.get('name')
size = request.POST.get('size')
value = request.POST.get('value')
if name:
if size:
if value:
beverage = Beverage.create(name, size, value)
if beverage is None:
messages.error(request, "Erro ao cadastrar a bebida")
else:
messages.error(request,
"Valor da bebida nao pode estar vazia")
else:
messages.error(request,
"Tamanho da bebida nao pode estar vazia")
else:
messages.error(request, "Nome nao pode estar vazio")
else:
messages.error(request, "Erro no rest")
return redirect('/beverage/create')
def edit_submit(request):
if request.POST:
id = request.POST.get('id')
name = request.POST.get('name')
size = request.POST.get('size')
price = request.POST.get('price')
if name or size or price:
beverage = Beverage.update(id, name, size, price)
if beverage is None:
messages.error(request, "Erro ao atualizar a bebida")
else:
if price is None:
messages.error(request, "preço nao pode estar vazio")
if size is None:
messages.error(request, "Lista de tamanho vazia")
if name is None:
messages.error(request, "Nome nao pode estar vazio")
return redirect('/beverage/edit/{}'.format(id))
else:
messages.error(request, "Erro no rest")
return redirect('/beverage/list')
def edit(request, id):
if id:
dados = {
'title': 'Editar bebida',
'header': 'Editar bebida',
'icon': 'fas fa-glass-martini',
'beverage': Beverage.get_by_id(id),
}
return render(request, 'beverage/create.html', dados)
else:
messages.error(request, "ID nao encontrado")
return redirect('/beverage/list')
def setUp(self):
ingredient = Ingredient.create("Pão")
Ingredient.set_value(ingredient, 1.25)
ingredient = Ingredient.create("Hamburger")
Ingredient.set_value(ingredient, 1)
ingredient = Ingredient.create("Bacon")
Ingredient.set_value(ingredient, 2)
name = "X-Bacon"
list = ["Pão", "Hamburger", "Bacon"]
profit = 5.5
sandwich = Sandwich.create(name, profit, list)
beverage = Beverage.create("Coca-cola", "2 Litros", 6)
def filter_submit(request):
if request.GET:
search = request.GET.get("search")
if search:
beverages = Beverage.filter(search)
if len(beverages) > 0:
return return_list(request, beverages)
else:
messages.error(request, 'Bebida não encontrada')
else:
messages.error(request, 'Busca esta vazia')
else:
messages.error(request, 'Erro durante a solicitação')
return redirect('/beverage/filter')
def list(request):
return return_list(request, Beverage.get_all())
def test_get_beverage_update_size(self):
Beverage.update_size(1, "3 litros")
beverage = Beverage.get_by_id(1)
self.assertEqual(beverage.size, "3 litros")
def test_get_beverage_update_name(self):
Beverage.update_name(1, "Suco de laranja")
beverage = Beverage.get_by_id(1)
self.assertEqual(beverage.name, "Suco de laranja")
def test_get_beverage_by_id(self):
beverage = Beverage.get_by_id(1)
self.assertEqual(beverage.name,"Coca-cola")
def test_filter_beverage_name(self):
beverages = Beverage.filter("Coca-cola")
self.assertNotEqual(len(beverages), 0)
def test_get_all_beverage(self):
beverage = Beverage.get_all()
self.assertIsNotNone(beverage)
def test_create_beverage(self):
beverage = Beverage.create("Suco de laranja", "1 Litro", 3.9)
self.assertEqual(beverage.id, 2)