def test_quantidade_produtos(self):
loja = Loja('Lojão Tabajara')
loja.add_estoque('123', 15, 10)
loja.add_estoque('1234', 20, 5)
loja.add_estoque('12345', 20, -1)
loja.add_estoque('123456', 20, 0)
assert loja.quantidade_produtos('123') == 10
assert loja.quantidade_produtos('1234') == 5
assert loja.quantidade_produtos('12345') == 0
assert loja.quantidade_produtos('123456') == 0
def test_devolver_carrinho(self):
loja = Loja('Lojão Tabajara')
loja.add_estoque('123', 15, 10)
loja.add_estoque('1234', 20, 5)
assert len(loja.estoque) == 15
cliente = Cliente('John Doe')
pedido = Pedido(cliente)
pedido.add_item(loja.comprar('1234'))
pedido.add_item(loja.comprar('123'))
assert len(pedido.itens) == 2
assert len(loja.estoque) == 13
assert loja.quantidade_produtos('1234') == 4
assert loja.quantidade_produtos('123') == 9
loja.devolver_carrinho(pedido)
assert len(pedido.itens) == 0
assert len(loja.estoque) == 15
