def setUp(self, plugin_name='linux'):
plugins_to_load = [
'example_plugin', 'linux', 'linux_network', 'subscription_plugin',
'virtual_machine_plugin', 'alert_plugin'
]
assert plugin_name in plugins_to_load
super(ResourceManagerTestCase, self).setUp()
self.host = self._create_host(fqdn='myaddress.mycompany.com',
nodename='myaddress.mycompany.com',
address='myaddress.mycompany.com')
self.manager = load_plugins(plugins_to_load)
mock.patch(
'chroma_core.lib.storage_plugin.manager.storage_plugin_manager',
self.manager).start()
self.resource_manager = ResourceManager()
# Mock out the plugin queue otherwise we get all sorts of issues in ci. We really shouldn't
# need all that ampq stuff just for the unit tests.
mock.patch('chroma_core.services.queue.AgentRxQueue').start()
self.plugin = AgentPluginHandlerCollection(
self.resource_manager
).handlers[plugin_name]._create_plugin_instance(self.host)
self.addCleanup(mock.patch.stopall)
def setUp(self):
super(TestLoad, self).setUp()
import loadable_plugin
self.loadable_plugin = loadable_plugin
self.manager = load_plugins(['loadable_plugin'])
self.assertEquals(self.manager.get_errored_plugins(), [])
import chroma_core
self.old_manager = chroma_core.lib.storage_plugin.manager.storage_plugin_manager
chroma_core.lib.storage_plugin.manager.storage_plugin_manager = self.manager
def setUp(self):
super(TestStorageResourceResource, self).setUp()
self.manager = load_plugins(['loadable_plugin'])
self.assertEquals(self.manager.get_errored_plugins(), [])
import chroma_core
self.old_manager = chroma_core.lib.storage_plugin.manager.storage_plugin_manager
chroma_core.lib.storage_plugin.manager.storage_plugin_manager = self.manager
# Re-initialize queryset to pick up loaded plugins
from chroma_api.storage_resource import StorageResourceResource, filter_class_ids
from chroma_core.models import StorageResourceRecord
StorageResourceResource._meta.queryset = StorageResourceRecord.objects.filter(resource_class__id__in = filter_class_ids())
def setUp(self):
super(TestReferenceAttribute, self).setUp()
import chroma_core.lib.storage_plugin.manager
self.original_mgr = chroma_core.lib.storage_plugin.manager.storage_plugin_manager
mgr = load_plugins(['loadable_plugin'])
chroma_core.lib.storage_plugin.manager.storage_plugin_manager = mgr
from chroma_core.models import StorageResourceRecord
resource_class, resource_class_id = mgr.get_plugin_resource_class('loadable_plugin', 'TestScannableResource')
record, created = StorageResourceRecord.get_or_create_root(resource_class, resource_class_id, {'name': 'foobar'})
self.record_pk = record.pk
self.manager = mgr
def test_load(self):
"""Test that the thousand_drives plugin used for stats load testing loads"""
manager = load_plugins(['thousand_drives'])
self.assertEquals(manager.get_errored_plugins(), [])
def test_load(self):
"""Test that the example plugin used in documentation loads"""
manager = load_plugins(['linux', 'example_plugin'])
self.assertEquals(manager.get_errored_plugins(), [])
def test_submodule(self):
# Check we can load a plugin via a dotted reference
mgr = load_plugins(['submodule.loadable_submodule_plugin'])
self.assertEquals(mgr.get_errored_plugins(), [])
# Check that the path has been stripped from the reference
mgr.get_plugin_class('loadable_submodule_plugin')
def test_dupemodule(self):
manager = load_plugins(['loadable_plugin', 'loadable_plugin'])
self.assertEqual(manager.get_errored_plugins(), ['loadable_plugin'])
def test_2classes(self):
manager = load_plugins(['unloadable_plugin_2classes'])
self.assertEqual(manager.get_errored_plugins(),
['unloadable_plugin_2classes'])
def test_0classes(self):
manager = load_plugins(["unloadable_plugin_0classes"])
self.assertEqual(manager.get_errored_plugins(),
["unloadable_plugin_0classes"])
