#
# Copyright (C) 2011 Google Inc.
+# All rights reserved.
#
-# 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.
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are
+# met:
#
-# 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.
+# 1. Redistributions of source code must retain the above copyright notice,
+# this list of conditions and the following disclaimer.
#
-# 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.
+# 2. Redistributions in binary form must reproduce the above copyright
+# notice, this list of conditions and the following disclaimer in the
+# documentation and/or other materials provided with the distribution.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS
+# IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
+# TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
+# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR
+# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
+# EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
+# PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
+# PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
+# LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
+# NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
"""Script for testing ganeti.client.gnt_cluster"""
import unittest
import optparse
+import os
+import shutil
+import tempfile
from ganeti import errors
from ganeti.client import gnt_cluster
from ganeti import utils
from ganeti import compat
from ganeti import constants
+from ganeti import ssh
+from ganeti import cli
import mock
import testutils
return self._OFF_EXITCODE
def _Test(self, *args, **kwargs):
- defaults = dict(cl=NotImplemented, _on_fn=NotImplemented,
+ defaults = dict(qcl=NotImplemented, _on_fn=NotImplemented,
_off_fn=NotImplemented,
_stdout_fn=lambda *args: None,
_stderr_fn=lambda *args: None)
off_fn = compat.partial(self._Off, ["node1.example.com"])
- result = self._Test(opts, [], cl=client, qcl=client, _off_fn=off_fn,
+ result = self._Test(opts, [], qcl=client, _off_fn=off_fn,
_confirm_fn=confirm_fn)
if force or confirm_result:
self.assertEqual(result, self._OFF_EXITCODE)
self.assertFalse(inst_map)
return self._ON_EXITCODE
- result = self._Test(opts, [], cl=client, qcl=client, _on_fn=_On,
+ result = self._Test(opts, [], qcl=client, _on_fn=_On,
_confirm_fn=self._ConfirmForce)
self.assertEqual(result, self._ON_EXITCODE)
client = _ClientForEpo(NotImplemented, [
("node1.example.com", True, [], [], True, False),
])
- result = self._Test(opts, [], cl=client, qcl=client,
- _confirm_fn=self._ConfirmForce)
+ result = self._Test(opts, [], qcl=client, _confirm_fn=self._ConfirmForce)
self.assertEqual(result, constants.EXIT_FAILURE)
opts = mock.Mock()
opts.drbd_helper = None
self.disableDrbd()
- helper = gnt_cluster._InitDrbdHelper(opts, self.enabled_disk_templates)
+ helper = gnt_cluster._InitDrbdHelper(opts, self.enabled_disk_templates,
+ feedback_fn=mock.Mock())
self.assertEquals(None, helper)
def testNoDrbdHelper(self):
opts = mock.Mock()
self.disableDrbd()
opts.drbd_helper = "/bin/true"
- helper = gnt_cluster._InitDrbdHelper(opts, self.enabled_disk_templates)
+ helper = gnt_cluster._InitDrbdHelper(opts, self.enabled_disk_templates,
+ feedback_fn=mock.Mock())
self.assertEquals(opts.drbd_helper, helper)
def testDrbdHelperNone(self):
opts = mock.Mock()
self.enableDrbd()
opts.drbd_helper = None
- helper = gnt_cluster._InitDrbdHelper(opts, self.enabled_disk_templates)
+ helper = gnt_cluster._InitDrbdHelper(opts, self.enabled_disk_templates,
+ feedback_fn=mock.Mock())
self.assertEquals(constants.DEFAULT_DRBD_HELPER, helper)
def testDrbdHelperEmpty(self):
self.enableDrbd()
opts.drbd_helper = ''
self.assertRaises(errors.OpPrereqError, gnt_cluster._InitDrbdHelper, opts,
- self.enabled_disk_templates)
+ self.enabled_disk_templates, feedback_fn=mock.Mock())
def testDrbdHelper(self):
opts = mock.Mock()
self.enableDrbd()
opts.drbd_helper = "/bin/true"
- helper = gnt_cluster._InitDrbdHelper(opts, self.enabled_disk_templates)
+ helper = gnt_cluster._InitDrbdHelper(opts, self.enabled_disk_templates,
+ feedback_fn=mock.Mock())
self.assertEquals(opts.drbd_helper, helper)
self.assertEquals(opts.drbd_helper, helper)
+class TestBuildGanetiPubKeys(testutils.GanetiTestCase):
+
+ _SOME_KEY_DICT = {"rsa": "key_rsa",
+ "dsa": "key_dsa"}
+ _MASTER_NODE_NAME = "master_node"
+ _MASTER_NODE_UUID = "master_uuid"
+ _NUM_NODES = 2 # excluding master node
+ _ONLINE_NODE_NAMES = ["node%s_name" % i for i in range(_NUM_NODES)]
+ _ONLINE_NODE_UUIDS = ["node%s_uuid" % i for i in range(_NUM_NODES)]
+ _CLUSTER_NAME = "cluster_name"
+ _PRIV_KEY = "master_private_key"
+ _PUB_KEY = "master_public_key"
+ _AUTH_KEYS = "a\nb\nc"
+
+ def _setUpFakeKeys(self):
+ os.makedirs(os.path.join(self.tmpdir, ".ssh"))
+
+ for key_type in ["rsa", "dsa"]:
+ self.priv_filename = os.path.join(self.tmpdir, ".ssh", "id_%s" % key_type)
+ utils.WriteFile(self.priv_filename, data=self._PRIV_KEY)
+
+ self.pub_filename = os.path.join(
+ self.tmpdir, ".ssh", "id_%s.pub" % key_type)
+ utils.WriteFile(self.pub_filename, data=self._PUB_KEY)
+
+ self.auth_filename = os.path.join(self.tmpdir, ".ssh", "authorized_keys")
+ utils.WriteFile(self.auth_filename, data=self._AUTH_KEYS)
+
+ def setUp(self):
+ testutils.GanetiTestCase.setUp(self)
+ self.tmpdir = tempfile.mkdtemp()
+ self.pub_key_filename = os.path.join(self.tmpdir, "ganeti_test_pub_keys")
+ self._setUpFakeKeys()
+
+ self._ssh_read_remote_ssh_pub_keys_patcher = testutils \
+ .patch_object(ssh, "ReadRemoteSshPubKeys")
+ self._ssh_read_remote_ssh_pub_keys_mock = \
+ self._ssh_read_remote_ssh_pub_keys_patcher.start()
+ self._ssh_read_remote_ssh_pub_keys_mock.return_value = self._SOME_KEY_DICT
+
+ self.mock_cl = mock.Mock()
+ self.mock_cl.QueryConfigValues = mock.Mock()
+ self.mock_cl.QueryConfigValues.return_value = \
+ (self._CLUSTER_NAME, self._MASTER_NODE_NAME)
+
+ self._get_online_nodes_mock = mock.Mock()
+ self._get_online_nodes_mock.return_value = \
+ self._ONLINE_NODE_NAMES
+
+ self._get_nodes_ssh_ports_mock = mock.Mock()
+ self._get_nodes_ssh_ports_mock.return_value = \
+ [22 for i in range(self._NUM_NODES + 1)]
+
+ self._get_node_uuids_mock = mock.Mock()
+ self._get_node_uuids_mock.return_value = \
+ self._ONLINE_NODE_UUIDS + [self._MASTER_NODE_UUID]
+
+ self._options = mock.Mock()
+ self._options.ssh_key_check = False
+
+ def _GetTempHomedir(self, _):
+ return self.tmpdir
+
+ def tearDown(self):
+ super(testutils.GanetiTestCase, self).tearDown()
+ shutil.rmtree(self.tmpdir)
+ self._ssh_read_remote_ssh_pub_keys_patcher.stop()
+
+ def testNewPubKeyFile(self):
+ gnt_cluster._BuildGanetiPubKeys(
+ self._options,
+ pub_key_file=self.pub_key_filename,
+ cl=self.mock_cl,
+ get_online_nodes_fn=self._get_online_nodes_mock,
+ get_nodes_ssh_ports_fn=self._get_nodes_ssh_ports_mock,
+ get_node_uuids_fn=self._get_node_uuids_mock,
+ homedir_fn=self._GetTempHomedir)
+ key_file_result = utils.ReadFile(self.pub_key_filename)
+ for node_uuid in self._ONLINE_NODE_UUIDS + [self._MASTER_NODE_UUID]:
+ self.assertTrue(node_uuid in key_file_result)
+ self.assertTrue(self._PUB_KEY in key_file_result)
+
+ def testOverridePubKeyFile(self):
+ fd = open(self.pub_key_filename, "w")
+ fd.write("Pink Bunny")
+ fd.close()
+ gnt_cluster._BuildGanetiPubKeys(
+ self._options,
+ pub_key_file=self.pub_key_filename,
+ cl=self.mock_cl,
+ get_online_nodes_fn=self._get_online_nodes_mock,
+ get_nodes_ssh_ports_fn=self._get_nodes_ssh_ports_mock,
+ get_node_uuids_fn=self._get_node_uuids_mock,
+ homedir_fn=self._GetTempHomedir)
+ self.assertFalse("Pink Bunny" in self.pub_key_filename)
+
+
if __name__ == "__main__":
testutils.GanetiTestProgram()
projects / ganeti-github.git / blobdiff