def test_remove_from_favourite_with_empty_list(self, user, test_adverts,
add_favourites):
Favourite.remove_from_favourite(user_id=user.id, adverts=[])
result_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(result_adverts) == list(expected_adverts)
assert len(result_adverts) == 3
def test_remove_from_favourite_when_object_not_in_favourites(
self, user, test_adverts):
advert = Advert.objects.filter(place="Dębe Wielkie")
Favourite.remove_from_favourite(user_id=user.id, adverts=advert)
result_adverts = Favourite.objects.select_related(
"adverts").values_list("adverts__place", "adverts__price",
"adverts__area")
assert list(result_adverts) == []
assert len(result_adverts) == 0
def test_remove_from_favourite(self, user, test_adverts, add_favourites):
advert = Advert.objects.filter(place="Dębe Wielkie")
Favourite.remove_from_favourite(user_id=user.id, adverts=advert)
result_adverts = Favourite.objects.select_related(
"adverts").values_list("adverts__place", "adverts__price",
"adverts__area")
expected_adverts = test_adverts.exclude(
place="Dębe Wielkie").values_list("place", "price", "area")
assert list(result_adverts) == list(expected_adverts)
assert len(result_adverts) == 2