def test_find_nearby(self):
stations = self.lib.find_nearby(Position(0, 0))
other = [
Station(self.lib, 8600626, "København H",
Position(55673063, 12565796)),
Station(self.lib, 10844, "Hovedbanegården, Tivoli",
Position(55672838, 12566191)),
Station(self.lib, 100240000,
"København, Bernstorffsgade (fjernbus)",
Position(55673899, 12565112))
]
self.assertEqual(other, stations)
def test_station_library_find_nearby(self):
self.assertEqual(
self.lib.find_nearby(Position(0, 0), 1000),
self.handler.handle(['1.0', 'Stations', 'findNearby'], {
'long': '0',
'lat': 0,
'radius': 1000
}))
def setup_v1_0(self):
"""
Sets up v1.0 of the API
:return:
"""
handler = self.add_handler('1.0')
site_library_handler = handler.add_handler('Stations')
site_library_handler.add_function(
'findNearby', lambda name, inp: self.library.find_nearby(
Position(int(inp['lat']), int(inp['long'])), int(inp['radius'])
))
site_library_handler \
.add_handler('departures') \
.add_function('*', lambda name, inp: self.library.station_from_id(int(name[0])).departures())
def setUp(self):
self.query_strategy = StubQueryStrategy()
self.lib = StationLibrary(self.query_strategy)
self.station = self.lib.find_nearby(Position(0, 0))[0]
self.query_strategy.called = []
def test_station_from_id_is_empty(self):
self.assertEqual(Station(self.lib, 123, '', Position(0, 0)),
self.lib.station_from_id(123))
def test_strategy_find_called_with_right_parameters(self):
self.lib.find_nearby(Position(1, 2))
self.assertEqual([('find_nearby', [1, 2, 100, 50])],
self.query_strategy.called)
def setUp(self):
self.query_strategy = StubQueryStrategy()
self.lib = StationLibrary(self.query_strategy)
self.station = self.lib.find_nearby(Position(0, 0))[0]
self.handler = RESTHandler(self.query_strategy)