Revision 2395c322 test/ganeti.hooks_unittest.py
| b/test/ganeti.hooks_unittest.py | ||
|---|---|---|
| 36 | 36 |
from ganeti import cmdlib |
| 37 | 37 |
from ganeti.constants import HKR_SUCCESS, HKR_FAIL, HKR_SKIP |
| 38 | 38 |
|
| 39 |
from mocks import FakeConfig, FakeSStore |
|
| 39 |
from mocks import FakeConfig, FakeSStore, FakeProc
|
|
| 40 | 40 |
|
| 41 | 41 |
class FakeLU(cmdlib.LogicalUnit): |
| 42 | 42 |
HPATH = "test" |
| ... | ... | |
| 231 | 231 |
sstore = FakeSStore() |
| 232 | 232 |
op = opcodes.OpCode() |
| 233 | 233 |
lu = FakeLU(None, op, cfg, sstore) |
| 234 |
hm = mcpu.HooksMaster(self._call_false, lu) |
|
| 234 |
hm = mcpu.HooksMaster(self._call_false, FakeProc(), lu)
|
|
| 235 | 235 |
self.failUnlessRaises(errors.HooksFailure, |
| 236 | 236 |
hm.RunPhase, constants.HOOKS_PHASE_PRE) |
| 237 | 237 |
hm.RunPhase(constants.HOOKS_PHASE_POST) |
| 238 | 238 |
|
| 239 | 239 |
def testIndividualFalse(self): |
| 240 |
"""Test individual rpc failure"""
|
|
| 240 |
"""Test individual node failure"""
|
|
| 241 | 241 |
cfg = FakeConfig() |
| 242 | 242 |
sstore = FakeSStore() |
| 243 | 243 |
op = opcodes.OpCode() |
| 244 | 244 |
lu = FakeLU(None, op, cfg, sstore) |
| 245 |
hm = mcpu.HooksMaster(self._call_nodes_false, lu) |
|
| 246 |
self.failUnlessRaises(errors.HooksFailure, |
|
| 247 |
hm.RunPhase, constants.HOOKS_PHASE_PRE) |
|
| 245 |
hm = mcpu.HooksMaster(self._call_nodes_false, FakeProc(), lu) |
|
| 246 |
hm.RunPhase(constants.HOOKS_PHASE_PRE) |
|
| 247 |
#self.failUnlessRaises(errors.HooksFailure, |
|
| 248 |
# hm.RunPhase, constants.HOOKS_PHASE_PRE) |
|
| 248 | 249 |
hm.RunPhase(constants.HOOKS_PHASE_POST) |
| 249 | 250 |
|
| 250 | 251 |
def testScriptFalse(self): |
| ... | ... | |
| 253 | 254 |
op = opcodes.OpCode() |
| 254 | 255 |
sstore = FakeSStore() |
| 255 | 256 |
lu = FakeLU(None, op, cfg, sstore) |
| 256 |
hm = mcpu.HooksMaster(self._call_script_fail, lu) |
|
| 257 |
hm = mcpu.HooksMaster(self._call_script_fail, FakeProc(), lu)
|
|
| 257 | 258 |
self.failUnlessRaises(errors.HooksAbort, |
| 258 | 259 |
hm.RunPhase, constants.HOOKS_PHASE_PRE) |
| 259 | 260 |
hm.RunPhase(constants.HOOKS_PHASE_POST) |
| ... | ... | |
| 264 | 265 |
op = opcodes.OpCode() |
| 265 | 266 |
sstore = FakeSStore() |
| 266 | 267 |
lu = FakeLU(None, op, cfg, sstore) |
| 267 |
hm = mcpu.HooksMaster(self._call_script_succeed, lu) |
|
| 268 |
hm = mcpu.HooksMaster(self._call_script_succeed, FakeProc(), lu)
|
|
| 268 | 269 |
for phase in (constants.HOOKS_PHASE_PRE, constants.HOOKS_PHASE_POST): |
| 269 | 270 |
hm.RunPhase(phase) |
| 270 | 271 |
|
Also available in: Unified diff