def test_get_vessel_fleet_with_callback(self):
def callback(a, b):
pass
spider = MarineTrafficSpider(show_fleet='',
getkey=self.api_key,
fleet_name='FLEET')
idx = -1
for idx, item in enumerate(spider.get_vessel_fleet(cb=callback)):
self.assertIn('/api/getfleet/{}'.format(self.api_key), item.url)
self.assertIs(item.callback, callback)
self.assertEqual(idx, 0)
def test_get_vessel_fleet(self):
api_key = '123456'
spider = MarineTrafficSpider(show_fleet='',
getkey=api_key,
fleet_name='FLEET',
fleet_id='xyz')
idx = -1
for idx, item in enumerate(spider.get_vessel_fleet()):
self.assertIn('/api/getfleet/{}/fleet_id:xyz'.format(self.api_key),
item.url)
self.assertIs(item.callback, spider.show_fleet)
self.assertEqual(idx, 0)