def test_deferred_hooks(self):
"""
Test that our plugins can return deferreds as part of their hooks and
that we run the plugin chain correctly based on the result of these
deferreds
"""
core.load("gyro.impl.twisted.core.TwistedCore")
class FirstPlugin(plugin.Plugin):
priority = 0
def on_first(self, arg):
return core.defer_later(0.01, lambda: "first" + arg)
def on_second(self, arg):
return core.defer_later(0.01, lambda: None)
class SecondPlugin(plugin.Plugin):
priority = 0
def on_first(self, arg):
raise AssertionError("Should not be called")
def on_second(self, arg):
return core.defer_later(0.01, lambda: "second" + arg)
plugin.enable_plugin("FirstPlugin")
plugin.enable_plugin("SecondPlugin")
self.assertEquals(len(plugin.get_active_plugins()), 2)
r = yield plugin.run_hook("on_first", "result")
self.assertEquals(r, "firstresult")
r = yield plugin.run_hook("on_second", "result")
self.assertEquals(r, "secondresult")
def test_run_hooks(self):
"""
Check that we execute hooks correctly according to priority and that
the execution chain is correct based on return result. Note that this
only tests synchronous hooks
"""
core.load("gyro.impl.twisted.core.TwistedCore")
class HighPriorityPlugin(plugin.Plugin):
priority = 0
def on_high(self, arg):
return "high" + arg
def on_medium(self, arg):
return None
def on_low(self, arg):
return None
class MediumPriorityPlugin(plugin.Plugin):
priority = 50
def on_high(self, arg):
return "medium" + arg
def on_medium(self, arg):
return "medium" + arg
def on_low(self, arg):
return None
class LowerPriorityPlugin(plugin.Plugin):
priority = 100
def on_high(self, arg):
return "low" + arg
def on_medium(self, arg):
return "low" + arg
def on_low(self, arg):
return "low" + arg
self.failUnless(HighPriorityPlugin in plugin.Plugin.get_plugins())
self.failUnless(MediumPriorityPlugin in plugin.Plugin.get_plugins())
self.failUnless(LowerPriorityPlugin in plugin.Plugin.get_plugins())
plugin.enable_plugin("HighPriorityPlugin")
plugin.enable_plugin("MediumPriorityPlugin")
plugin.enable_plugin("LowerPriorityPlugin")
self.assertEquals(len(plugin.get_active_plugins()), 3)
r = yield plugin.run_hook("on_high", "result")
self.assertEquals(r, "highresult")
r = yield plugin.run_hook("on_medium", "result")
self.assertEquals(r, "mediumresult")
r = yield plugin.run_hook("on_low", "result")
self.assertEquals(r, "lowresult")
plugin.disable_plugin("HighPriorityPlugin")
r = yield plugin.run_hook("on_high", "result")
self.assertEquals(r, "mediumresult")