def test_user(self): item_shorts = Item(name='shorts', price=20, description='Some nice shorts') item_tshirt = Item(name='tshirt', price=15, description='A nice t-shirt') item_vest = Item(name='vest', price=15, description='A nice vest') user = User('Mike') self.assertIsInstance(user, User) cart5 = Cart() cart5.add_item(item_shorts, 2) cart5.add_item(item_tshirt, 3) cart5.add_item(item_vest, 1) user.add_cart(cart5) retrieved_cart5 = user.get_cart_by_id(cart_id=cart5.cart_id) self.assertIsInstance(retrieved_cart5, Cart) shorts = retrieved_cart5.get_item('shorts') self.assertEqual(shorts['num_items'], 2) cart6 = Cart() cart6.add_item(item_shorts, 1) cart6.add_item(item_tshirt, 2) cart6.add_item(item_vest, 3) user.add_cart(cart6) retrieved_cart6 = user.get_cart_by_id(cart_id=cart6.cart_id) self.assertIsInstance(retrieved_cart6, Cart) vest = retrieved_cart6.get_item('vest') self.assertEqual(vest['num_items'], 3) cart5_vest = retrieved_cart5.get_item('vest') self.assertEqual(cart5_vest['num_items'], 1) self.assertFalse(retrieved_cart5.get_item('pants'))
def main(): item_shorts = Item(name='shorts', price=10, description='Some nice shorts') item_tshirt = Item(name='tshirt', price=20, description='A nice t-shirt') item_pants = Item(name='pants', price=30, description='Lovely pants') # create user user = User('Eric') # create a new cart cart = Cart() user.add_cart(cart) # Eric does some shopping cart.add_item(item_shorts, 4) cart.add_item(item_tshirt, 2) cart.add_item(item_pants, 3) # we want to display his cart retrieved_cart = user.get_cart_by_id(cart_id=cart.cart_id) print(retrieved_cart) pants = retrieved_cart.get_item('pants') print(pants) cart2 = Cart() cart2.add_item(item_shorts, 1) cart2.add_item(item_tshirt, 2) cart2.add_item(item_pants, 3) user.add_cart(cart2) retrieved_cart2 = user.get_cart_by_id(cart_id=2) tshirt = retrieved_cart2.get_item('tshirt') print(tshirt) tshirt_total = cart2.get_item_total('tshirt') print(tshirt_total)