Revision ff1c051b test/py/ganeti.hooks_unittest.py
| b/test/py/ganeti.hooks_unittest.py | ||
|---|---|---|
| 495 | 495 |
assert isinstance(self.lu, FakeNoHooksLU), "LU was replaced" |
| 496 | 496 |
|
| 497 | 497 |
|
| 498 |
class FakeEnvWithNodeNameLU(cmdlib.LogicalUnit):
|
|
| 498 |
class FakeEnvWithCustomPostHookNodesLU(cmdlib.LogicalUnit):
|
|
| 499 | 499 |
HPATH = "env_test_lu" |
| 500 | 500 |
HTYPE = constants.HTYPE_GROUP |
| 501 | 501 |
|
| ... | ... | |
| 506 | 506 |
return {}
|
| 507 | 507 |
|
| 508 | 508 |
def BuildHooksNodes(self): |
| 509 |
return (["a"], ["a"], ["explicit.node1.com", "explicit.node2.com"]) |
|
| 509 |
return (["a"], ["a"]) |
|
| 510 |
|
|
| 511 |
def PreparePostHookNodes(self, post_hook_node_uuids): |
|
| 512 |
return post_hook_node_uuids + ["b"] |
|
| 510 | 513 |
|
| 511 | 514 |
|
| 512 | 515 |
class TestHooksRunnerEnv(unittest.TestCase): |
| ... | ... | |
| 514 | 517 |
self._rpcs = [] |
| 515 | 518 |
|
| 516 | 519 |
self.op = opcodes.OpTestDummy(result=False, messages=[], fail=False) |
| 517 |
self.lu = FakeEnvWithNodeNameLU(FakeProc(), self.op, FakeContext(), None) |
|
| 520 |
self.lu = FakeEnvWithCustomPostHookNodesLU(FakeProc(), self.op, |
|
| 521 |
FakeContext(), None) |
|
| 518 | 522 |
|
| 519 | 523 |
def _HooksRpc(self, *args): |
| 520 | 524 |
self._rpcs.append(args) |
| ... | ... | |
| 526 | 530 |
hm.RunPhase(constants.HOOKS_PHASE_PRE) |
| 527 | 531 |
|
| 528 | 532 |
(node_list, hpath, phase, env) = self._rpcs.pop(0) |
| 529 |
self.assertEqual(node_list, set(["node_a.example.com"]))
|
|
| 533 |
self.assertEqual(node_list, set(["a"]))
|
|
| 530 | 534 |
|
| 531 | 535 |
# Check post-phase hook |
| 532 | 536 |
hm.RunPhase(constants.HOOKS_PHASE_POST) |
| 533 | 537 |
|
| 534 | 538 |
(node_list, hpath, phase, env) = self._rpcs.pop(0) |
| 535 |
self.assertEqual(node_list, set(["node_a.example.com", |
|
| 536 |
"explicit.node1.com", |
|
| 537 |
"explicit.node2.com"])) |
|
| 539 |
self.assertEqual(node_list, set(["a", "b"])) |
|
| 538 | 540 |
|
| 539 | 541 |
self.assertRaises(IndexError, self._rpcs.pop) |
| 540 | 542 |
|
Also available in: Unified diff