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_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_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_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_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_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_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_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_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_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
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
