Synnefo » snf-ganeti

#!/usr/bin/python

#

# Copyright (C) 2008, 2011, 2012, 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.

"""Tests for LUCluster*

"""

import OpenSSL

import unittest

import operator

import os

import tempfile

import shutil

from ganeti import constants

from ganeti import errors

from ganeti import netutils

from ganeti import objects

from ganeti import opcodes

from ganeti import utils

from ganeti import pathutils

from ganeti import query

from ganeti.cmdlib import cluster

from ganeti.hypervisor import hv_xen

from testsupport import *

import testutils

class TestCertVerification(testutils.GanetiTestCase):

  def setUp(self):

    testutils.GanetiTestCase.setUp(self)

    self.tmpdir = tempfile.mkdtemp()

  def tearDown(self):

    shutil.rmtree(self.tmpdir)

  def testVerifyCertificate(self):

    cluster._VerifyCertificate(testutils.TestDataFilename("cert1.pem"))

    nonexist_filename = os.path.join(self.tmpdir, "does-not-exist")

    (errcode, msg) = cluster._VerifyCertificate(nonexist_filename)

    self.assertEqual(errcode, cluster.LUClusterVerifyConfig.ETYPE_ERROR)

    # Try to load non-certificate file

    invalid_cert = testutils.TestDataFilename("bdev-net.txt")

    (errcode, msg) = cluster._VerifyCertificate(invalid_cert)

    self.assertEqual(errcode, cluster.LUClusterVerifyConfig.ETYPE_ERROR)

class TestClusterVerifySsh(unittest.TestCase):

  def testMultipleGroups(self):

    fn = cluster.LUClusterVerifyGroup._SelectSshCheckNodes

    mygroupnodes = [

      objects.Node(name="node20", group="my", offline=False),

      objects.Node(name="node21", group="my", offline=False),

      objects.Node(name="node22", group="my", offline=False),

      objects.Node(name="node23", group="my", offline=False),

      objects.Node(name="node24", group="my", offline=False),

      objects.Node(name="node25", group="my", offline=False),

      objects.Node(name="node26", group="my", offline=True),

      ]

    nodes = [

      objects.Node(name="node1", group="g1", offline=True),

      objects.Node(name="node2", group="g1", offline=False),

      objects.Node(name="node3", group="g1", offline=False),

      objects.Node(name="node4", group="g1", offline=True),

      objects.Node(name="node5", group="g1", offline=False),

      objects.Node(name="node10", group="xyz", offline=False),

      objects.Node(name="node11", group="xyz", offline=False),

      objects.Node(name="node40", group="alloff", offline=True),

      objects.Node(name="node41", group="alloff", offline=True),

      objects.Node(name="node50", group="aaa", offline=False),

      ] + mygroupnodes

    assert not utils.FindDuplicates(map(operator.attrgetter("name"), nodes))

    (online, perhost) = fn(mygroupnodes, "my", nodes)

    self.assertEqual(online, ["node%s" % i for i in range(20, 26)])

    self.assertEqual(set(perhost.keys()), set(online))

    self.assertEqual(perhost, {

      "node20": ["node10", "node2", "node50"],

      "node21": ["node11", "node3", "node50"],

      "node22": ["node10", "node5", "node50"],

      "node23": ["node11", "node2", "node50"],

      "node24": ["node10", "node3", "node50"],

      "node25": ["node11", "node5", "node50"],

      })

  def testSingleGroup(self):

    fn = cluster.LUClusterVerifyGroup._SelectSshCheckNodes

    nodes = [

      objects.Node(name="node1", group="default", offline=True),

      objects.Node(name="node2", group="default", offline=False),

      objects.Node(name="node3", group="default", offline=False),

      objects.Node(name="node4", group="default", offline=True),

      ]

    assert not utils.FindDuplicates(map(operator.attrgetter("name"), nodes))

    (online, perhost) = fn(nodes, "default", nodes)

    self.assertEqual(online, ["node2", "node3"])

    self.assertEqual(set(perhost.keys()), set(online))

    self.assertEqual(perhost, {

      "node2": [],

      "node3": [],

      })

class TestLUClusterActivateMasterIp(CmdlibTestCase):

  def testSuccess(self):

    op = opcodes.OpClusterActivateMasterIp()

    self.rpc.call_node_activate_master_ip.return_value = \

      self.RpcResultsBuilder() \

        .CreateSuccessfulNodeResult(self.master)

    self.ExecOpCode(op)

    self.rpc.call_node_activate_master_ip.assert_called_once_with(

      self.master_uuid, self.cfg.GetMasterNetworkParameters(), False)

  def testFailure(self):

    op = opcodes.OpClusterActivateMasterIp()

    self.rpc.call_node_activate_master_ip.return_value = \

      self.RpcResultsBuilder() \

        .CreateFailedNodeResult(self.master) \

    self.ExecOpCodeExpectOpExecError(op)

class TestLUClusterDeactivateMasterIp(CmdlibTestCase):

  def testSuccess(self):

    op = opcodes.OpClusterDeactivateMasterIp()

    self.rpc.call_node_deactivate_master_ip.return_value = \

      self.RpcResultsBuilder() \

        .CreateSuccessfulNodeResult(self.master)

    self.ExecOpCode(op)

    self.rpc.call_node_deactivate_master_ip.assert_called_once_with(

      self.master_uuid, self.cfg.GetMasterNetworkParameters(), False)

  def testFailure(self):

    op = opcodes.OpClusterDeactivateMasterIp()

    self.rpc.call_node_deactivate_master_ip.return_value = \

      self.RpcResultsBuilder() \

        .CreateFailedNodeResult(self.master) \

    self.ExecOpCodeExpectOpExecError(op)

class TestLUClusterConfigQuery(CmdlibTestCase):

  def testInvalidField(self):

    op = opcodes.OpClusterConfigQuery(output_fields=["pinky_bunny"])

    self.ExecOpCodeExpectOpPrereqError(op, "pinky_bunny")

  def testAllFields(self):

    op = opcodes.OpClusterConfigQuery(output_fields=query.CLUSTER_FIELDS.keys())

    self.rpc.call_get_watcher_pause.return_value = \

      self.RpcResultsBuilder() \

        .CreateSuccessfulNodeResult(self.master, -1)

    ret = self.ExecOpCode(op)

    self.assertEqual(1, self.rpc.call_get_watcher_pause.call_count)

    self.assertEqual(len(ret), len(query.CLUSTER_FIELDS))

  def testEmpytFields(self):

    op = opcodes.OpClusterConfigQuery(output_fields=[])

    self.ExecOpCode(op)

    self.assertFalse(self.rpc.call_get_watcher_pause.called)

class TestLUClusterDestroy(CmdlibTestCase):

  def testExistingNodes(self):

    op = opcodes.OpClusterDestroy()

    self.cfg.AddNewNode()

    self.cfg.AddNewNode()

    self.ExecOpCodeExpectOpPrereqError(op, "still 2 node\(s\)")

  def testExistingInstances(self):

    op = opcodes.OpClusterDestroy()

    self.cfg.AddNewInstance()

    self.cfg.AddNewInstance()

    self.ExecOpCodeExpectOpPrereqError(op, "still 2 instance\(s\)")

  def testEmptyCluster(self):

    op = opcodes.OpClusterDestroy()

    self.ExecOpCode(op)

    self.assertSingleHooksCall([self.master.name],

                               "cluster-destroy",

                               constants.HOOKS_PHASE_POST)

class TestLUClusterPostInit(CmdlibTestCase):

  def testExecuion(self):

    op = opcodes.OpClusterPostInit()

    self.ExecOpCode(op)

    self.assertSingleHooksCall([self.master.name],

                               "cluster-init",

                               constants.HOOKS_PHASE_POST)

class TestLUClusterQuery(CmdlibTestCase):

  def testSimpleInvocation(self):

    op = opcodes.OpClusterQuery()

    self.ExecOpCode(op)

  def testIPv6Cluster(self):

    op = opcodes.OpClusterQuery()

    self.cluster.primary_ip_family = netutils.IP6Address.family

    self.ExecOpCode(op)

class TestLUClusterRedistConf(CmdlibTestCase):

  def testSimpleInvocation(self):

    op = opcodes.OpClusterRedistConf()

    self.ExecOpCode(op)

class TestLUClusterRename(CmdlibTestCase):

  NEW_NAME = "new-name.example.com"

  NEW_IP = "203.0.113.100"

  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.master_uuid, self.cfg.GetMasterNetworkParameters(), False)

    self.rpc.call_node_activate_master_ip.assert_called_once_with(

      self.master_uuid, self.cfg.GetMasterNetworkParameters(), False)

  def testRenameOfflineMaster(self):

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

    self.master.offline = True

    self.netutils_mod.GetHostname.return_value = \

      HostnameMock(self.NEW_NAME, self.NEW_IP)

    self.ExecOpCode(op)

class TestLUClusterRepairDiskSizes(CmdlibTestCase):

  def testNoInstances(self):

    op = opcodes.OpClusterRepairDiskSizes()

    self.ExecOpCode(op)

  def _SetUpInstanceSingleDisk(self, dev_type=constants.DT_PLAIN):

    pnode = self.master

    snode = self.cfg.AddNewNode()

    disk = self.cfg.CreateDisk(dev_type=dev_type,

                               primary_node=pnode,

                               secondary_node=snode)

    inst = self.cfg.AddNewInstance(disks=[disk])

    return (inst, disk)

  def testSingleInstanceOnFailingNode(self):

    (inst, _) = self._SetUpInstanceSingleDisk()

    op = opcodes.OpClusterRepairDiskSizes(instances=[inst.name])

    self.rpc.call_blockdev_getdimensions.return_value = \

      self.RpcResultsBuilder() \

        .CreateFailedNodeResult(self.master)

    self.ExecOpCode(op)

    self.mcpu.assertLogContainsRegex("Failure in blockdev_getdimensions")

  def _ExecOpClusterRepairDiskSizes(self, node_data):

    # not specifying instances repairs all

    op = opcodes.OpClusterRepairDiskSizes()

    self.rpc.call_blockdev_getdimensions.return_value = \

      self.RpcResultsBuilder() \

        .CreateSuccessfulNodeResult(self.master, node_data)

    return self.ExecOpCode(op)

  def testInvalidResultData(self):

    for data in [[], [None], ["invalid"], [("still", "invalid")]]:

      self.ResetMocks()

      self._SetUpInstanceSingleDisk()

      self._ExecOpClusterRepairDiskSizes(data)

      self.mcpu.assertLogContainsRegex("ignoring")

  def testCorrectSize(self):

    self._SetUpInstanceSingleDisk()

    changed = self._ExecOpClusterRepairDiskSizes([(1024 * 1024 * 1024, None)])

    self.mcpu.assertLogIsEmpty()

    self.assertEqual(0, len(changed))

  def testWrongSize(self):

    self._SetUpInstanceSingleDisk()

    changed = self._ExecOpClusterRepairDiskSizes([(512 * 1024 * 1024, None)])

    self.assertEqual(1, len(changed))

  def testCorrectDRBD(self):

    self._SetUpInstanceSingleDisk(dev_type=constants.DT_DRBD8)

    changed = self._ExecOpClusterRepairDiskSizes([(1024 * 1024 * 1024, None)])

    self.mcpu.assertLogIsEmpty()

    self.assertEqual(0, len(changed))

  def testWrongDRBDChild(self):

    (_, disk) = self._SetUpInstanceSingleDisk(dev_type=constants.DT_DRBD8)

    disk.children[0].size = 512

    changed = self._ExecOpClusterRepairDiskSizes([(1024 * 1024 * 1024, None)])

    self.assertEqual(1, len(changed))

  def testExclusiveStorageInvalidResultData(self):

    self._SetUpInstanceSingleDisk()

    self.master.ndparams[constants.ND_EXCLUSIVE_STORAGE] = True

    self._ExecOpClusterRepairDiskSizes([(1024 * 1024 * 1024, None)])

    self.mcpu.assertLogContainsRegex(

      "did not return valid spindles information")

  def testExclusiveStorageCorrectSpindles(self):

    (_, disk) = self._SetUpInstanceSingleDisk()

    disk.spindles = 1

    self.master.ndparams[constants.ND_EXCLUSIVE_STORAGE] = True

    changed = self._ExecOpClusterRepairDiskSizes([(1024 * 1024 * 1024, 1)])

    self.assertEqual(0, len(changed))

  def testExclusiveStorageWrongSpindles(self):

    self._SetUpInstanceSingleDisk()

    self.master.ndparams[constants.ND_EXCLUSIVE_STORAGE] = True

    changed = self._ExecOpClusterRepairDiskSizes([(1024 * 1024 * 1024, 1)])

    self.assertEqual(1, len(changed))

class TestLUClusterSetParams(CmdlibTestCase):

  UID_POOL = [(10, 1000)]

  def testUidPool(self):

    op = opcodes.OpClusterSetParams(uid_pool=self.UID_POOL)

    self.ExecOpCode(op)

    self.assertEqual(self.UID_POOL, self.cluster.uid_pool)

  def testAddUids(self):

    old_pool = [(1, 9)]

    self.cluster.uid_pool = list(old_pool)

    op = opcodes.OpClusterSetParams(add_uids=self.UID_POOL)

    self.ExecOpCode(op)

    self.assertEqual(set(self.UID_POOL + old_pool),

                     set(self.cluster.uid_pool))

  def testRemoveUids(self):

    additional_pool = [(1, 9)]

    self.cluster.uid_pool = self.UID_POOL + additional_pool

    op = opcodes.OpClusterSetParams(remove_uids=self.UID_POOL)

    self.ExecOpCode(op)

    self.assertEqual(additional_pool, self.cluster.uid_pool)

  def testMasterNetmask(self):

    op = opcodes.OpClusterSetParams(master_netmask=26)

    self.ExecOpCode(op)

    self.assertEqual(26, self.cluster.master_netmask)

  def testInvalidDiskparams(self):

    for diskparams in [{constants.DT_DISKLESS: {constants.LV_STRIPES: 0}},

                       {constants.DT_DRBD8: {constants.RBD_POOL: "pool"}},

                       {constants.DT_DRBD8: {constants.RBD_ACCESS: "bunny"}}]:

      self.ResetMocks()

      op = opcodes.OpClusterSetParams(diskparams=diskparams)

      self.ExecOpCodeExpectOpPrereqError(op, "verify diskparams")

  def testValidDiskparams(self):

    diskparams = {constants.DT_RBD: {constants.RBD_POOL: "mock_pool",

                                     constants.RBD_ACCESS: "kernelspace"}}

    op = opcodes.OpClusterSetParams(diskparams=diskparams)

    self.ExecOpCode(op)

    self.assertEqual(diskparams[constants.DT_RBD],

                     self.cluster.diskparams[constants.DT_RBD])

  def testMinimalDiskparams(self):

    diskparams = {constants.DT_RBD: {constants.RBD_POOL: "mock_pool"}}

    self.cluster.diskparams = {}

    op = opcodes.OpClusterSetParams(diskparams=diskparams)

    self.ExecOpCode(op)

    self.assertEqual(diskparams, self.cluster.diskparams)

  def testValidDiskparamsAccess(self):

    for value in constants.DISK_VALID_ACCESS_MODES:

      self.ResetMocks()

      op = opcodes.OpClusterSetParams(diskparams={

        constants.DT_RBD: {constants.RBD_ACCESS: value}

      })

      self.ExecOpCode(op)

      got = self.cluster.diskparams[constants.DT_RBD][constants.RBD_ACCESS]

      self.assertEqual(value, got)

  def testInvalidDiskparamsAccess(self):

    for value in ["default", "pinky_bunny"]:

      self.ResetMocks()

      op = opcodes.OpClusterSetParams(diskparams={

        constants.DT_RBD: {constants.RBD_ACCESS: value}

      })

      self.ExecOpCodeExpectOpPrereqError(op, "Invalid value of 'rbd:access'")

  def testUnsetDrbdHelperWithDrbdDisks(self):

    self.cfg.AddNewInstance(disks=[

      self.cfg.CreateDisk(dev_type=constants.DT_DRBD8, create_nodes=True)])

    op = opcodes.OpClusterSetParams(drbd_helper="")

    self.ExecOpCodeExpectOpPrereqError(op, "Cannot disable drbd helper")

  def testFileStorageDir(self):

    op = opcodes.OpClusterSetParams(file_storage_dir="/random/path")

    self.ExecOpCode(op)

  def testSetFileStorageDirToCurrentValue(self):

    op = opcodes.OpClusterSetParams(

           file_storage_dir=self.cluster.file_storage_dir)

    self.ExecOpCode(op)

    self.mcpu.assertLogContainsRegex("file storage dir already set to value")

  def testUnsetFileStorageDirFileStorageEnabled(self):

    self.cfg.SetEnabledDiskTemplates([constants.DT_FILE])

    op = opcodes.OpClusterSetParams(file_storage_dir='')

    self.ExecOpCodeExpectOpPrereqError(op, "Unsetting the 'file' storage")

  def testUnsetFileStorageDirFileStorageDisabled(self):

    self.cfg.SetEnabledDiskTemplates([constants.DT_PLAIN])

    op = opcodes.OpClusterSetParams(file_storage_dir='')

    self.ExecOpCode(op)

  def testSetFileStorageDirFileStorageDisabled(self):

    self.cfg.SetEnabledDiskTemplates([constants.DT_PLAIN])

    op = opcodes.OpClusterSetParams(file_storage_dir='/some/path/')

    self.ExecOpCode(op)

    self.mcpu.assertLogContainsRegex("although file storage is not enabled")

  def testValidDrbdHelper(self):

    node1 = self.cfg.AddNewNode()

    node1.offline = True

    self.rpc.call_drbd_helper.return_value = \

      self.RpcResultsBuilder() \

        .AddSuccessfulNode(self.master, "/bin/true") \

        .AddOfflineNode(node1) \

        .Build()

    op = opcodes.OpClusterSetParams(drbd_helper="/bin/true")

    self.ExecOpCode(op)

    self.mcpu.assertLogContainsRegex("Not checking drbd helper on offline node")

  def testDrbdHelperFailingNode(self):

    self.rpc.call_drbd_helper.return_value = \

      self.RpcResultsBuilder() \

        .AddFailedNode(self.master) \

        .Build()

    op = opcodes.OpClusterSetParams(drbd_helper="/bin/true")

    self.ExecOpCodeExpectOpPrereqError(op, "Error checking drbd helper")

  def testInvalidDrbdHelper(self):

    self.rpc.call_drbd_helper.return_value = \

      self.RpcResultsBuilder() \

        .AddSuccessfulNode(self.master, "/bin/false") \

        .Build()

    op = opcodes.OpClusterSetParams(drbd_helper="/bin/true")

    self.ExecOpCodeExpectOpPrereqError(op, "drbd helper is /bin/false")

  def testDrbdHelperWithoutDrbdDiskTemplate(self):

    drbd_helper = "/bin/random_helper"

    self.cfg.SetEnabledDiskTemplates([constants.DT_DISKLESS])

    self.rpc.call_drbd_helper.return_value = \

      self.RpcResultsBuilder() \

        .AddSuccessfulNode(self.master, drbd_helper) \

        .Build()

    op = opcodes.OpClusterSetParams(drbd_helper=drbd_helper)

    self.ExecOpCode(op)

    self.mcpu.assertLogContainsRegex("but did not enable")

  def testResetDrbdHelperDrbdDisabled(self):

    drbd_helper = ""

    self.cfg.SetEnabledDiskTemplates([constants.DT_DISKLESS])

    op = opcodes.OpClusterSetParams(drbd_helper=drbd_helper)

    self.ExecOpCode(op)

    self.assertEqual(None, self.cluster.drbd_usermode_helper)

  def testResetDrbdHelperDrbdEnabled(self):

    drbd_helper = ""

    self.cluster.enabled_disk_templates = [constants.DT_DRBD8]

    op = opcodes.OpClusterSetParams(drbd_helper=drbd_helper)

    self.ExecOpCodeExpectOpPrereqError(

        op, "Cannot disable drbd helper while DRBD is enabled.")

  def testEnableDrbdNoHelper(self):

    self.cluster.enabled_disk_templates = [constants.DT_DISKLESS]

    self.cluster.drbd_usermode_helper = None

    enabled_disk_templates = [constants.DT_DRBD8]

    op = opcodes.OpClusterSetParams(

        enabled_disk_templates=enabled_disk_templates)

    self.ExecOpCodeExpectOpPrereqError(

        op, "Cannot enable DRBD without a DRBD usermode helper set")

  def testEnableDrbdHelperSet(self):

    drbd_helper = "/bin/random_helper"

    self.rpc.call_drbd_helper.return_value = \

      self.RpcResultsBuilder() \

        .AddSuccessfulNode(self.master, drbd_helper) \

        .Build()

    self.cfg.SetEnabledDiskTemplates([constants.DT_DISKLESS])

    self.cluster.drbd_usermode_helper = drbd_helper

    enabled_disk_templates = [constants.DT_DRBD8]

    op = opcodes.OpClusterSetParams(

        enabled_disk_templates=enabled_disk_templates,

        ipolicy={constants.IPOLICY_DTS: enabled_disk_templates})

    self.ExecOpCode(op)

    self.assertEqual(drbd_helper, self.cluster.drbd_usermode_helper)

  def testDrbdHelperAlreadySet(self):

    drbd_helper = "/bin/true"

    self.rpc.call_drbd_helper.return_value = \

      self.RpcResultsBuilder() \

        .AddSuccessfulNode(self.master, "/bin/true") \

        .Build()

    self.cfg.SetEnabledDiskTemplates([constants.DT_DISKLESS])

    op = opcodes.OpClusterSetParams(drbd_helper=drbd_helper)

    self.ExecOpCode(op)

    self.assertEqual(drbd_helper, self.cluster.drbd_usermode_helper)

    self.mcpu.assertLogContainsRegex("DRBD helper already in desired state")

  def testSetDrbdHelper(self):

    drbd_helper = "/bin/true"

    self.rpc.call_drbd_helper.return_value = \

      self.RpcResultsBuilder() \

        .AddSuccessfulNode(self.master, "/bin/true") \

        .Build()

    self.cluster.drbd_usermode_helper = "/bin/false"

    self.cfg.SetEnabledDiskTemplates([constants.DT_DRBD8])

    op = opcodes.OpClusterSetParams(drbd_helper=drbd_helper)

    self.ExecOpCode(op)

    self.assertEqual(drbd_helper, self.cluster.drbd_usermode_helper)

  def testBeparams(self):

    beparams = {constants.BE_VCPUS: 32}

    op = opcodes.OpClusterSetParams(beparams=beparams)

    self.ExecOpCode(op)

    self.assertEqual(32, self.cluster

                           .beparams[constants.PP_DEFAULT][constants.BE_VCPUS])

  def testNdparams(self):

    ndparams = {constants.ND_EXCLUSIVE_STORAGE: True}

    op = opcodes.OpClusterSetParams(ndparams=ndparams)

    self.ExecOpCode(op)

    self.assertEqual(True, self.cluster

                             .ndparams[constants.ND_EXCLUSIVE_STORAGE])

  def testNdparamsResetOobProgram(self):

    ndparams = {constants.ND_OOB_PROGRAM: ""}

    op = opcodes.OpClusterSetParams(ndparams=ndparams)

    self.ExecOpCode(op)

    self.assertEqual(constants.NDC_DEFAULTS[constants.ND_OOB_PROGRAM],

                     self.cluster.ndparams[constants.ND_OOB_PROGRAM])

  def testHvState(self):

    hv_state = {constants.HT_FAKE: {constants.HVST_CPU_TOTAL: 8}}

    op = opcodes.OpClusterSetParams(hv_state=hv_state)

    self.ExecOpCode(op)

    self.assertEqual(8, self.cluster.hv_state_static

                          [constants.HT_FAKE][constants.HVST_CPU_TOTAL])

  def testDiskState(self):

    disk_state = {

      constants.DT_PLAIN: {

        "mock_vg": {constants.DS_DISK_TOTAL: 10}

      }

    }

    op = opcodes.OpClusterSetParams(disk_state=disk_state)

    self.ExecOpCode(op)

    self.assertEqual(10, self.cluster

                           .disk_state_static[constants.DT_PLAIN]["mock_vg"]

                             [constants.DS_DISK_TOTAL])

  def testDefaultIPolicy(self):

    ipolicy = constants.IPOLICY_DEFAULTS

    op = opcodes.OpClusterSetParams(ipolicy=ipolicy)

    self.ExecOpCode(op)

  def testIPolicyNewViolation(self):

    import ganeti.constants as C

    ipolicy = C.IPOLICY_DEFAULTS

    ipolicy[C.ISPECS_MINMAX][0][C.ISPECS_MIN][C.ISPEC_MEM_SIZE] = 128

    ipolicy[C.ISPECS_MINMAX][0][C.ISPECS_MAX][C.ISPEC_MEM_SIZE] = 128

    self.cfg.AddNewInstance(beparams={C.BE_MINMEM: 512, C.BE_MAXMEM: 512})

    op = opcodes.OpClusterSetParams(ipolicy=ipolicy)

    self.ExecOpCode(op)

    self.mcpu.assertLogContainsRegex("instances violate them")

  def testNicparamsNoInstance(self):

    nicparams = {

      constants.NIC_LINK: "mock_bridge"

    }

    op = opcodes.OpClusterSetParams(nicparams=nicparams)

    self.ExecOpCode(op)

    self.assertEqual("mock_bridge",

                     self.cluster.nicparams

                       [constants.PP_DEFAULT][constants.NIC_LINK])

  def testNicparamsInvalidConf(self):

    nicparams = {

      constants.NIC_MODE: constants.NIC_MODE_BRIDGED,

      constants.NIC_LINK: ""

    }

    op = opcodes.OpClusterSetParams(nicparams=nicparams)

    self.ExecOpCodeExpectException(op, errors.ConfigurationError, "NIC link")

  def testNicparamsInvalidInstanceConf(self):

    nicparams = {

      constants.NIC_MODE: constants.NIC_MODE_BRIDGED,

      constants.NIC_LINK: "mock_bridge"

    }

    self.cfg.AddNewInstance(nics=[

      self.cfg.CreateNic(nicparams={constants.NIC_LINK: None})])

    op = opcodes.OpClusterSetParams(nicparams=nicparams)

    self.ExecOpCodeExpectOpPrereqError(op, "Missing bridged NIC link")

  def testNicparamsMissingIp(self):

    nicparams = {

      constants.NIC_MODE: constants.NIC_MODE_ROUTED

    }

    self.cfg.AddNewInstance()

    op = opcodes.OpClusterSetParams(nicparams=nicparams)

    self.ExecOpCodeExpectOpPrereqError(op, "routed NIC with no ip address")

  def testNicparamsWithInstance(self):

    nicparams = {

      constants.NIC_LINK: "mock_bridge"

    }

    self.cfg.AddNewInstance()

    op = opcodes.OpClusterSetParams(nicparams=nicparams)

    self.ExecOpCode(op)

  def testDefaultHvparams(self):

    hvparams = constants.HVC_DEFAULTS

    op = opcodes.OpClusterSetParams(hvparams=hvparams)

    self.ExecOpCode(op)

    self.assertEqual(hvparams, self.cluster.hvparams)

  def testMinimalHvparams(self):

    hvparams = {

      constants.HT_FAKE: {

        constants.HV_MIGRATION_MODE: constants.HT_MIGRATION_NONLIVE

      }

    }

    self.cluster.hvparams = {}

    op = opcodes.OpClusterSetParams(hvparams=hvparams)

    self.ExecOpCode(op)

    self.assertEqual(hvparams, self.cluster.hvparams)

  def testOsHvp(self):

    os_hvp = {

      "mocked_os": {

        constants.HT_FAKE: {

          constants.HV_MIGRATION_MODE: constants.HT_MIGRATION_NONLIVE

        }

      },

      "other_os": constants.HVC_DEFAULTS

    }

    op = opcodes.OpClusterSetParams(os_hvp=os_hvp)

    self.ExecOpCode(op)

    self.assertEqual(constants.HT_MIGRATION_NONLIVE,

                     self.cluster.os_hvp["mocked_os"][constants.HT_FAKE]

                       [constants.HV_MIGRATION_MODE])

    self.assertEqual(constants.HVC_DEFAULTS, self.cluster.os_hvp["other_os"])

  def testRemoveOsHvp(self):

    os_hvp = {"mocked_os": {constants.HT_FAKE: None}}

    op = opcodes.OpClusterSetParams(os_hvp=os_hvp)

    self.ExecOpCode(op)

    assert constants.HT_FAKE not in self.cluster.os_hvp["mocked_os"]

  def testDefaultOsHvp(self):

    os_hvp = {"mocked_os": constants.HVC_DEFAULTS.copy()}

    self.cluster.os_hvp = {"mocked_os": {}}

    op = opcodes.OpClusterSetParams(os_hvp=os_hvp)

    self.ExecOpCode(op)

    self.assertEqual(os_hvp, self.cluster.os_hvp)

  def testOsparams(self):

    osparams = {

      "mocked_os": {

        "param1": "value1",

        "param2": None

      },

      "other_os": {

        "param1": None

      }

    }

    self.cluster.osparams = {"other_os": {"param1": "value1"}}

    op = opcodes.OpClusterSetParams(osparams=osparams)

    self.ExecOpCode(op)

    self.assertEqual({"mocked_os": {"param1": "value1"}}, self.cluster.osparams)

  def testEnabledHypervisors(self):

    enabled_hypervisors = [constants.HT_XEN_HVM, constants.HT_XEN_PVM]

    op = opcodes.OpClusterSetParams(enabled_hypervisors=enabled_hypervisors)

    self.ExecOpCode(op)

    self.assertEqual(enabled_hypervisors, self.cluster.enabled_hypervisors)

  def testEnabledHypervisorsWithoutHypervisorParams(self):