class TestPluginNotRunning(TestPluginBase):
def setUp(self):
self.config = self.set_up_base_config()
self.p_base = None
def tearDown(self):
pass
def multithread_config_test(self, multithread_flag, threads_in_config,
threads_wanted):
self.config.set('base', 'threads', threads_in_config)
self.p_base = AnalysisBasePlugin(self,
self.config,
no_multithread=multithread_flag)
self.assertEqual(self.p_base.config[self.p_base.NAME]['threads'],
threads_wanted, 'number of threads not correct')
self.p_base.shutdown()
def test_no_multithread(self):
self.multithread_config_test(True, '4', '1')
def test_normal_multithread(self):
self.multithread_config_test(False, '2', '2')
def test_init_result_dict(self):
self.p_base = AnalysisBasePlugin(self, self.config)
resultdict = self.p_base.init_dict()
self.assertIn('analysis_date', resultdict, 'analysis date missing')
self.assertEqual(resultdict['plugin_version'], 'not set',
'plugin version field not correct')
self.p_base.shutdown()
class TestPluginBase(unittest.TestCase):
def setUp(self):
config = self.set_up_base_config()
self.base_plugin = AnalysisBasePlugin(self, config)
@staticmethod
def set_up_base_config():
config = ConfigParser()
config.add_section('base')
config.set('base', 'threads', '2')
config.add_section('ExpertSettings')
config.set('ExpertSettings', 'block_delay', '0.1')
return config
def tearDown(self):
self.base_plugin.shutdown()
gc.collect()
def register_plugin(self, name, plugin_object):
'''
This is a mock checking if the plugin registers correctly
'''
self.assertEqual(name, 'base', 'plugin registers with wrong name')
self.assertEqual(plugin_object.NAME, 'base',
'plugin object has wrong name')