def test_add_item(self): sale = self.create_sale() delivery = Delivery(store=self.store) service = self.create_service() service_item = sale.add_sellable(service.sellable, quantity=1, price=10) self.assertFalse(service_item in list(delivery.delivery_items)) self.assertFalse(service_item in delivery.get_items()) delivery.add_item(service_item) self.assertTrue(service_item in list(delivery.delivery_items)) self.assertTrue(service_item in delivery.get_items()) product = self.create_product() product_item = sale.add_sellable(product.sellable, quantity=1, price=10) self.assertFalse(product_item in list(delivery.delivery_items)) self.assertFalse(product_item in delivery.get_items()) delivery.add_item(product_item) self.assertTrue(product_item in list(delivery.delivery_items)) self.assertTrue(product_item in delivery.get_items())