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)
def test_80_run(self):
ctxt_agent = CtxtAgent("/tmp/gen_data.json")
ctxt_agent.logger = self.logger
ctxt_agent.contextualize_vm = MagicMock()
ctxt_agent.contextualize_vm.return_value = {'SSH_WAIT': True, 'OK': True, 'CHANGE_CREDS': True, 'basic': True}
with open("/tmp/gen_data.json", "w+") as f:
json.dump(self.gen_general_conf(), f)
with open("/tmp/vm_data.json", "w+") as f:
json.dump(self.gen_vm_conf(["basic"]), f)
res = ctxt_agent.run("/tmp/vm_data.json")
self.assertTrue(res)