Merge branch 'stable-2.15' into stable-2.16
[ganeti-github.git] / test / py / ganeti.client.gnt_cluster_unittest.py
index a087a54..38bda23 100755 (executable)
@@ -384,7 +384,9 @@ class TestBuildGanetiPubKeys(testutils.GanetiTestCase):
   _CLUSTER_NAME = "cluster_name"
   _PRIV_KEY = "master_private_key"
   _PUB_KEY = "master_public_key"
+  _MODIFY_SSH_SETUP = True
   _AUTH_KEYS = "a\nb\nc"
+  _SSH_KEY_TYPE = "dsa"
 
   def _setUpFakeKeys(self):
     os.makedirs(os.path.join(self.tmpdir, ".ssh"))
@@ -415,7 +417,8 @@ class TestBuildGanetiPubKeys(testutils.GanetiTestCase):
     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._CLUSTER_NAME, self._MASTER_NODE_NAME, self._MODIFY_SSH_SETUP,
+       self._SSH_KEY_TYPE)
 
     self._get_online_nodes_mock = mock.Mock()
     self._get_online_nodes_mock.return_value = \