Example #1
0
  def test_relationships(self):
    food1 = FoodItem(name='food1',
    img_url="",
    nf_calories=Decimal(1),
    nf_total_fat=Decimal(2),
    nf_cholesterol=Decimal(3),
    nf_saturated_fat=Decimal(4),
    nf_sodium=Decimal(5),
    nf_total_carbohydrate=Decimal(6),
    nf_dietary_fiber=Decimal(7),
    nf_sugars=Decimal(8),
    nf_protein=Decimal(9))

    food2 = FoodItem(name='food2',
    img_url="",
    nf_calories=Decimal(11),
    nf_total_fat=Decimal(12),
    nf_cholesterol=Decimal(13),
    nf_saturated_fat=Decimal(14),
    nf_sodium=Decimal(15),
    nf_total_carbohydrate=Decimal(16),
    nf_dietary_fiber=Decimal(17),
    nf_sugars=Decimal(18),
    nf_protein=Decimal(19))

    cart1 = Cart()
    cart2 = Cart()

    user1 = User(email="*****@*****.**",username="******",password="******")
    user2 = User(email="*****@*****.**",username="******",password="******")

    food1.cart = cart1
    cart1.user = user1
    food2.cart = cart2
    cart2.user = user2

    # check relationships
    self.assertTrue(food1.cart == cart1)
    self.assertTrue(cart1.user == user1)
    self.assertTrue(food2.cart == cart2)
    self.assertTrue(cart2.user == user2)

    # test relationships are showed in __repr__
    self.assertTrue(food1.__repr__() == 'food1 in Cart of <User one>')
    self.assertTrue(cart1.__repr__() == 'Cart of <User one>')
    self.assertTrue(user1.__repr__() == '<User one>')