Synnefo » snf-ganeti

	test/py/cmdlib/testsupport/netutils_mock.py \

	test/py/cmdlib/testsupport/rpc_runner_mock.py \

	test/py/cmdlib/testsupport/ssh_mock.py \

	test/py/cmdlib/testsupport/util.py

class TestLUClusterRename(CmdlibTestCase):

  NEW_NAME = "new-name.example.com"

  NEW_IP = "1.2.3.4"

  def testNoChanges(self):

    op = opcodes.OpClusterRename(name=self.cfg.GetClusterName())

    self.ExecOpCodeExpectOpPrereqError(op, "name nor the IP address")

  def testReachableIp(self):

    op = opcodes.OpClusterRename(name=self.NEW_NAME)

    self.netutils_mod.GetHostname.return_value = \

      HostnameMock(self.NEW_NAME, self.NEW_IP)

    self.netutils_mod.TcpPing.return_value = True

    self.ExecOpCodeExpectOpPrereqError(op, "is reachable on the network")

  def testValidRename(self):

    op = opcodes.OpClusterRename(name=self.NEW_NAME)

    self.netutils_mod.GetHostname.return_value = \

      HostnameMock(self.NEW_NAME, self.NEW_IP)

    self.ExecOpCode(op)

    self.assertEqual(1, self.ssh_mod.WriteKnownHostsFile.call_count)

    self.rpc.call_node_deactivate_master_ip.assert_called_once_with(

      self.cfg.GetMasterNode(),

      self.cfg.GetMasterNetworkParameters(),

      False)

    self.rpc.call_node_activate_master_ip.assert_called_once_with(

      self.cfg.GetMasterNode(),

      self.cfg.GetMasterNetworkParameters(),

      False)

  def testRenameOfflineMaster(self):

    op = opcodes.OpClusterRename(name=self.NEW_NAME)

    self.cfg.GetMasterNodeInfo().offline = True

    self.netutils_mod.GetHostname.return_value = \

      HostnameMock(self.NEW_NAME, self.NEW_IP)

    self.ExecOpCode(op)

from cmdlib.testsupport.netutils_mock import patchNetutils, HostnameMock

from cmdlib.testsupport.ssh_mock import patchSsh

           "HostnameMock",

           "patchIAllocator",

           "patchNetutils",

           "patchSsh",

from cmdlib.testsupport.netutils_mock import patchNetutils, \

  SetupDefaultNetutilsMock

from cmdlib.testsupport.ssh_mock import patchSsh

    * C{netutils_mod}: @see L{patchNetutils}

    * C{ssh_mod}: @see L{patchSsh}

    self._netutils_patcher = None

    self._ssh_patcher = None

    if self._netutils_patcher is not None:

      self._netutils_patcher.stop()

      self._netutils_patcher = None

    if self._ssh_patcher is not None:

      self._ssh_patcher.stop()

      self._ssh_patcher = None

    ctx = GanetiContextMock(self.cfg, self.glm, self.rpc)

    self.mcpu = ProcessorMock(ctx)

    except (ImportError, AttributeError):

    try:

      self._netutils_patcher = patchNetutils(self._GetTestModule())

      self.netutils_mod = self._netutils_patcher.start()

      SetupDefaultNetutilsMock(self.netutils_mod, self.cfg)

    except (ImportError, AttributeError):

      # this test module does not use netutils, no patching performed

      self._netutils_patcher = None

    try:

      self._ssh_patcher = patchSsh(self._GetTestModule())

      self.ssh_mod = self._ssh_patcher.start()

    except (ImportError, AttributeError):

      # this test module does not use ssh, no patching performed

      self._ssh_patcher = None

from cmdlib.testsupport.util import patchModule

  return patchModule(module_under_test, "iallocator.IAllocator")

#

#

# Copyright (C) 2013 Google Inc.

#

# This program is free software; you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation; either version 2 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful, but

# WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

# General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with this program; if not, write to the Free Software

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA

# 02110-1301, USA.

"""Support for mocking the netutils module"""

import mock

from ganeti import compat

from ganeti import errors

from ganeti import netutils

from cmdlib.testsupport.util import patchModule

# pylint: disable=C0103

def patchNetutils(module_under_test):

  """Patches the L{ganeti.netutils} module for tests.

  This function is meant to be used as a decorator for test methods.

  @type module_under_test: string

  @param module_under_test: the module within cmdlib which is tested. The

        "ganeti.cmdlib" prefix is optional.

  """

  return patchModule(module_under_test, "netutils")

class HostnameMock(object):

  """Simple mocked version of L{netutils.Hostname}.

  """

  def __init__(self, name, ip):

    self.name = name

    self.ip = ip

def _IsOverwrittenReturnValue(value):

  return value is not None and value != mock.DEFAULT and \

      not isinstance(value, mock.Mock)

# pylint: disable=W0613

def _GetHostnameMock(cfg, mock_fct, name=None, family=None):

  if _IsOverwrittenReturnValue(mock_fct.return_value):

    return mock.DEFAULT

  if name is None:

    name = cfg.GetMasterNodeName()

  if name == cfg.GetClusterName():

    cluster = cfg.GetClusterInfo()

    return HostnameMock(cluster.cluster_name, cluster.master_ip)

  node = cfg.GetNodeInfoByName(name)

  if node is None:

    raise errors.OpPrereqError(

      "Mock error: The given name '%s' is not in the config" % name,

      errors.ECODE_RESOLVER)

  return HostnameMock(node.name, node.primary_ip)

# pylint: disable=W0613

def _TcpPingMock(cfg, mock_fct, target, port, timeout=None,

                 live_port_needed=None, source=None):

  if _IsOverwrittenReturnValue(mock_fct.return_value):

    return mock.DEFAULT

  if target == cfg.GetClusterName():

    return True

  if cfg.GetNodeInfoByName(target) is not None:

    return True

  if target in [node.primary_ip for node in cfg.GetAllNodesInfo().values()]:

    return True

  if target in [node.secondary_ip for node in cfg.GetAllNodesInfo().values()]:

    return True

  return False

def SetupDefaultNetutilsMock(netutils_mod, cfg):

  """Configures the given netutils_mod mock to work with the given config.

  All relevant functions in netutils_mod are stubbed in such a way that they

  are consistent with the configuration.

  @param netutils_mod: the mock module to configure

  @type cfg: cmdlib.testsupport.ConfigMock

  @param cfg: the configuration to query for netutils request

  """

  netutils_mod.GetHostname.side_effect = \

    compat.partial(_GetHostnameMock, cfg, netutils_mod.GetHostname)

  netutils_mod.TcpPing.side_effect = \

    compat.partial(_TcpPingMock, cfg, netutils_mod.TcpPing)

  netutils_mod.GetDaemonPort.side_effect = netutils.GetDaemonPort

  netutils_mod.FormatAddress.side_effect = netutils.FormatAddress

#

#

# Copyright (C) 2013 Google Inc.

#

# This program is free software; you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation; either version 2 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful, but

# WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

# General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with this program; if not, write to the Free Software

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA

# 02110-1301, USA.

"""Support for mocking the ssh module"""

from cmdlib.testsupport.util import patchModule

# pylint: disable=C0103

def patchSsh(module_under_test):

  """Patches the L{ganeti.ssh} module for tests.

  This function is meant to be used as a decorator for test methods.

  @type module_under_test: string

  @param module_under_test: the module within cmdlib which is tested. The

        "ganeti.cmdlib" prefix is optional.

  """

  return patchModule(module_under_test, "ssh")

#

#

# Copyright (C) 2013 Google Inc.

#

# This program is free software; you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation; either version 2 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful, but

# WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU

# General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with this program; if not, write to the Free Software

# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA

# 02110-1301, USA.

"""Utility functions or the cmdlib test framework"""

import mock

# pylint: disable=C0103

def patchModule(module_under_test, mock_module):

  """Computes the module prefix required to mock parts of the Ganeti code.

  @type module_under_test: string

  @param module_under_test: the module within cmdlib which is tested. The

        "ganeti.cmdlib" prefix is optional.

  @type mock_module

  @param mock_module: the module which should be mocked.

  """

  if not module_under_test.startswith("ganeti.cmdlib"):

    module_under_test = "ganeti.cmdlib." + module_under_test

  return mock.patch("%s.%s" % (module_under_test, mock_module))