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()