class TestPluginManager(unittest.TestCase):
@classmethod
def setupClass(cls):
cls.tmppath, cls.pythonpath = install_package()
import pkg_resources
dist = list(pkg_resources.find_distributions(cls.pythonpath))[0]
pkg_resources.working_set.add(dist)
def setUp(self):
self.pm = PluginManager()
self.pm.debug = False
def test_autoload(self):
assert self.pm._plugin == {}
self.pm.plugins('test.c1')
assert 'test.c1' in self.pm._plugin
def test_dynamic_plugin(self):
self.pm.add_plugin('test.dynamic3', AlgoPlugin1)
self.pm.add_plugin('test.dynamic3', AlgoPlugin2)
assert len(self.pm.plugins('test.dynamic3')) == 2
objc3c1 = plugin_instance('test.dynamic3', 'AlgoPlugin1')
assert objc3c1
assert isinstance(objc3c1, Algo)
objc3c2 = plugin_instance('test.dynamic3', 'MyAlgoPlugin2')
assert objc3c2
assert isinstance(objc3c1, Algo)
def test_proxy_plugin(self):
from openalea.core.plugin.manager import SimpleClassPluginProxy
self.pm.add_plugin('test.dynamic4', Algo, plugin_proxy=SimpleClassPluginProxy)
objc4c1 = plugin_instance('test.dynamic4', 'Algo')
objc4c1_2 = plugin_instance('test.dynamic4', 'Algo')
objc4c1_3 = new_plugin_instance('test.dynamic4', 'Algo')
assert objc4c1
assert isinstance(objc4c1, Algo)
assert objc4c1 is objc4c1_2
assert objc4c1 is not objc4c1_3
def test_plugin_name(self):
import tstpkg1.plugin
import tstpkg1.impl
# Check manager use plugin name attribute if defined (instead of class name)
c1plugin1 = self.pm.plugin('test.c1', 'MyPlugin1')
assert c1plugin1 is tstpkg1.plugin.C1Plugin1
# Check there is no conflict if two plugins with same alias (in setup.py) but in different categories
# Check plugin name use class name if no attribute "name"
c2plugin1 = self.pm.plugin('test.c2', 'C2Plugin1')
assert c2plugin1 is tstpkg1.plugin.C2Plugin1
def test_new_instance(self):
import tstpkg1.impl
objc1c1 = plugin_instance('test.c1', 'MyPlugin1')
assert objc1c1
assert isinstance(objc1c1, tstpkg1.impl.C1Class1)
objc1c2 = plugin_instance('test.c1', 'MyPlugin2')
assert objc1c2
assert isinstance(objc1c2, tstpkg1.impl.C1Class2)
objc2c1 = plugin_instance('test.c2', 'C2Plugin1')
assert objc2c1
assert isinstance(objc2c1, tstpkg1.impl.C2Class1)
objc2c2 = plugin_instance('test.c2', 'C2Plugin2')
assert objc2c1
assert isinstance(objc2c2, tstpkg1.impl.C2Class2)
# Check instance return existing instance
objc1c1_2 = plugin_instance('test.c1', 'MyPlugin1')
assert objc1c1 is objc1c1_2
assert plugin_instances('test.c1', 'MyPlugin1')[0] is objc1c1_2
def test_multiple_instance(self):
# Assert weakref work correctly and plugin instances have been lost
assert not plugin_instances('test.c1', 'MyPlugin1')
objc1c1_0 = plugin_instance('test.c1', 'MyPlugin1')
objc1c1_1 = new_plugin_instance('test.c1', 'MyPlugin1')
objc1c1_2 = new_plugin_instance('test.c1', 'MyPlugin1')
# Assert all object have been created correctly
assert objc1c1_0
assert objc1c1_1
assert objc1c1_2
# Assert all instances are different as we use new instead of instance
assert objc1c1_0 is not objc1c1_1
assert objc1c1_1 is not objc1c1_2
objc1c2 = new_plugin_instance('test.c1', 'MyPlugin2')
assert len(plugin_instances('test.c1', 'MyPlugin1')) == 3
assert len(plugin_instances('test.c1')) == 4
del objc1c1_2
assert len(plugin_instances('test.c1', 'MyPlugin1')) == 2
assert len(plugin_instances('test.c1')) == 3
del objc1c2
assert len(plugin_instances('test.c1', 'MyPlugin1')) == 2
assert len(plugin_instances('test.c1', 'MyPlugin2')) == 0
assert len(plugin_instances('test.c1')) == 2
def test_debug_ep_load(self):
clear_plugin_instances()
self.pm.debug = True
with self.assertRaises(ImportError):
self.pm.plugins('test.err1')
self.pm.clear()
self.pm.debug = False
self.pm.plugins('test.err1')
def test_plugin_proxy(self):
from openalea.core.plugin.manager import SimpleClassPluginProxy
pm = PluginManager()
pm.set_proxy('oalab.modelclass', SimpleClassPluginProxy)
clear_plugin_instances()
w1 = plugin_instance('oalab.modelclass', 'PythonModel')
w2 = plugin_instance('oalab.modelclass', 'PythonModel')
w3 = plugin_instance('oalab.modelclass', 'PythonModel')
w4 = new_plugin_instance('oalab.modelclass', 'PythonModel')
assert w1
assert w1 is w2 is w3
assert w1 is not w4
assert len(plugin_instances('oalab.modelclass', 'PythonModel')) == 2
@classmethod
def tearDownClass(cls):
cls.tmppath.rmtree()
def menu_actions(self):
return [self.menu_edit, self.action_menu_1]
class TestAppletPlugin(object):
name = 'TestApplet'
alias = 'Test Applet'
def __call__(self):
return TestApplet
from openalea.core.plugin.manager import PluginManager
pm = PluginManager()
pm.discover('oalab.applet')
pm.add_plugin('oalab.applet', TestAppletPlugin)
if __name__ == '__main__':
instance = QtGui.QApplication.instance()
if instance is None:
app = QtGui.QApplication([])
else:
app = instance
tissuelab_conf = {
'children': {0: [1, 2], 2: [3, 4], 3: [5, 6], 4: [7, 8], 7: [11, 12], 8: [9, 10]},
'parents': {0: None, 1: 0, 2: 0, 3: 2, 4: 2, 5: 3, 6: 3, 7: 4, 8: 4, 9: 8, 10: 8, 11: 7, 12: 7},
'properties': {
0: {'amount': 0.04774535809018567, 'splitDirection': 2},
class TestPluginManager(unittest.TestCase):
@classmethod
def setupClass(cls):
cls.tmppath, cls.pythonpath = install_package()
import pkg_resources
dist = list(pkg_resources.find_distributions(cls.pythonpath))[0]
pkg_resources.working_set.add(dist)
def setUp(self):
self.pm = PluginManager()
self.pm.debug = False
def test_autoload(self):
assert self.pm._plugin == {}
self.pm.plugins('test.c1')
assert 'test.c1' in self.pm._plugin
def test_dynamic_plugin(self):
self.pm.add_plugin('test.dynamic3', AlgoPlugin1)
self.pm.add_plugin('test.dynamic3', AlgoPlugin2)
assert len(self.pm.plugins('test.dynamic3')) == 2
objc3c1 = plugin_instance('test.dynamic3', 'AlgoPlugin1')
assert objc3c1
assert isinstance(objc3c1, Algo)
objc3c2 = plugin_instance('test.dynamic3', 'MyAlgoPlugin2')
assert objc3c2
assert isinstance(objc3c1, Algo)
def test_proxy_plugin(self):
from openalea.core.plugin.manager import SimpleClassPluginProxy
self.pm.add_plugin('test.dynamic4',
Algo,
plugin_proxy=SimpleClassPluginProxy)
objc4c1 = plugin_instance('test.dynamic4', 'Algo')
objc4c1_2 = plugin_instance('test.dynamic4', 'Algo')
objc4c1_3 = new_plugin_instance('test.dynamic4', 'Algo')
assert objc4c1
assert isinstance(objc4c1, Algo)
assert objc4c1 is objc4c1_2
assert objc4c1 is not objc4c1_3
def test_plugin_name(self):
import tstpkg1.plugin
import tstpkg1.impl
# Check manager use plugin name attribute if defined (instead of class name)
c1plugin1 = self.pm.plugin('test.c1', 'MyPlugin1')
assert c1plugin1 is tstpkg1.plugin.C1Plugin1
# Check there is no conflict if two plugins with same alias (in setup.py) but in different categories
# Check plugin name use class name if no attribute "name"
c2plugin1 = self.pm.plugin('test.c2', 'C2Plugin1')
assert c2plugin1 is tstpkg1.plugin.C2Plugin1
def test_new_instance(self):
import tstpkg1.impl
objc1c1 = plugin_instance('test.c1', 'MyPlugin1')
assert objc1c1
assert isinstance(objc1c1, tstpkg1.impl.C1Class1)
objc1c2 = plugin_instance('test.c1', 'MyPlugin2')
assert objc1c2
assert isinstance(objc1c2, tstpkg1.impl.C1Class2)
objc2c1 = plugin_instance('test.c2', 'C2Plugin1')
assert objc2c1
assert isinstance(objc2c1, tstpkg1.impl.C2Class1)
objc2c2 = plugin_instance('test.c2', 'C2Plugin2')
assert objc2c1
assert isinstance(objc2c2, tstpkg1.impl.C2Class2)
# Check instance return existing instance
objc1c1_2 = plugin_instance('test.c1', 'MyPlugin1')
assert objc1c1 is objc1c1_2
assert plugin_instances('test.c1', 'MyPlugin1')[0] is objc1c1_2
def test_multiple_instance(self):
# Assert weakref work correctly and plugin instances have been lost
assert not plugin_instances('test.c1', 'MyPlugin1')
objc1c1_0 = plugin_instance('test.c1', 'MyPlugin1')
objc1c1_1 = new_plugin_instance('test.c1', 'MyPlugin1')
objc1c1_2 = new_plugin_instance('test.c1', 'MyPlugin1')
# Assert all object have been created correctly
assert objc1c1_0
assert objc1c1_1
assert objc1c1_2
# Assert all instances are different as we use new instead of instance
assert objc1c1_0 is not objc1c1_1
assert objc1c1_1 is not objc1c1_2
objc1c2 = new_plugin_instance('test.c1', 'MyPlugin2')
assert len(plugin_instances('test.c1', 'MyPlugin1')) == 3
assert len(plugin_instances('test.c1')) == 4
del objc1c1_2
assert len(plugin_instances('test.c1', 'MyPlugin1')) == 2
assert len(plugin_instances('test.c1')) == 3
del objc1c2
assert len(plugin_instances('test.c1', 'MyPlugin1')) == 2
assert len(plugin_instances('test.c1', 'MyPlugin2')) == 0
assert len(plugin_instances('test.c1')) == 2
def test_debug_ep_load(self):
clear_plugin_instances()
self.pm.debug = True
with self.assertRaises(ImportError):
self.pm.plugins('test.err1')
self.pm.clear()
self.pm.debug = False
self.pm.plugins('test.err1')
def test_plugin_proxy(self):
from openalea.core.plugin.manager import SimpleClassPluginProxy
pm = PluginManager()
pm.set_proxy('oalab.modelclass', SimpleClassPluginProxy)
clear_plugin_instances()
w1 = plugin_instance('oalab.modelclass', 'PythonModel')
w2 = plugin_instance('oalab.modelclass', 'PythonModel')
w3 = plugin_instance('oalab.modelclass', 'PythonModel')
w4 = new_plugin_instance('oalab.modelclass', 'PythonModel')
assert w1
assert w1 is w2 is w3
assert w1 is not w4
assert len(plugin_instances('oalab.modelclass', 'PythonModel')) == 2
@classmethod
def tearDownClass(cls):
cls.tmppath.rmtree()