def test_file_exists_negative(self, mock_access):
validator = validation.file_exists('param')
mock_access.return_value = False
result = validator(param='/tmp/bah')
mock_access.assert_called_with('/tmp/bah', os.R_OK)
self.assertFalse(result.is_valid)
class VMTasks(nova_utils.NovaScenario, vm_utils.VMScenario):
def __init__(self, *args, **kwargs):
super(VMTasks, self).__init__(*args, **kwargs)
@types.set(image=types.ImageResourceType, flavor=types.FlavorResourceType)
@validation.add(validation.image_valid_on_flavor("flavor", "image"))
@validation.add(validation.file_exists("script"))
@validation.add(
validation.number("port",
minval=1,
maxval=65535,
nullable=True,
integer_only=True))
@validation.add(
validation.external_network_exists("floating_network",
"use_floatingip"))
@base.scenario(context={
"cleanup": ["nova"],
"keypair": {},
"allow_ssh": {}
})
@validation.required_services(consts.Service.NOVA)
def boot_runcommand_delete(self,
image,
flavor,
script,
interpreter,
username,
fixed_network="private",
floating_network="public",
ip_version=4,
port=22,
use_floatingip=True,
**kwargs):
"""Boot server, run a script that outputs JSON, delete server.
:param script: script to run on the server, must output JSON mapping
metric names to values. See sample script below.
:param interpreter: The shell interpreter to use when running script
:param username: User to SSH to instance as
:param fixed_network: Network where instance is part of
:param floating_network: External network used to get floating ip from
:param ip_version: Version of ip protocol to use for connection
:param port: Port to use for SSH connection
:param use_floatingip: Whether to associate a floating ip for
connection
:returns: Dictionary containing two keys, data and errors. Data is JSON
data output by the script. Errors is raw data from the
script's standard error stream.
Example Script in doc/samples/tasks/support/instance_dd_test.sh
"""
server = None
floating_ip = None
try:
server = self._boot_server(
self._generate_random_name("rally_novaserver_"),
image,
flavor,
key_name='rally_ssh_key',
**kwargs)
self.check_network(server, fixed_network)
fixed_ip = [
ip for ip in server.addresses[fixed_network]
if ip["version"] == ip_version
][0]["addr"]
if use_floatingip:
floating_ip = self._create_floating_ip(floating_network)
self._associate_floating_ip(server, floating_ip)
server_ip = floating_ip.ip
else:
server_ip = fixed_ip
code, out, err = self.run_command(server_ip, port, username,
interpreter, script)
if code:
raise exceptions.ScriptError("Error running script %(script)s."
"Error %(code)s: %(error)s" % {
"script": script,
"code": code,
"error": err
})
try:
out = json.loads(out)
except ValueError as e:
raise exceptions.ScriptError(
"Script %(script)s did not output valid JSON: %(error)s" %
{
"script": script,
"error": str(e)
})
# Always try to free resources
finally:
if use_floatingip:
self._release_server_floating_ip(server, floating_ip)
if server:
self._delete_server(server)
return {"data": out, "errors": err}
def _release_server_floating_ip(self, server, floating_ip):
"""Release a floating ip associated to a server.
This method check that the given floating ip is associated with the
specified server and tries to dissociate it.
Once dissociated, release the floating ip to reintegrate
it to the pool of available ips.
:param server: The server to dissociate the floating ip from
:param floating_ip: The floating ip to release
"""
if floating_ip and server:
if self.check_ip_address(floating_ip)(server):
self._dissociate_floating_ip(server, floating_ip)
if floating_ip:
self._delete_floating_ip(floating_ip)
def test_file_exists(self, mock_access):
validator = validation.file_exists('param')
result = validator(param='/tmp/foo')
mock_access.assert_called_once_with('/tmp/foo', os.R_OK)
self.assertTrue(result.is_valid)
def test_file_exists(self, mock_access):
validator = validation.file_exists('param')
result = validator(param='/tmp/foo')
mock_access.assert_called_once_with('/tmp/foo', os.R_OK)
self.assertTrue(result.is_valid)
def test_file_exists_negative(self, mock_access):
validator = validation.file_exists('param')
mock_access.return_value = False
result = validator(param='/tmp/bah')
mock_access.assert_called_with('/tmp/bah', os.R_OK)
self.assertFalse(result.is_valid)
def test_file_exists(self, mock_access):
validator = validation.file_exists('param')
config = {"args": {"param": "/tmp/foo"}}
result = validator(config, None, None)
mock_access.assert_called_once_with('/tmp/foo', os.R_OK)
self.assertTrue(result.is_valid)
def test_file_exists_negative(self, mock_access):
validator = validation.file_exists('param')
config = {"args": {"param": "/tmp/bah"}}
result = validator(config, None, None)
mock_access.assert_called_with('/tmp/bah', os.R_OK)
self.assertFalse(result.is_valid)
class VMTasks(nova_utils.NovaScenario, vm_utils.VMScenario):
def __init__(self, *args, **kwargs):
super(VMTasks, self).__init__(*args, **kwargs)
@types.set(image=types.ImageResourceType, flavor=types.FlavorResourceType)
@valid.add_validator(valid.image_valid_on_flavor("flavor", "image"))
@valid.add_validator(valid.file_exists("script"))
@valid.add_validator(
valid.number("port",
minval=1,
maxval=65535,
nullable=True,
integer_only=True))
@base.scenario(context={
"cleanup": ["nova"],
"keypair": {},
"allow_ssh": {}
})
def boot_runcommand_delete(self,
image,
flavor,
script,
interpreter,
network='private',
username='******',
ip_version=4,
port=22,
**kwargs):
"""Boot server, run a script that outputs JSON, delete server.
Parameters:
script: script to run on the server, must output JSON mapping metric
names to values. See sample script below.
network: Network to choose address to connect to instance from
username: User to SSH to instance as
ip_version: Version of ip protocol to use for connection
returns: Dictionary containing two keys, data and errors. Data is JSON
data output by the script. Errors is raw data from the
script's standard error stream.
Example Script in doc/samples/support/instance_dd_test.sh
"""
server = self._boot_server(
self._generate_random_name("rally_novaserver_"),
image,
flavor,
key_name='rally_ssh_key',
**kwargs)
code, out, err = self.run_command(server, username, network, port,
ip_version, interpreter, script)
if code:
LOG.error(
_("Error running script on instance via SSH. "
"Error: %s") % err)
try:
out = json.loads(out)
except ValueError:
LOG.warning(_("Script %s did not output valid JSON.") % script)
self._delete_server(server)
LOG.debug("Output streams from in-instance script execution: "
"stdout: %(stdout)s, stderr: $(stderr)s" %
dict(stdout=out, stderr=err))
return {"data": out, "errors": err}
def test_file_exists_negative(self, mock_access):
validator = validation.file_exists("param")
result = validator(param="/tmp/bah")
mock_access.assert_called_with("/tmp/bah", os.R_OK)
self.assertFalse(result.is_valid)
def test_file_exists(self, mock_access):
validator = validation.file_exists("param")
result = validator(param="/tmp/foo")
mock_access.assert_called_once_with("/tmp/foo", os.R_OK)
self.assertTrue(result.is_valid)
def test_file_exists_negative(self, mock_access):
validator = validation.file_exists('param')
config = {"args": {"param": "/tmp/bah"}}
result = validator(config, None, None)
mock_access.assert_called_with('/tmp/bah', os.R_OK)
self.assertFalse(result.is_valid)
def test_file_exists(self, mock_access):
validator = validation.file_exists('param')
config = {"args": {"param": "/tmp/foo"}}
result = validator(config, None, None)
mock_access.assert_called_once_with('/tmp/foo', os.R_OK)
self.assertTrue(result.is_valid)
