def setUp(self): """ """ self.p1 = Product.objects.create(name="Product 1", slug="product-1", active=True) self.p2 = Product.objects.create(name="Product 2", slug="product-2", active=True) self.p3 = Product.objects.create(name="Product 3", slug="product-3", active=True) self.p4 = Product.objects.create(name="Product 4", slug="product-4", active=True) self.c1 = Category.objects.create(name="Category 1", slug="category-1") self.c1.save() self.c11 = Category.objects.create(name="Category 11", slug="category-11", parent=self.c1) self.c11.products = (self.p1, self.p2) self.c11.save() self.c12 = Category.objects.create(name="Category 12", slug="category-12", parent=self.c1) self.c12.products = (self.p3, self.p4) self.c12.save() address = Address.objects.create() self.o = Order.objects.create(invoice_address=address, shipping_address=address) self.oi1 = OrderItem.objects.create(order=self.o, product_amount=1, product=self.p1) self.oi2 = OrderItem.objects.create(order=self.o, product_amount=2, product=self.p2) self.oi3 = OrderItem.objects.create(order=self.o, product_amount=3, product=self.p3) self.oi4 = OrderItem.objects.create(order=self.o, product_amount=4, product=self.p4) calculate_product_sales()
def test_calculate_product_sales(self): calculate_product_sales() # This should not break calculate_product_sales() self.oi1.product = None self.oi1.save() calculate_product_sales()