def test_cannot_register_for_unknown_phase(self):
plugin = team.PluginDescriptor("unittest-plugin")
hook = InstallHookHandlerTests.UnitTestHook(phase="this_is_an_unknown_install_phase")
handler = provisioner.InstallHookHandler(plugin, loader_class=InstallHookHandlerTests.UnitTestComponentLoader)
handler.loader.registration_function = hook
with self.assertRaises(exceptions.SystemSetupError) as ctx:
handler.load()
self.assertEqual("Provisioning phase [this_is_an_unknown_install_phase] is unknown. Valid phases are: ['post_install'].",
ctx.exception.args[0])
def test_loads_module(self):
plugin = team.PluginDescriptor("unittest-plugin")
hook = InstallHookHandlerTests.UnitTestHook()
handler = provisioner.InstallHookHandler(plugin, loader_class=InstallHookHandlerTests.UnitTestComponentLoader)
handler.loader.registration_function = hook
handler.load()
handler.invoke("post_install", {"increment": 4})
# we registered our hook twice. Check that it has been called twice.
self.assertEqual(hook.call_counter, 2 * 4)