def test_consumer_reset(self, mock_get_rpc_server):
mock_server = mock.Mock()
mock_get_rpc_server.return_value = mock_server
consumer = octavia_messaging_consumer.OctaviaConsumer(self.driver)
consumer.start()
consumer.reset()
mock_server.reset.assert_called_once_with()
def test_consumer_graceful_stop(self, mock_get_rpc_server):
mock_server = mock.Mock()
mock_get_rpc_server.return_value = mock_server
consumer = octavia_messaging_consumer.OctaviaConsumer(self.driver)
consumer.start()
consumer.stop(graceful=True)
mock_server.stop.assert_called_once_with()
mock_server.wait.assert_called_once_with()
def set_db_mocks(self):
TOPIC = 'neutron_lbaas_event'
self.addCleanup(cfg.CONF.clear_override, 'event_stream_topic',
group='oslo_messaging')
cfg.CONF.set_override('event_stream_topic', TOPIC,
group='oslo_messaging')
self.payload = {'operating_status': 'ONLINE'}
self.consumer = octavia_messaging_consumer.OctaviaConsumer(
self.driver)
def test_set_consumer_topic(self):
TOPIC = 'neutron_lbaas_event'
self.addCleanup(cfg.CONF.clear_override, 'event_stream_topic',
group='oslo_messaging')
cfg.CONF.set_override('event_stream_topic', TOPIC,
group='oslo_messaging')
consumer = octavia_messaging_consumer.OctaviaConsumer(self.driver)
self.assertIsNotNone(consumer.transport)
self.assertEqual(TOPIC, consumer.target.topic)
self.assertEqual(cfg.CONF.host, consumer.target.server)
def test_consumer_start(self, mock_get_rpc_server):
mock_server = mock.Mock()
mock_get_rpc_server.return_value = mock_server
TOPIC = 'neutron_lbaas_event'
self.addCleanup(cfg.CONF.clear_override, 'event_stream_topic',
group='oslo_messaging')
cfg.CONF.set_override('event_stream_topic', TOPIC,
group='oslo_messaging')
consumer = octavia_messaging_consumer.OctaviaConsumer(self.driver)
consumer.start()
mock_get_rpc_server.assert_called_once_with(
consumer.transport, consumer.target, consumer.endpoints,
executor='eventlet'
)
mock_server.start.assert_called_once_with()
