def test_add_to_favourite(self, user, test_adverts): Favourite.add_to_favourite(user_id=user.id, adverts=test_adverts) added_adverts = Favourite.objects.select_related( "adverts").values_list("adverts__place", "adverts__price", "adverts__area") expected_adverts = test_adverts.values_list("place", "price", "area") assert list(added_adverts) == list(expected_adverts)
def add_favourites(user, test_adverts): Favourite.add_to_favourite(user_id=user.id, adverts=test_adverts)
def test_add_to_favourite_with_empty_list(self, user): Favourite.add_to_favourite(user_id=user.id, adverts=[]) added_adverts = Favourite.objects.select_related( "adverts").values_list("adverts__place", "adverts__price", "adverts__area") assert list(added_adverts) == [(None, None, None)]