Revision d2525573 test/ganeti.hooks_unittest.py
| b/test/ganeti.hooks_unittest.py | ||
|---|---|---|
| 225 | 225 |
return dict([(node_name, [("unittest", constants.HKR_SUCCESS, "ok")])
|
| 226 | 226 |
for node_name in node_list]) |
| 227 | 227 |
|
| 228 |
def setUp(self): |
|
| 229 |
self.cfg = FakeConfig() |
|
| 230 |
self.sstore = FakeSStore() |
|
| 231 |
self.op = opcodes.OpCode() |
|
| 232 |
self.lu = FakeLU(None, self.op, self.cfg, self.sstore) |
|
| 233 |
|
|
| 228 | 234 |
def testTotalFalse(self): |
| 229 | 235 |
"""Test complete rpc failure""" |
| 230 |
cfg = FakeConfig() |
|
| 231 |
sstore = FakeSStore() |
|
| 232 |
op = opcodes.OpCode() |
|
| 233 |
lu = FakeLU(None, op, cfg, sstore) |
|
| 234 |
hm = mcpu.HooksMaster(self._call_false, FakeProc(), lu) |
|
| 236 |
hm = mcpu.HooksMaster(self._call_false, FakeProc(), self.lu) |
|
| 235 | 237 |
self.failUnlessRaises(errors.HooksFailure, |
| 236 | 238 |
hm.RunPhase, constants.HOOKS_PHASE_PRE) |
| 237 | 239 |
hm.RunPhase(constants.HOOKS_PHASE_POST) |
| 238 | 240 |
|
| 239 | 241 |
def testIndividualFalse(self): |
| 240 | 242 |
"""Test individual node failure""" |
| 241 |
cfg = FakeConfig() |
|
| 242 |
sstore = FakeSStore() |
|
| 243 |
op = opcodes.OpCode() |
|
| 244 |
lu = FakeLU(None, op, cfg, sstore) |
|
| 245 |
hm = mcpu.HooksMaster(self._call_nodes_false, FakeProc(), lu) |
|
| 243 |
hm = mcpu.HooksMaster(self._call_nodes_false, FakeProc(), self.lu) |
|
| 246 | 244 |
hm.RunPhase(constants.HOOKS_PHASE_PRE) |
| 247 | 245 |
#self.failUnlessRaises(errors.HooksFailure, |
| 248 | 246 |
# hm.RunPhase, constants.HOOKS_PHASE_PRE) |
| ... | ... | |
| 250 | 248 |
|
| 251 | 249 |
def testScriptFalse(self): |
| 252 | 250 |
"""Test individual rpc failure""" |
| 253 |
cfg = FakeConfig() |
|
| 254 |
op = opcodes.OpCode() |
|
| 255 |
sstore = FakeSStore() |
|
| 256 |
lu = FakeLU(None, op, cfg, sstore) |
|
| 257 |
hm = mcpu.HooksMaster(self._call_script_fail, FakeProc(), lu) |
|
| 251 |
hm = mcpu.HooksMaster(self._call_script_fail, FakeProc(), self.lu) |
|
| 258 | 252 |
self.failUnlessRaises(errors.HooksAbort, |
| 259 | 253 |
hm.RunPhase, constants.HOOKS_PHASE_PRE) |
| 260 | 254 |
hm.RunPhase(constants.HOOKS_PHASE_POST) |
| 261 | 255 |
|
| 262 | 256 |
def testScriptSucceed(self): |
| 263 | 257 |
"""Test individual rpc failure""" |
| 264 |
cfg = FakeConfig() |
|
| 265 |
op = opcodes.OpCode() |
|
| 266 |
sstore = FakeSStore() |
|
| 267 |
lu = FakeLU(None, op, cfg, sstore) |
|
| 268 |
hm = mcpu.HooksMaster(self._call_script_succeed, FakeProc(), lu) |
|
| 258 |
hm = mcpu.HooksMaster(self._call_script_succeed, FakeProc(), self.lu) |
|
| 269 | 259 |
for phase in (constants.HOOKS_PHASE_PRE, constants.HOOKS_PHASE_POST): |
| 270 | 260 |
hm.RunPhase(phase) |
| 271 | 261 |
|
Also available in: Unified diff