class Bike: def __init__(self): self.http_service = HttpService() self.producer = Producer() def get_bikes_station_information(self, url, params): res = self.http_service.get(url, params).json() for msg in res['data']['stations']: print('bikes_station_information ==> ', msg) self.producer.data_producer(BIKES_STATION_INFORMATION_TOPIC, msg) def get_bikes_station_status(self, url, params): res = self.http_service.get(url, params).json() for msg in res['data']['stations']: print('bikes_station_status ==> ', msg) self.producer.data_producer(BIKES_STATION_INFORMATION_TOPIC, msg)
def __init__(self): self.http_service = HttpService() self.producer = Producer()
def test_call_without_user_agent(self): s = HttpService('host', 42, 'path') s._call = Mock(return_value=(200, 'data', 3)) s.call('body') s._call.assert_called_with('body', {})
def test_call_with_user_agent(self): s = HttpService('host', 42, 'path', user_agent='test agent') s._call = Mock(return_value=(200, 'data', 3)) s.call('body') s._call.assert_called_with('body', {'User-Agent': 'test agent'})
def test_call_helper_not_implemented(self): s = HttpService() with self.assertRaises(NotImplementedError): s._call('whatever', {})