Revision 57da0458 test/py/cmdlib/testsupport/cmdlib_testcase.py
b/test/py/cmdlib/testsupport/cmdlib_testcase.py | ||
---|---|---|
23 | 23 |
|
24 | 24 |
|
25 | 25 |
import inspect |
26 |
import mock |
|
26 | 27 |
import re |
27 | 28 |
import traceback |
28 | 29 |
|
... | ... | |
36 | 37 |
RpcResultsBuilder |
37 | 38 |
from cmdlib.testsupport.ssh_mock import patchSsh |
38 | 39 |
|
40 |
from ganeti.cmdlib.base import LogicalUnit |
|
39 | 41 |
from ganeti import errors |
40 | 42 |
from ganeti import objects |
41 | 43 |
from ganeti import opcodes |
... | ... | |
45 | 47 |
|
46 | 48 |
|
47 | 49 |
class GanetiContextMock(object): |
48 |
def __init__(self, cfg, glm, rpc): |
|
49 |
self.cfg = cfg |
|
50 |
self.glm = glm |
|
51 |
self.rpc = rpc |
|
50 |
# pylint: disable=W0212 |
|
51 |
cfg = property(fget=lambda self: self._test_case.cfg) |
|
52 |
# pylint: disable=W0212 |
|
53 |
glm = property(fget=lambda self: self._test_case.glm) |
|
54 |
# pylint: disable=W0212 |
|
55 |
rpc = property(fget=lambda self: self._test_case.rpc) |
|
56 |
|
|
57 |
def __init__(self, test_case): |
|
58 |
self._test_case = test_case |
|
59 |
|
|
60 |
|
|
61 |
class MockLU(LogicalUnit): |
|
62 |
def BuildHooksNodes(self): |
|
63 |
pass |
|
64 |
|
|
65 |
def BuildHooksEnv(self): |
|
66 |
pass |
|
52 | 67 |
|
53 | 68 |
|
54 | 69 |
# pylint: disable=R0904 |
... | ... | |
137 | 152 |
self.cfg = ConfigMock() |
138 | 153 |
self.glm = LockManagerMock() |
139 | 154 |
self.rpc = CreateRpcRunnerMock() |
140 |
ctx = GanetiContextMock(self.cfg, self.glm, self.rpc)
|
|
141 |
self.mcpu = ProcessorMock(ctx) |
|
155 |
self.ctx = GanetiContextMock(self)
|
|
156 |
self.mcpu = ProcessorMock(self.ctx)
|
|
142 | 157 |
|
143 | 158 |
self._StopPatchers() |
144 | 159 |
try: |
... | ... | |
163 | 178 |
# this test module does not use ssh, no patching performed |
164 | 179 |
self._ssh_patcher = None |
165 | 180 |
|
181 |
def GetMockLU(self): |
|
182 |
"""Creates a mock L{LogialUnit} with access to the mocked config etc. |
|
183 |
|
|
184 |
@rtype: L{LogialUnit} |
|
185 |
@return: A mock LU |
|
186 |
|
|
187 |
""" |
|
188 |
return MockLU(self.mcpu, mock.MagicMock(), self.ctx, self.rpc) |
|
189 |
|
|
166 | 190 |
def RpcResultsBuilder(self, use_node_names=False): |
167 | 191 |
"""Creates a pre-configured L{RpcResultBuilder} |
168 | 192 |
|
Also available in: Unified diff