LGTM On Tue, Nov 24, 2015 at 3:15 PM 'Helga Velroyen' via ganeti-devel < [email protected]> wrote:
> This patch adds a unit test for bulk-removing potential > master candidates. > > Signed-off-by: Helga Velroyen <[email protected]> > --- > test/py/ganeti.backend_unittest.py | 31 +++++++++++++++++++++++++++++++ > 1 file changed, 31 insertions(+) > > diff --git a/test/py/ganeti.backend_unittest.py b/test/py/ > ganeti.backend_unittest.py > index 484348f..c551639 100755 > --- a/test/py/ganeti.backend_unittest.py > +++ b/test/py/ganeti.backend_unittest.py > @@ -1423,6 +1423,37 @@ class > TestAddRemoveGenerateNodeSshKey(testutils.GanetiTestCase): > self.assertEqual(1, > len(self._ssh_file_manager.GetAuthorizedKeysOfNode(node_name))) > > + def testRemovePotentialMasterCandidateBulk(self): > + node_list = [] > + key_map = {} > + for node_name, (node_uuid, node_key, _, _, _) in \ > + self._ssh_file_manager.GetAllPurePotentialMasterCandidates()[:3]: > + node_list.append(backend.SshRemoveNodeInfo(uuid=node_uuid, > + name=node_name, > + > from_authorized_keys=False, > + from_public_keys=True, > + > clear_authorized_keys=True, > + clear_public_keys=True)) > + key_map[node_name] = node_key > + > + backend.RemoveNodeSshKeyBulk(node_list, > + self._master_candidate_uuids, > + self._potential_master_candidates, > + pub_key_file=self._pub_key_file, > + ssconf_store=self._ssconf_mock, > + noded_cert_file=self.noded_cert_file, > + run_cmd_fn=self._run_cmd_mock) > + > + for node_info in node_list: > + self._ssh_file_manager.AssertNoNodeHasPublicKey( > + node_info.uuid, key_map[node_info.name]) > + self._ssh_file_manager.AssertNodeSetOnlyHasAuthorizedKey( > + [node_info.name], key_map[node_info.name]) > + self.assertEqual(0, > + len(self._ssh_file_manager.GetPublicKeysOfNode(node_info.name > ))) > + self.assertEqual(1, > + len(self._ssh_file_manager.GetAuthorizedKeysOfNode( > node_info.name))) > + > def testRemoveNormalNode(self): > node_name, node_info = self._ssh_file_manager.GetAllNormalNodes()[0] > > -- > 2.6.0.rc2.230.g3dd15c0 > > -- Lisa Velden Software Engineer [email protected] Google Germany GmbH Dienerstraße 12 80331 München Geschäftsführer: Matthew Scott Sucherman, Paul Terence Manicle Registergericht und -nummer: Hamburg, HRB 86891 Sitz der Gesellschaft: Hamburg
