Skip to content

Commit 2e93623

Browse files
committed
Moct tests
1 parent b5b69d5 commit 2e93623

File tree

1 file changed

+10
-9
lines changed

1 file changed

+10
-9
lines changed

search/tests.py

+10-9
Original file line numberDiff line numberDiff line change
@@ -162,17 +162,18 @@ def test_dining_info(self, mock_get_places):
162162
assert len(dining_info) == 2
163163
assert dining_info[0]["name"] == "Restaurant A"
164164

165-
def test_outdoor_info(self):
165+
@patch('info.helpers.places.FourSquarePlacesHelper.get_places')
166+
def test_outdoor_info(self, mock_get_places):
166167
"""
167168
Tests that outdoor activities information can be retrieved
168169
"""
169-
city = "New York City"
170-
country = "US"
171-
172-
outdoor_info = FourSquarePlacesHelper().get_places(
173-
city=f"{city}, {country}", categories="16000", sort="RELEVANCE", limit=5
174-
)
175-
assert outdoor_info is not None and len(outdoor_info) > 0 # Ensure we got outdoor info
170+
mock_get_places.return_value = [
171+
{"name": "Outdoor A", "address": "Address A"},
172+
{"name": "Outdoor B", "address": "Address B"},
173+
]
174+
outdoor_info = FourSquarePlacesHelper().get_places(city="New York", categories="13065", limit=5)
175+
assert len(outdoor_info) == 2
176+
assert outdoor_info[0]["name"] == "Outdoor A"
176177

177178
@patch('info.helpers.places.FourSquarePlacesHelper.get_places')
178179
def test_airport_info(self, mock_get_places):
@@ -193,7 +194,7 @@ def test_arts_info(self, mock_get_places):
193194
art_info = FourSquarePlacesHelper().get_places(city="New York", categories="19040", limit=5)
194195
assert len(art_info) == 2
195196
assert art_info[0]["name"] == "Arts A"
196-
197+
197198
def test_city_photo(self):
198199
"""
199200
Tests that a city photo can be retrieved for New York City.

0 commit comments

Comments
 (0)