def test_return_data_hook_duplicate_key(self):
hooks = [
Hook({
"path": "stacker.tests.test_util.result_hook",
"data_key": "my_hook_results"
}),
Hook({
"path": "stacker.tests.test_util.result_hook",
"data_key": "my_hook_results"
})
]
with self.assertRaises(KeyError):
handle_hooks("result", hooks, "us-east-1", self.context)
def test_hook_failure(self):
hooks = [
Hook({"path": "stacker.tests.test_util.fail_hook",
"required": True})]
with self.assertRaises(SystemExit):
handle_hooks("fail", hooks, self.provider, self.context)
hooks = [{"path": "stacker.tests.test_util.exception_hook",
"required": True}]
with self.assertRaises(Exception):
handle_hooks("fail", hooks, self.provider, self.context)
hooks = [
Hook({"path": "stacker.tests.test_util.exception_hook",
"required": False})]
# Should pass
handle_hooks("ignore_exception", hooks, self.provider, self.context)
def test_return_data_hook(self):
hooks = [
Hook({
"path": "stacker.tests.test_util.result_hook",
"data_key": "my_hook_results"
}),
# Shouldn't return data
Hook({"path": "stacker.tests.test_util.context_hook"})
]
handle_hooks("result", hooks, "us-east-1", self.context)
self.assertEqual(self.context.hook_data["my_hook_results"]["foo"],
"bar")
# Verify only the first hook resulted in stored data
self.assertEqual(self.context.hook_data.keys(), ["my_hook_results"])
def test_context_provided_to_hook(self):
hooks = [
Hook({
"path": "stacker.tests.test_util.context_hook",
"required": True
})
]
handle_hooks("missing", hooks, "us-east-1", self.context)
def test_valid_hook(self):
hooks = [
Hook({"path": "stacker.tests.test_util.mock_hook",
"required": True})]
handle_hooks("missing", hooks, self.provider, self.context)
good = hook_queue.get_nowait()
self.assertEqual(good["provider"].region, "us-east-1")
with self.assertRaises(Queue.Empty):
hook_queue.get_nowait()
def test_valid_enabled_false_hook(self):
hooks = [
Hook({
"path": "stacker.tests.test_util.mock_hook",
"required": True,
"enabled": False
})
]
handle_hooks("missing", hooks, self.provider, self.context)
self.assertTrue(hook_queue.empty())
def test_default_required_hook(self):
hooks = [Hook({"path": "stacker.hooks.blah"})]
with self.assertRaises(AttributeError):
handle_hooks("missing", hooks, self.provider, self.context)
def test_missing_non_required_hook_method(self):
hooks = [Hook({"path": "stacker.hooks.blah", "required": False})]
handle_hooks("missing", hooks, self.provider, self.context)
self.assertTrue(hook_queue.empty())
def test_missing_required_hook(self):
hooks = [Hook({"path": "not.a.real.path", "required": True})]
with self.assertRaises(ImportError):
handle_hooks("missing", hooks, self.provider, self.context)