def test_wrap_module(self):
"""??wrap_module???
"""
plugin_module = imp.new_module("test_plugin")
plugin_module.execute = lambda ctx, arg: "execute {}".format(arg)
plugin = Plugin.wrap_module(plugin_module)
self.assertEquals(plugin.name, "test_plugin")
plugin.sys_prepare(None)
self.assertEquals(plugin.execute(None, "cat"), "execute cat")
plugin_module.execute = lambda: "execute"
self.failUnlessException(
InvalidPluginException,
Plugin.wrap_module,
plugin_module
)
评论列表
文章目录