class TestABasket(TestCase): def setUp(self): self.basket = Basket() self.product = create_product() def test_an_empty_basket_has_zero_lines(self): self.assertEqual(0, self.basket.num_lines) def test_new_baskets_are_empty(self): self.assertTrue(self.basket.is_empty) def test_adding_product_creates_line(self): self.basket.add_product(self.product) self.assertEqual(1, self.basket.num_lines) def test_adding_multiproduct_line_returns_correct_number_of_items(self): self.basket.add_product(self.product, 10) self.assertEqual(self.basket.num_items, 10) self.assertEqual(self.basket.num_lines, 1) def test_add_product_creates_line(self): self.basket.add_product(self.product) self.assertTrue(self.basket.num_lines == 1) def test_add_product_sets_line_prices(self): self.basket.add_product(self.product) basket_line = self.basket.lines.all()[0] self.assertEqual(basket_line.price_incl_tax, D('10.00')) self.assertEqual(basket_line.price_excl_tax, D('10.00')) def test_flushing_basket_removes_all_lines(self): self.basket.add_product(self.product, 10) self.assertEqual(self.basket.num_items, 10) self.basket.flush() self.assertEqual(self.basket.num_items, 0) def test_returns_correct_quantity_for_missing_product(self): self.assertEqual(0, self.basket.line_quantity(self.product)) def test_returns_correct_quantity_for_existing_product(self): self.basket.add_product(self.product) self.assertEqual(1, self.basket.line_quantity(self.product)) def test_returns_correct_quantity_for_existing_product_with_options(self): option = Option.objects.create(name="Message") options = [{"option": option, "value": "2"}] self.basket.add_product(self.product, options=options) self.assertEqual(0, self.basket.line_quantity(self.product)) self.assertEqual(1, self.basket.line_quantity(self.product, options)) self.assertEqual(1, self.basket.product_quantity(self.product)) def test_has_method_to_test_if_submitted(self): self.basket.set_as_submitted() self.assertTrue(self.basket.is_submitted())