Example #1
0
    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'))
Example #2
0
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)