Revision 66222813 test/py/cmdlib/testsupport/netutils_mock.py
| b/test/py/cmdlib/testsupport/netutils_mock.py | ||
|---|---|---|
| 23 | 23 |
|
| 24 | 24 |
import mock |
| 25 | 25 |
|
| 26 |
|
|
| 27 | 26 |
from ganeti import compat |
| 28 |
from ganeti import errors |
|
| 29 | 27 |
from ganeti import netutils |
| 30 | 28 |
from cmdlib.testsupport.util import patchModule |
| 31 | 29 |
|
| ... | ... | |
| 71 | 69 |
return HostnameMock(cluster.cluster_name, cluster.master_ip) |
| 72 | 70 |
|
| 73 | 71 |
node = cfg.GetNodeInfoByName(name) |
| 74 |
if node is None: |
|
| 75 |
|
|
| 76 |
raise errors.OpPrereqError( |
|
| 77 |
"Mock error: The given name '%s' is not in the config" % name, |
|
| 78 |
errors.ECODE_RESOLVER) |
|
| 72 |
if node is not None: |
|
| 73 |
return HostnameMock(node.name, node.primary_ip) |
|
| 79 | 74 |
|
| 80 |
return HostnameMock(node.name, node.primary_ip)
|
|
| 75 |
return HostnameMock(name, "1.2.3.4")
|
|
| 81 | 76 |
|
| 82 | 77 |
|
| 83 | 78 |
# pylint: disable=W0613 |
| ... | ... | |
| 114 | 109 |
compat.partial(_TcpPingMock, cfg, netutils_mod.TcpPing) |
| 115 | 110 |
netutils_mod.GetDaemonPort.side_effect = netutils.GetDaemonPort |
| 116 | 111 |
netutils_mod.FormatAddress.side_effect = netutils.FormatAddress |
| 112 |
netutils_mod.Hostname.GetNormalizedName.side_effect = \ |
|
| 113 |
netutils.Hostname.GetNormalizedName |
|
Also available in: Unified diff