def test_register_query_module(self):
querier = loader.register_query_module(MOCK_RUNNER_NAME)
self.assertIsNotNone(querier)
self.assertEqual(MOCK_RUNNER_NAME, querier.__name__)
self.assertIn(MOCK_RUNNER_NAME, loader.QUERIER_MODULES_CACHE)
self.assertEqual(querier, loader.QUERIER_MODULES_CACHE[MOCK_RUNNER_NAME])
def test_register_query_module(self):
querier = loader.register_query_module(MOCK_RUNNER_NAME)
self.assertIsNotNone(querier)
self.assertEqual(MOCK_RUNNER_NAME, querier.__name__)
self.assertIn(MOCK_RUNNER_NAME, loader.QUERIER_MODULES_CACHE)
self.assertEqual(querier, loader.QUERIER_MODULES_CACHE[MOCK_RUNNER_NAME])
def test_register_query_module(self):
querier = loader.register_query_module('mock_runner')
self.assertIsNotNone(querier)
self.assertEqual('mock_runner', querier.__name__)
self.assertIn('mock_runner', loader.QUERIER_MODULES_CACHE)
self.assertEqual(querier, loader.QUERIER_MODULES_CACHE['mock_runner'])
def test_register_query_module_again(self):
querier1 = loader.register_query_module(MOCK_RUNNER_NAME)
self.assertEqual(1, imp.load_source.call_count)
self.assertIsNotNone(querier1)
self.assertEqual(MOCK_RUNNER_NAME, querier1.__name__)
self.assertIn(MOCK_RUNNER_NAME, loader.QUERIER_MODULES_CACHE)
self.assertEqual(querier1, loader.QUERIER_MODULES_CACHE[MOCK_RUNNER_NAME])
querier2 = loader.register_query_module(MOCK_RUNNER_NAME)
self.assertEqual(1, imp.load_source.call_count)
self.assertEqual(querier1, querier2)
self.assertIsNotNone(querier2)
self.assertEqual(MOCK_RUNNER_NAME, querier2.__name__)
self.assertIn(MOCK_RUNNER_NAME, loader.QUERIER_MODULES_CACHE)
self.assertEqual(querier2, loader.QUERIER_MODULES_CACHE[MOCK_RUNNER_NAME])
def test_register_query_module_again(self):
querier1 = loader.register_query_module(MOCK_RUNNER_NAME)
self.assertEqual(1, imp.load_source.call_count)
self.assertIsNotNone(querier1)
self.assertEqual(MOCK_RUNNER_NAME, querier1.__name__)
self.assertIn(MOCK_RUNNER_NAME, loader.QUERIER_MODULES_CACHE)
self.assertEqual(querier1, loader.QUERIER_MODULES_CACHE[MOCK_RUNNER_NAME])
querier2 = loader.register_query_module(MOCK_RUNNER_NAME)
self.assertEqual(1, imp.load_source.call_count)
self.assertEqual(querier1, querier2)
self.assertIsNotNone(querier2)
self.assertEqual(MOCK_RUNNER_NAME, querier2.__name__)
self.assertIn(MOCK_RUNNER_NAME, loader.QUERIER_MODULES_CACHE)
self.assertEqual(querier2, loader.QUERIER_MODULES_CACHE[MOCK_RUNNER_NAME])
def test_register_query_module_again(self):
querier1 = loader.register_query_module('mock_runner')
self.assertEqual(1, imp.load_source.call_count)
self.assertIsNotNone(querier1)
self.assertEqual('mock_runner', querier1.__name__)
self.assertIn('mock_runner', loader.QUERIER_MODULES_CACHE)
self.assertEqual(querier1, loader.QUERIER_MODULES_CACHE['mock_runner'])
querier2 = loader.register_query_module('mock_runner')
self.assertEqual(1, imp.load_source.call_count)
self.assertEqual(querier1, querier2)
self.assertIsNotNone(querier2)
self.assertEqual('mock_runner', querier2.__name__)
self.assertIn('mock_runner', loader.QUERIER_MODULES_CACHE)
self.assertEqual(querier2, loader.QUERIER_MODULES_CACHE['mock_runner'])
def setUpClass(cls):
super(MistralQuerierTest, cls).setUpClass()
# Override the retry configuration here otherwise st2tests.config.parse_args
# in DbTestCase.setUpClass will reset these overrides.
cfg.CONF.set_override('retry_exp_msec', 100, group='mistral')
cfg.CONF.set_override('retry_exp_max_msec', 200, group='mistral')
cfg.CONF.set_override('retry_stop_max_msec', 200, group='mistral')
# Register query module.
cls.query_module = loader.register_query_module('mistral_v2')
def setUpClass(cls):
super(MistralQuerierTest, cls).setUpClass()
# Override the retry configuration here otherwise st2tests.config.parse_args
# in DbTestCase.setUpClass will reset these overrides.
cfg.CONF.set_override('retry_exp_msec', 100, group='mistral')
cfg.CONF.set_override('retry_exp_max_msec', 200, group='mistral')
cfg.CONF.set_override('retry_stop_max_msec', 200, group='mistral')
# Register query module.
cls.query_module = loader.register_query_module('mistral_v2')
def get_querier(self, query_module_name):
if (query_module_name not in self._queriers and
query_module_name not in self._failed_imports):
try:
query_module = register_query_module(query_module_name)
except:
LOG.exception('Failed importing query module: %s', query_module_name)
self._failed_imports.add(query_module_name)
self._queriers[query_module_name] = None
else:
querier = query_module.get_instance()
self._queriers[query_module_name] = querier
self._query_threads.append(eventlet.spawn(querier.start))
return self._queriers[query_module_name]
def get_querier(self, query_module_name):
if (query_module_name not in self._queriers
and query_module_name not in self._failed_imports):
try:
query_module = register_query_module(query_module_name)
except:
LOG.exception('Failed importing query module: %s',
query_module_name)
self._failed_imports.add(query_module_name)
self._queriers[query_module_name] = None
else:
querier = query_module.get_instance()
self._queriers[query_module_name] = querier
self._query_threads.append(eventlet.spawn(querier.start))
return self._queriers[query_module_name]
