def __init__(self): self.products = Product.get_all() self.heads = [ "name", "material", "acabado", "cristal" ] self.data = self.get_empty_data() self.add_data()
def test_should_not_save_changes_given_invalid_data(self): quote_request = QuoteRequest(self.quote) url = url_for('quote.edit', id=self.quote.id) quote_data = dict( customer_name="", email="test.email.com", address="New address", name1="", material1="New Material", acabado1="", cristal1="New Cristal", name="Test Two", material="", acabado="Acabado two", cristal="Cristal two" ) with self.request_context(url, quote_data): quote_request.edit() self.db.session.rollback() self.assertEqual(self.customer.name, "Test") self.assertEqual(self.product.name, "Test") self.assertNotEqual(self.quote.address, "New address") self.assertEqual(len(Product.get_all()), 1)
def test_should_return_all_products(self): products = Product.get_all() self.assertEqual(products, [self.product, self.product2])