def test_can_purchase_product_from_stock(self): """Test that we can purchase a product from stock. """ gift_repo = Product.get_gift_repository(self.db_conn) product = gift_repo[0] original_stock_level = product.in_stock_quantity self.assertEqual(product.in_stock_quantity, original_stock_level) product.purchase() self.assertEqual(product.in_stock_quantity, original_stock_level - 1)
def test_wedding_list_is_list(self): """Test core wedding list functionality. """ # List of wedding gift items. example_product = Product.get_gift_repository(self.db_conn)[0] example_gift = GiftItem.get_new_gift_item(self.db_conn, example_product) wedding_list = WeddingList() # Add a gift to the list: wedding_list.append(example_gift) self.assertEqual(wedding_list, [example_gift]) # Remove gift to the list: wedding_list.remove(example_gift) self.assertEqual(wedding_list, [])