def test_add_get_favorite(self):
"""
test adding a favorite for the user on FavoriteItem
no duplicates for favorites!
"""
favitem = FavoriteItem(name="Product 1")
favitem.save()
self.assertFalse(Favorites.objects.get_favorite(self.user, favitem))
fav = Favorites.objects.add_favorite(self.user, favitem)
fav_len = len(Favorites.objects.all())
self.assertTrue(fav_len == 1)
# should not duplicate
if not Favorites.objects.get_favorite(self.user, favitem):
termprint("WARNING", "Not found....adding")
fav_duplicate = Favorites.objects.add_favorite(self.user, favitem)
fav_len = len(Favorites.objects.all())
termprint("INFO", "%s rows found " % fav_len)
self.assertTrue(fav_len == 1)
favget = Favorites.objects.get_favorite(self.user, favitem)
self.assertTrue(favget)
# test the templat tag get
termprint("INFO", my_favorites(self.user))
self.assertTrue(my_favorites(self.user))
def test_remove_get_favorite(self):
"""
test adding a favorite for the user on FavoriteItem
no duplicates for favorites!
"""
favitem = FavoriteItem(name="Product 1")
favitem.save()
self.assertFalse(Favorites.objects.get_favorite(self.user, favitem))
fav = Favorites.objects.add_favorite(self.user, favitem)
fav_len = len(Favorites.objects.all())
self.assertTrue(fav_len == 1)
# delete it
fav_del = Favorites.objects.delete_favorite(self.user, favitem)
fav_len = len(Favorites.objects.all())
self.assertTrue(fav_len == 0)
termprint("INFO", my_favorites(self.user))
self.assertFalse(my_favorites(self.user))
