Synnefo » snf-ganeti » ganeti-local

from ganeti import compat

from ganeti import rpc

class TestClusterVerifyFiles(unittest.TestCase):

  @staticmethod

  def _FakeErrorIf(errors, cond, ecode, item, msg, *args, **kwargs):

    assert ((ecode == cmdlib.LUClusterVerifyGroup.ENODEFILECHECK and

             ht.TNonEmptyString(item)) or

            (ecode == cmdlib.LUClusterVerifyGroup.ECLUSTERFILECHECK and

             item is None))

    if args:

      msg = msg % args

    if cond:

      errors.append((item, msg))

  _VerifyFiles = cmdlib.LUClusterVerifyGroup._VerifyFiles

  def test(self):

    errors = []

    master_name = "master.example.com"

    nodeinfo = [

      objects.Node(name=master_name, offline=False),

      objects.Node(name="node2.example.com", offline=False),

      objects.Node(name="node3.example.com", master_candidate=True),

      objects.Node(name="node4.example.com", offline=False),

      objects.Node(name="nodata.example.com"),

      objects.Node(name="offline.example.com", offline=True),

      ]

    cluster = objects.Cluster(modify_etc_hosts=True,

                              enabled_hypervisors=[constants.HT_XEN_HVM])

    files_all = set([

      constants.CLUSTER_DOMAIN_SECRET_FILE,

      constants.RAPI_CERT_FILE,

      ])

    files_all_opt = set([

      constants.RAPI_USERS_FILE,

      ])

    files_mc = set([

      constants.CLUSTER_CONF_FILE,

      ])

    files_vm = set()

    nvinfo = {

      master_name: rpc.RpcResult(data=(True, {

        constants.NV_FILELIST: {

          constants.CLUSTER_CONF_FILE: "82314f897f38b35f9dab2f7c6b1593e0",

          constants.RAPI_CERT_FILE: "babbce8f387bc082228e544a2146fee4",

          constants.CLUSTER_DOMAIN_SECRET_FILE: "cds-47b5b3f19202936bb4",

        }})),

      "node2.example.com": rpc.RpcResult(data=(True, {

        constants.NV_FILELIST: {

          constants.RAPI_CERT_FILE: "97f0356500e866387f4b84233848cc4a",

          }

        })),

      "node3.example.com": rpc.RpcResult(data=(True, {

        constants.NV_FILELIST: {

          constants.RAPI_CERT_FILE: "97f0356500e866387f4b84233848cc4a",

          constants.CLUSTER_DOMAIN_SECRET_FILE: "cds-47b5b3f19202936bb4",

          }

        })),

      "node4.example.com": rpc.RpcResult(data=(True, {

        constants.NV_FILELIST: {

          constants.RAPI_CERT_FILE: "97f0356500e866387f4b84233848cc4a",

          constants.CLUSTER_CONF_FILE: "conf-a6d4b13e407867f7a7b4f0f232a8f527",

          constants.CLUSTER_DOMAIN_SECRET_FILE: "cds-47b5b3f19202936bb4",

          constants.RAPI_USERS_FILE: "rapiusers-ea3271e8d810ef3",

          }

        })),

      "nodata.example.com": rpc.RpcResult(data=(True, {})),

      "offline.example.com": rpc.RpcResult(offline=True),

      }

    assert set(nvinfo.keys()) == set(map(operator.attrgetter("name"), nodeinfo))

    self._VerifyFiles(compat.partial(self._FakeErrorIf, errors), nodeinfo,

                      master_name, nvinfo,

                      (files_all, files_all_opt, files_mc, files_vm))

    self.assertEqual(sorted(errors), sorted([

      (None, ("File %s found with 2 different checksums (variant 1 on"

              " node2.example.com, node3.example.com, node4.example.com;"

              " variant 2 on master.example.com)" % constants.RAPI_CERT_FILE)),

      (None, ("File %s is missing from node(s) node2.example.com" %

              constants.CLUSTER_DOMAIN_SECRET_FILE)),

      (None, ("File %s should not exist on node(s) node4.example.com" %

              constants.CLUSTER_CONF_FILE)),

      (None, ("File %s is missing from node(s) node3.example.com" %

              constants.CLUSTER_CONF_FILE)),

      (None, ("File %s found with 2 different checksums (variant 1 on"

              " master.example.com; variant 2 on node4.example.com)" %

              constants.CLUSTER_CONF_FILE)),

      (None, ("File %s is optional, but it must exist on all or no nodes (not"

              " found on master.example.com, node2.example.com,"

              " node3.example.com)" % constants.RAPI_USERS_FILE)),

      ("nodata.example.com", "Node did not return file checksum data"),

      ]))