def test_emit_location(self, mock_producer): client = StreamClient(kafka_addr='addr', kafka_topic='topic') resp = client.emit_location(self.mock_location) self.assertTrue(mock_producer().send.calledWith( 'topic', self.mock_location)) self.assertIsNone(resp)
def __init__(self, app): """ Create an application context object, :param app: Flask application instance. """ self.config = ConfigClient() self.cache = CacheClient( addr=self.config.get_value('redis.addr'), prefix='orion', ) self.db = DbClient( app, user=self.config.get_value('database.user'), password=self.config.get_value('database.password'), host=self.config.get_value('database.host'), port=self.config.get_value('database.port'), name=self.config.get_value('database.name'), ) self.geocode = ReverseGeocodingClient( mapbox_access_token=self.config.get_value('mapbox_access_token'), ) self.metrics_event = EventMetricsClient( addr=self.config.get_value('statsd.addr'), prefix='orion', ) self.metrics_latency = LatencyMetricsClient( addr=self.config.get_value('statsd.addr'), prefix='orion', ) self.stream = StreamClient( kafka_addr=self.config.get_value('kafka.addr'), kafka_topic=self.config.get_value('kafka.topic'), )
def __init__(self, app): """ Create an application context object, :param app: Flask application instance. """ self.config = ConfigClient() self.db = DbClient( app, user=self.config.get_value('database.user'), password=self.config.get_value('database.password'), host=self.config.get_value('database.host'), port=self.config.get_value('database.port'), name=self.config.get_value('database.name'), ) self.geocode = ReverseGeocodingClient( google_api_key=self.config.get_value('google_api_key'), ) self.stream = StreamClient( kafka_addr=self.config.get_value('kafka.addr'), kafka_topic=self.config.get_value('kafka.topic'), )
def test_init_noop_producer(self, mock_producer): client = StreamClient(kafka_addr=None, kafka_topic='topic') self.assertFalse(mock_producer.called) self.assertEqual(client.topic, 'topic')
def test_init_kafka_producer(self, mock_producer): client = StreamClient(kafka_addr='addr', kafka_topic='topic') self.assertTrue(mock_producer.called) self.assertEqual(client.topic, 'topic')