def test_60_changeVMCredentials(self, winrm_session, execute, execute_timeout): ctxt_agent = CtxtAgent(None) ctxt_agent.logger = self.logger execute.return_value = "", "", 0 execute_timeout.return_value = "", "", 0 vm = self.gen_vm_data() vm['ctxt_ip'] = "10.0.0.1" vm['ctxt_port'] = 22 res = ctxt_agent.changeVMCredentials(vm, None) self.assertTrue(res) vm = self.gen_vm_data() del vm['new_passwd'] vm['new_public_key'] = "new_public_key" vm['new_private_key'] = "new_private_key" vm['ctxt_ip'] = "10.0.0.1" vm['ctxt_port'] = 22 res = ctxt_agent.changeVMCredentials(vm, None) self.assertTrue(res) session = MagicMock() req = MagicMock req.status_code = 0 session.run_cmd.return_value = req winrm_session.return_value = session vm = self.gen_vm_data("windows") res = ctxt_agent.changeVMCredentials(vm, None) self.assertTrue(res)
def test_70_contextualize_vm(self, test_connectivity): ctxt_agent = CtxtAgent(None) ctxt_agent.logger = self.logger ctxt_agent.changeVMCredentials = MagicMock() ctxt_agent.changeVMCredentials.return_value = True ctxt_agent.LaunchAnsiblePlaybook = MagicMock() queue = MagicMock() queue.get.return_value = None, 0, None ctxt_agent.LaunchAnsiblePlaybook.return_value = (MagicMock(), queue) ctxt_agent.wait_winrm_access = MagicMock() ctxt_agent.wait_winrm_access.return_value = True ctxt_agent.wait_ssh_access = MagicMock() ctxt_agent.wait_ssh_access.return_value = 'init' ctxt_agent.removeRequiretty = MagicMock() ctxt_agent.removeRequiretty.return_value = True res = ctxt_agent.contextualize_vm(self.gen_general_conf(), self.gen_vm_conf(["basic"])) expected_res = { 'SSH_WAIT': True, 'OK': True, 'CHANGE_CREDS': True, 'basic': True } self.assertEqual(res, expected_res) res = ctxt_agent.contextualize_vm(self.gen_general_conf(), self.gen_vm_conf(["main", "front"])) expected_res = {'OK': True, 'front': True, 'main': True} self.assertEqual(res, expected_res)