Revision 880478f8 testing/ganeti.hooks_unittest.py
| b/testing/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 fake_config import FakeConfig
|
|
| 39 |
from mocks import FakeConfig, FakeSStore
|
|
| 40 | 40 |
|
| 41 | 41 |
class FakeLU(cmdlib.LogicalUnit): |
| 42 | 42 |
HPATH = "test" |
| ... | ... | |
| 228 | 228 |
def testTotalFalse(self): |
| 229 | 229 |
"""Test complete rpc failure""" |
| 230 | 230 |
cfg = FakeConfig() |
| 231 |
sstore = FakeSStore() |
|
| 231 | 232 |
op = opcodes.OpCode() |
| 232 |
lu = FakeLU(None, op, cfg, None)
|
|
| 233 |
hm = mcpu.HooksMaster(self._call_false, cfg, lu) |
|
| 233 |
lu = FakeLU(None, op, cfg, sstore)
|
|
| 234 |
hm = mcpu.HooksMaster(self._call_false, cfg, sstore, lu)
|
|
| 234 | 235 |
self.failUnlessRaises(errors.HooksFailure, |
| 235 | 236 |
hm.RunPhase, constants.HOOKS_PHASE_PRE) |
| 236 | 237 |
hm.RunPhase(constants.HOOKS_PHASE_POST) |
| ... | ... | |
| 238 | 239 |
def testIndividualFalse(self): |
| 239 | 240 |
"""Test individual rpc failure""" |
| 240 | 241 |
cfg = FakeConfig() |
| 242 |
sstore = FakeSStore() |
|
| 241 | 243 |
op = opcodes.OpCode() |
| 242 |
lu = FakeLU(None, op, cfg, None)
|
|
| 243 |
hm = mcpu.HooksMaster(self._call_nodes_false, cfg, lu) |
|
| 244 |
lu = FakeLU(None, op, cfg, sstore)
|
|
| 245 |
hm = mcpu.HooksMaster(self._call_nodes_false, cfg, sstore, lu)
|
|
| 244 | 246 |
self.failUnlessRaises(errors.HooksFailure, |
| 245 | 247 |
hm.RunPhase, constants.HOOKS_PHASE_PRE) |
| 246 | 248 |
hm.RunPhase(constants.HOOKS_PHASE_POST) |
| ... | ... | |
| 249 | 251 |
"""Test individual rpc failure""" |
| 250 | 252 |
cfg = FakeConfig() |
| 251 | 253 |
op = opcodes.OpCode() |
| 252 |
lu = FakeLU(None, op, cfg, None) |
|
| 253 |
hm = mcpu.HooksMaster(self._call_script_fail, cfg, lu) |
|
| 254 |
sstore = FakeSStore() |
|
| 255 |
lu = FakeLU(None, op, cfg, sstore) |
|
| 256 |
hm = mcpu.HooksMaster(self._call_script_fail, cfg, sstore, lu) |
|
| 254 | 257 |
self.failUnlessRaises(errors.HooksAbort, |
| 255 | 258 |
hm.RunPhase, constants.HOOKS_PHASE_PRE) |
| 256 | 259 |
hm.RunPhase(constants.HOOKS_PHASE_POST) |
| ... | ... | |
| 259 | 262 |
"""Test individual rpc failure""" |
| 260 | 263 |
cfg = FakeConfig() |
| 261 | 264 |
op = opcodes.OpCode() |
| 262 |
lu = FakeLU(None, op, cfg, None) |
|
| 263 |
hm = mcpu.HooksMaster(self._call_script_succeed, cfg, lu) |
|
| 265 |
sstore = FakeSStore() |
|
| 266 |
lu = FakeLU(None, op, cfg, sstore) |
|
| 267 |
hm = mcpu.HooksMaster(self._call_script_succeed, cfg, sstore, lu) |
|
| 264 | 268 |
for phase in (constants.HOOKS_PHASE_PRE, constants.HOOKS_PHASE_POST): |
| 265 | 269 |
hm.RunPhase(phase) |
| 266 | 270 |
|
Also available in: Unified diff