Revision bd39b6bb test/py/cmdlib/testsupport/rpc_runner_mock.py
| b/test/py/cmdlib/testsupport/rpc_runner_mock.py | ||
|---|---|---|
| 18 | 18 |
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA |
| 19 | 19 |
# 02110-1301, USA. |
| 20 | 20 |
|
| 21 |
|
|
| 22 |
"""Support for mocking the RPC runner""" |
|
| 23 |
|
|
| 24 |
|
|
| 21 | 25 |
import mock |
| 22 | 26 |
|
| 23 | 27 |
from ganeti import objects |
| ... | ... | |
| 91 | 95 |
else: |
| 92 | 96 |
return node.uuid |
| 93 | 97 |
|
| 94 |
def CreateSuccessfulNodeResult(self, node, data={}):
|
|
| 98 |
def CreateSuccessfulNodeResult(self, node, data=None):
|
|
| 95 | 99 |
"""@see L{RpcResultsBuilder}
|
| 96 | 100 |
|
| 97 | 101 |
@param node: @see L{RpcResultsBuilder}.
|
| ... | ... | |
| 99 | 103 |
@param data: the data as returned by the RPC |
| 100 | 104 |
@rtype: L{rpc.RpcResult}
|
| 101 | 105 |
""" |
| 106 |
if data is None: |
|
| 107 |
data = {}
|
|
| 102 | 108 |
return rpc.RpcResult(data=(True, data), node=self._GetNodeId(node)) |
| 103 | 109 |
|
| 104 | 110 |
def CreateFailedNodeResult(self, node): |
| ... | ... | |
| 127 | 133 |
""" |
| 128 | 134 |
return rpc.RpcResult(data=(False, error_msg), node=self._GetNodeId(node)) |
| 129 | 135 |
|
| 130 |
def AddSuccessfulNode(self, node, data={}):
|
|
| 136 |
def AddSuccessfulNode(self, node, data=None):
|
|
| 131 | 137 |
"""@see L{CreateSuccessfulNode}"""
|
| 132 | 138 |
self._results.append(self.CreateSuccessfulNodeResult(node, data)) |
| 133 | 139 |
return self |
| ... | ... | |
| 150 | 156 |
|
| 151 | 157 |
@rtype: dict |
| 152 | 158 |
""" |
| 153 |
return dict((result.node, result) for result in self._results) |
|
| 159 |
return dict((result.node, result) for result in self._results) |
|
Also available in: Unified diff