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)]