def test_rebootvm(self, server_proxy, requests): """ Test rebootvm operation """ proxy = MagicMock() proxy.RebootVM.return_value = (True, "") server_proxy.return_value = proxy options = MagicMock() options.auth_file = get_abs_path("../../auth.dat") options.restapi = None parser = MagicMock() out = StringIO() oldstdout = sys.stdout sys.stdout = out res = main("rebootvm", options, ["infid", "vmid"], parser) self.assertEquals(res, True) output = out.getvalue().strip() self.assertIn("VM successfully rebooted", output) out = StringIO() sys.stdout = out options.xmlrpc = None options.restapi = "https://localhost:8800" requests.side_effect = self.get_response res = main("rebootvm", options, ["infid", "vmid"], parser) self.assertEquals(res, True) output = out.getvalue().strip() self.assertIn("VM successfully rebooted", output) sys.stdout = oldstdout
def test_getversion(self, server_proxy, requests): """ Test getversion operation """ proxy = MagicMock() proxy.GetVersion.return_value = (True, "1.0") server_proxy.return_value = proxy options = MagicMock() options.auth_file = get_abs_path("../../auth.dat") options.restapi = None parser = MagicMock() out = StringIO() oldstdout = sys.stdout sys.stdout = out res = main("getversion", options, [], parser) self.assertEquals(res, True) output = out.getvalue().strip() self.assertIn("1.0", output) out = StringIO() sys.stdout = out options.xmlrpc = None options.restapi = "https://localhost:8800" requests.side_effect = self.get_response res = main("getversion", options, [], parser) self.assertEquals(res, True) output = out.getvalue().strip() self.assertIn("1.0", output) sys.stdout = oldstdout
def test_getinfo(self, server_proxy, requests): """ Test getinfo operation """ proxy = MagicMock() proxy.GetVMInfo.return_value = (True, "radltest") proxy.GetInfrastructureInfo.return_value = (True, ["vm1"]) server_proxy.return_value = proxy options = MagicMock() options.auth_file = get_abs_path("../../auth.dat") options.restapi = None parser = MagicMock() out = StringIO() oldstdout = sys.stdout sys.stdout = out res = main("getinfo", options, ["infid"], parser) self.assertEquals(res, True) output = out.getvalue().strip() self.assertIn("Info about VM with ID: vm1\nradltest", output) out = StringIO() sys.stdout = out options.xmlrpc = None options.restapi = "https://localhost:8800" requests.side_effect = self.get_response res = main("getinfo", options, ["infid"], parser) self.assertEquals(res, True) output = out.getvalue().strip() self.assertIn("Info about VM with ID: vm1\nradltest", output) sys.stdout = oldstdout
def test_getcontmsg(self, server_proxy, requests): """ Test getcontmsg operation """ proxy = MagicMock() proxy.GetInfrastructureContMsg.return_value = (True, "contmsg") server_proxy.return_value = proxy options = MagicMock() options.auth_file = get_abs_path("../../auth.dat") options.restapi = None parser = MagicMock() out = StringIO() oldstdout = sys.stdout sys.stdout = out res = main("getcontmsg", options, ["infid"], parser) self.assertEquals(res, True) output = out.getvalue().strip() self.assertIn("Msg Contextualizator: \n\ncontmsg", output) out = StringIO() sys.stdout = out options.xmlrpc = None options.restapi = "https://localhost:8800" requests.side_effect = self.get_response res = main("getcontmsg", options, ["infid"], parser) self.assertEquals(res, True) output = out.getvalue().strip() self.assertIn("Msg Contextualizator: \n\ncontmsg", output) sys.stdout = oldstdout
def test_removeresource(self, server_proxy, requests): """ Test removeresource operation """ proxy = MagicMock() proxy.RemoveResource.return_value = (True, ["1", "2"]) server_proxy.return_value = proxy options = MagicMock() options.restapi = None options.auth_file = get_abs_path("../../auth.dat") parser = MagicMock() out = StringIO() oldstdout = sys.stdout sys.stdout = out res = main("removeresource", options, ["infid", "1,2"], parser) self.assertEquals(res, True) output = out.getvalue().strip() self.assertIn("Resources with IDs: ['1', '2'] successfully deleted.", output) out = StringIO() sys.stdout = out options.xmlrpc = None options.restapi = "https://localhost:8800" requests.side_effect = self.get_response res = main("removeresource", options, ["infid", "1"], parser) self.assertEquals(res, True) output = out.getvalue().strip() self.assertIn("Resources with IDs: 1 successfully deleted.", output) sys.stdout = oldstdout
def test_import(self, server_proxy, requests): """ Test import operation """ proxy = MagicMock() proxy.ImportInfrastructure.return_value = (True, "newinfid") server_proxy.return_value = proxy options = MagicMock() options.auth_file = get_abs_path("../../auth.dat") options.restapi = None parser = MagicMock() out = StringIO() oldstdout = sys.stdout sys.stdout = out res = main("import", options, [get_abs_path("../files/test.radl")], parser) self.assertEquals(res, True) output = out.getvalue().strip() self.assertIn("New Inf: newinfid", output) out = StringIO() sys.stdout = out options.xmlrpc = None options.restapi = "https://localhost:8800" requests.side_effect = self.get_response res = main("import", options, [get_abs_path("../files/test.radl")], parser) self.assertEquals(res, True) output = out.getvalue().strip() self.assertIn("New Inf: newinfid", output) sys.stdout = oldstdout
def test_list(self, server_proxy, requests): """ Test list operation """ proxy = MagicMock() proxy.GetInfrastructureList.return_value = (True, ["inf1", "inf2"]) server_proxy.return_value = proxy options = MagicMock() options.auth_file = get_abs_path("../../auth.dat") options.xmlrpc = "https://localhost:8899" options.restapi = None options.verify = False parser = MagicMock() out = StringIO() oldstdout = sys.stdout sys.stdout = out res = main("list", options, [], parser) output = out.getvalue().strip() self.assertEquals(res, True) self.assertIn("IDs: \n inf1\n inf2", output) sys.stdout = oldstdout out = StringIO() sys.stdout = out options.xmlrpc = None options.restapi = "https://localhost:8800" requests.side_effect = self.get_response res = main("list", options, [], parser) self.assertEquals(res, True) output = out.getvalue().strip() self.assertIn("IDs: \n inf1\n inf2", output) sys.stdout = oldstdout
def test_getstate(self, server_proxy, requests): """ Test getstate operation """ proxy = MagicMock() proxy.GetInfrastructureState.return_value = (True, {"state": "running", "vm_states": {"vm1": "running"}}) server_proxy.return_value = proxy options = MagicMock() options.auth_file = get_abs_path("../../auth.dat") options.restapi = None parser = MagicMock() out = StringIO() oldstdout = sys.stdout sys.stdout = out res = main("getstate", options, ["infid"], parser) self.assertEquals(res, True) output = out.getvalue().strip() self.assertIn("The infrastructure is in state: running\nVM ID: vm1 is in state: running.", output) out = StringIO() sys.stdout = out options.xmlrpc = None options.restapi = "https://localhost:8800" requests.side_effect = self.get_response res = main("getstate", options, ["infid"], parser) self.assertEquals(res, True) output = out.getvalue().strip() self.assertIn("The infrastructure is in state: running\nVM ID: vm1 is in state: running.", output) sys.stdout = oldstdout
def test_addresource(self, server_proxy, requests): """ Test addresource operation """ proxy = MagicMock() proxy.AddResource.return_value = (True, ["1"]) server_proxy.return_value = proxy options = MagicMock() options.auth_file = get_abs_path("../../auth.dat") options.restapi = None parser = MagicMock() out = StringIO() oldstdout = sys.stdout sys.stdout = out res = main("addresource", options, ["infid", get_abs_path("../files/test.radl")], parser) self.assertEquals(res, True) output = out.getvalue().strip() self.assertIn("Resources with IDs: 1 successfully added.", output) out = StringIO() sys.stdout = out options.xmlrpc = None options.restapi = "https://localhost:8800" requests.side_effect = self.get_response res = main("addresource", options, ["infid", get_abs_path("../files/test.radl")], parser) self.assertEquals(res, True) output = out.getvalue().strip() self.assertIn("Resources with IDs: 1 successfully added.", output) sys.stdout = oldstdout
def test_create(self, server_proxy, requests): """ Test create operation """ proxy = MagicMock() proxy.CreateInfrastructure.return_value = (True, "inf1") server_proxy.return_value = proxy options = MagicMock() options.auth_file = get_abs_path("../../auth.dat") options.restapi = None parser = MagicMock() out = StringIO() oldstdout = sys.stdout sys.stdout = out res = main("create", options, [get_abs_path("../files/test.radl")], parser) self.assertEquals(res, True) output = out.getvalue().strip() self.assertIn("Infrastructure successfully created with ID: inf1", output) sys.stdout = oldstdout out = StringIO() sys.stdout = out options.xmlrpc = None options.restapi = "https://localhost:8800" requests.side_effect = self.get_response res = main("create", options, [get_abs_path("../files/test.radl")], parser) self.assertEquals(res, True) output = out.getvalue().strip() self.assertIn("Infrastructure successfully created with ID: inf1", output) sys.stdout = oldstdout
def test_sshvm(self, server_proxy): """ Test sshvm operation """ proxy = MagicMock() radl = open(get_abs_path("../files/test.radl"), 'r').read() proxy.GetVMInfo.return_value = (True, radl) server_proxy.return_value = proxy options = MagicMock() options.auth_file = get_abs_path("../../auth.dat") options.restapi = None parser = MagicMock() out = StringIO() oldstdout = sys.stdout oldstderr = sys.stderr sys.stdout = out sys.stderr = out res = main("sshvm", options, ["infid", "vmid", "1"], parser) self.assertEquals(res, True) output = out.getvalue().strip() self.assertIn( "sshpass -pyoyoyo ssh -p 1022 -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no " "[email protected]", output) sys.stdout = oldstdout sys.stderr = oldstderr sys.stdout = out sys.stderr = out res = main("ssh", options, ["infid", "1"], parser) self.assertEquals(res, True) output = out.getvalue().strip() self.assertIn( "sshpass -pyoyoyo ssh -p 1022 -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no " "[email protected]", output) sys.stdout = oldstdout sys.stderr = oldstderr # test proxy host radl = open(get_abs_path("../files/test_proxy.radl"), 'r').read() proxy.GetVMInfo.return_value = (True, radl) out = StringIO() oldstdout = sys.stdout oldstderr = sys.stderr sys.stdout = out sys.stderr = out res = main("sshvm", options, ["infid", "vmid", "1"], parser) self.assertEquals(res, True) output = out.getvalue().strip() self.assertIn( "sshpass -pyoyoyo ssh -p 22 -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no " "[email protected] -o ProxyCommand=sshpass -p passwd ssh -W %h:%p -p 22" " -o StrictHostKeyChecking=no [email protected]", output) sys.stdout = oldstdout sys.stderr = oldstderr
def test_sshvm_key(self, server_proxy): """ Test sshvm operation """ proxy = MagicMock() radl = open(get_abs_path("../files/test_priv.radl"), 'r').read() proxy.GetVMInfo.return_value = (True, radl) server_proxy.return_value = proxy options = MagicMock() options.auth_file = get_abs_path("../../auth.dat") options.restapi = None parser = MagicMock() out = StringIO() oldstdout = sys.stdout oldstderr = sys.stderr sys.stdout = out sys.stderr = out res = main("sshvm", options, ["infid", "vmid", "1"], parser) self.assertEquals(res, True) output = out.getvalue().strip() self.assertIn("ssh -p 1022 -i /tmp/", output) self.assertIn( " -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no [email protected]", output) sys.stdout = oldstdout sys.stderr = oldstderr
def test_sshvm_key(self, server_proxy): """ Test sshvm operation """ proxy = MagicMock() radl = open(get_abs_path("../files/test_priv.radl"), 'r').read() proxy.GetVMInfo.return_value = (True, radl) server_proxy.return_value = proxy options = MagicMock() options.auth_file = get_abs_path("../../auth.dat") options.restapi = None parser = MagicMock() out = StringIO() oldstdout = sys.stdout oldstderr = sys.stderr sys.stdout = out sys.stderr = out res = main("sshvm", options, ["infid", "vmid", "1"], parser) self.assertEquals(res, True) output = out.getvalue().strip() self.assertIn("ssh -p 1022 -i /tmp/", output) self.assertIn(" -o UserKnownHostsFile=/dev/null -o StrictHostKeyChecking=no [email protected]", output) sys.stdout = oldstdout sys.stderr = oldstderr
def test_getstate(self, server_proxy, requests): """ Test getstate operation """ proxy = MagicMock() proxy.GetInfrastructureState.return_value = (True, { "state": "running", "vm_states": { "vm1": "running" } }) server_proxy.return_value = proxy options = MagicMock() options.auth_file = get_abs_path("../../auth.dat") options.restapi = None parser = MagicMock() out = StringIO() oldstdout = sys.stdout sys.stdout = out res = main("getstate", options, ["infid"], parser) self.assertEquals(res, True) output = out.getvalue().strip() self.assertIn( "The infrastructure is in state: running\nVM ID: vm1 is in state: running.", output) out = StringIO() sys.stdout = out options.xmlrpc = None options.restapi = "https://localhost:8800" requests.side_effect = self.get_response res = main("getstate", options, ["infid"], parser) self.assertEquals(res, True) output = out.getvalue().strip() self.assertIn( "The infrastructure is in state: running\nVM ID: vm1 is in state: running.", output) sys.stdout = oldstdout
def test_getoutputs(self, server_proxy, requests): """ Test getoutputs operation """ options = MagicMock() options.auth_file = get_abs_path("../../auth.dat") parser = MagicMock() oldstdout = sys.stdout out = StringIO() sys.stdout = out options.xmlrpc = None options.restapi = "https://localhost:8800" requests.side_effect = self.get_response res = main("getoutputs", options, ["infid", "vmid"], parser) self.assertEquals(res, True) output = out.getvalue().strip() self.assertIn("The infrastructure outputs:\n", output) self.assertIn("\noutput2 = value2", output) self.assertIn("\noutput1 = value1", output) sys.stdout = oldstdout