def test_user_argument(self):
# User is the same as logged user, no sudo should be used
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = LOGGED_USER_USERNAME
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
self.assertEqual(command, '/tmp/foo.sh')
# User is different, sudo should be used
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = '******'
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
self.assertEqual(command, 'sudo -E -u mauser -- bash -c /tmp/foo.sh')
# sudo is used, it doesn't matter what user is specified since the
# command should run as root
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = True
kwargs['user'] = '******'
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
self.assertEqual(command, 'sudo -E -- bash -c /tmp/foo.sh')
def test_user_argument(self):
# User is the same as logged user, no sudo should be used
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = LOGGED_USER_USERNAME
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
self.assertEqual(command, '/tmp/foo.sh')
# User is different, sudo should be used
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = '******'
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
self.assertEqual(command, 'sudo -E -u mauser -- bash -c /tmp/foo.sh')
# sudo is used, it doesn't matter what user is specified since the
# command should run as root
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = True
kwargs['user'] = '******'
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
self.assertEqual(command, 'sudo -E -- bash -c /tmp/foo.sh')
def test_named_parameter_escaping(self):
# no sudo
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = LOGGED_USER_USERNAME
kwargs['named_args'] = OrderedDict([
('key1', 'value foo bar'),
('key2', 'value "bar" foo'),
('key3', 'date ; whoami'),
('key4', '"date ; whoami"'),
])
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
expected = self._get_fixture('escaping_test_command_1.txt')
self.assertEqual(command, expected)
# sudo
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = True
kwargs['user'] = LOGGED_USER_USERNAME
kwargs['named_args'] = OrderedDict([
('key1', 'value foo bar'),
('key2', 'value "bar" foo'),
('key3', 'date ; whoami'),
('key4', '"date ; whoami"'),
])
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
expected = self._get_fixture('escaping_test_command_2.txt')
self.assertEqual(command, expected)
def test_named_parameter_escaping(self):
# no sudo
kwargs = copy.deepcopy(self._base_kwargs)
kwargs["sudo"] = False
kwargs["user"] = LOGGED_USER_USERNAME
kwargs["named_args"] = OrderedDict([
("key1", "value foo bar"),
("key2", 'value "bar" foo'),
("key3", "date ; whoami"),
("key4", '"date ; whoami"'),
])
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
expected = self._get_fixture("escaping_test_command_1.txt")
self.assertEqual(command, expected)
# sudo
kwargs = copy.deepcopy(self._base_kwargs)
kwargs["sudo"] = True
kwargs["user"] = LOGGED_USER_USERNAME
kwargs["named_args"] = OrderedDict([
("key1", "value foo bar"),
("key2", 'value "bar" foo'),
("key3", "date ; whoami"),
("key4", '"date ; whoami"'),
])
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
expected = self._get_fixture("escaping_test_command_2.txt")
self.assertEqual(command, expected)
def test_named_parameter_escaping(self):
# no sudo
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = LOGGED_USER_USERNAME
kwargs['named_args'] = {
'key1': 'value foo bar',
'key2': 'value "bar" foo',
'key3': 'date ; whoami',
'key4': '"date ; whoami"',
}
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
expected = self._get_fixture('escaping_test_command_1.txt')
self.assertEqual(command, expected)
# sudo
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = True
kwargs['user'] = LOGGED_USER_USERNAME
kwargs['named_args'] = {
'key1': 'value foo bar',
'key2': 'value "bar" foo',
'key3': 'date ; whoami',
'key4': '"date ; whoami"',
}
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
expected = self._get_fixture('escaping_test_command_2.txt')
self.assertEqual(command, expected)
def test_user_argument(self):
# User is the same as logged user, no sudo should be used
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = LOGGED_USER_USERNAME
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
self.assertEqual(command, '/tmp/foo.sh')
# User is different, sudo should be used
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = '******'
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
self.assertEqual(command,
'sudo -E -H -u mauser -- bash -c /tmp/foo.sh')
# sudo with password
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = '******'
kwargs['sudo_password'] = '******'
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
expected_command = 'sudo -S -E -H -u mauser -- bash -c /tmp/foo.sh'
self.assertEqual(command, expected_command)
# complex sudo password which needs escaping
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = '******'
kwargs['sudo_password'] = '******'as"sss'
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
expected_command = ('sudo -S -E -H '
'-u mauser -- bash -c /tmp/foo.sh')
self.assertEqual(command, expected_command)
command = action.get_sanitized_full_command_string()
expected_command = ('echo -e \'%s\n\' | sudo -S -E -H '
'-u mauser -- bash -c /tmp/foo.sh' %
(MASKED_ATTRIBUTE_VALUE))
self.assertEqual(command, expected_command)
# sudo is used, it doesn't matter what user is specified since the
# command should run as root
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = True
kwargs['user'] = '******'
kwargs['sudo_password'] = '******'
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
expected_command = 'sudo -S -E -- bash -c /tmp/foo.sh'
self.assertEqual(command, expected_command)
def test_user_argument(self):
# User is the same as logged user, no sudo should be used
kwargs = copy.deepcopy(self._base_kwargs)
kwargs["sudo"] = False
kwargs["user"] = LOGGED_USER_USERNAME
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
self.assertEqual(command, "/tmp/foo.sh")
# User is different, sudo should be used
kwargs = copy.deepcopy(self._base_kwargs)
kwargs["sudo"] = False
kwargs["user"] = "******"
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
self.assertEqual(command,
"sudo -E -H -u mauser -- bash -c /tmp/foo.sh")
# sudo with password
kwargs = copy.deepcopy(self._base_kwargs)
kwargs["sudo"] = False
kwargs["user"] = "******"
kwargs["sudo_password"] = "******"
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
expected_command = "sudo -S -E -H -u mauser -- bash -c /tmp/foo.sh"
self.assertEqual(command, expected_command)
# complex sudo password which needs escaping
kwargs = copy.deepcopy(self._base_kwargs)
kwargs["sudo"] = False
kwargs["user"] = "******"
kwargs["sudo_password"] = "******"sss"
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
expected_command = "sudo -S -E -H " "-u mauser -- bash -c /tmp/foo.sh"
self.assertEqual(command, expected_command)
command = action.get_sanitized_full_command_string()
expected_command = ("echo -e '%s\n' | sudo -S -E -H "
"-u mauser -- bash -c /tmp/foo.sh" %
(MASKED_ATTRIBUTE_VALUE))
self.assertEqual(command, expected_command)
# sudo is used, it doesn't matter what user is specified since the
# command should run as root
kwargs = copy.deepcopy(self._base_kwargs)
kwargs["sudo"] = True
kwargs["user"] = "******"
kwargs["sudo_password"] = "******"
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
expected_command = "sudo -S -E -- bash -c /tmp/foo.sh"
self.assertEqual(command, expected_command)
def test_user_argument(self):
# User is the same as logged user, no sudo should be used
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = LOGGED_USER_USERNAME
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
self.assertEqual(command, '/tmp/foo.sh')
# User is different, sudo should be used
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = '******'
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
self.assertEqual(command, 'sudo -E -H -u mauser -- bash -c /tmp/foo.sh')
# sudo with password
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = '******'
kwargs['sudo_password'] = '******'
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
expected_command = 'sudo -S -E -H -u mauser -- bash -c /tmp/foo.sh'
self.assertEqual(command, expected_command)
# complex sudo password which needs escaping
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = '******'
kwargs['sudo_password'] = '******'as"sss'
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
expected_command = ('sudo -S -E -H '
'-u mauser -- bash -c /tmp/foo.sh')
self.assertEqual(command, expected_command)
command = action.get_sanitized_full_command_string()
expected_command = ('echo -e \'%s\n\' | sudo -S -E -H '
'-u mauser -- bash -c /tmp/foo.sh' % (MASKED_ATTRIBUTE_VALUE))
self.assertEqual(command, expected_command)
# sudo is used, it doesn't matter what user is specified since the
# command should run as root
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = True
kwargs['user'] = '******'
kwargs['sudo_password'] = '******'
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
expected_command = 'sudo -S -E -- bash -c /tmp/foo.sh'
self.assertEqual(command, expected_command)
def test_various_ascii_parameters(self):
kwargs = copy.deepcopy(self._base_kwargs)
kwargs["sudo"] = False
kwargs["user"] = LOGGED_USER_USERNAME
kwargs["named_args"] = {"foo1": "bar1", "foo2": "bar2"}
kwargs["positional_args"] = []
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
self.assertEqual(command, "/tmp/foo.sh foo1=bar1 foo2=bar2")
def test_various_ascii_parameters(self):
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = LOGGED_USER_USERNAME
kwargs['named_args'] = {'foo1': 'bar1', 'foo2': 'bar2'}
kwargs['positional_args'] = []
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
self.assertEqual(command, u"/tmp/foo.sh foo1=bar1 foo2=bar2")
def test_various_ascii_parameters(self):
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = LOGGED_USER_USERNAME
kwargs['named_args'] = {'foo1': 'bar1', 'foo2': 'bar2'}
kwargs['positional_args'] = []
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
self.assertEqual(command, u"/tmp/foo.sh foo1=bar1 foo2=bar2")
def test_unicode_parameter_specifing(self):
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = LOGGED_USER_USERNAME
kwargs['named_args'] = {u'ｆｏｏ': u'ｂａｒ'}
kwargs['positional_args'] = []
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
self.assertEqual(command, u"/tmp/foo.sh 'ｆｏｏ'='ｂａｒ'")
def test_unicode_parameter_specifing(self):
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = LOGGED_USER_USERNAME
kwargs['named_args'] = {u'ｆｏｏ': u'ｂａｒ'}
kwargs['positional_args'] = []
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
self.assertEqual(command, u"/tmp/foo.sh 'ｆｏｏ'='ｂａｒ'")
def test_unicode_parameter_specifing(self):
kwargs = copy.deepcopy(self._base_kwargs)
kwargs["sudo"] = False
kwargs["user"] = LOGGED_USER_USERNAME
kwargs["named_args"] = {"ｆｏｏ": "ｂａｒ"}
kwargs["positional_args"] = []
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
self.assertEqual(command, "/tmp/foo.sh 'ｆｏｏ'='ｂａｒ'")
def test_command_construction_with_parameters(self):
# same user, named args, no positional args
kwargs = copy.deepcopy(self._base_kwargs)
kwargs["sudo"] = False
kwargs["user"] = LOGGED_USER_USERNAME
kwargs["named_args"] = OrderedDict([("key1", "value1"),
("key2", "value2")])
kwargs["positional_args"] = []
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
self.assertEqual(command, "/tmp/foo.sh key1=value1 key2=value2")
# same user, named args, no positional args, sudo password
kwargs = copy.deepcopy(self._base_kwargs)
kwargs["sudo"] = True
kwargs["sudo_password"] = "******"
kwargs["user"] = LOGGED_USER_USERNAME
kwargs["named_args"] = OrderedDict([("key1", "value1"),
("key2", "value2")])
kwargs["positional_args"] = []
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
expected = "sudo -S -E -- bash -c " "'/tmp/foo.sh key1=value1 key2=value2'"
self.assertEqual(command, expected)
# different user, named args, no positional args
kwargs = copy.deepcopy(self._base_kwargs)
kwargs["sudo"] = False
kwargs["user"] = "******"
kwargs["named_args"] = OrderedDict([("key1", "value1"),
("key2", "value2")])
kwargs["positional_args"] = []
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
expected = (
"sudo -E -H -u mauser -- bash -c '/tmp/foo.sh key1=value1 key2=value2'"
)
self.assertEqual(command, expected)
# different user, named args, no positional args, sudo password
kwargs = copy.deepcopy(self._base_kwargs)
kwargs["sudo"] = False
kwargs["sudo_password"] = "******"
kwargs["user"] = "******"
kwargs["named_args"] = OrderedDict([("key1", "value1"),
("key2", "value2")])
kwargs["positional_args"] = []
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
expected = ("sudo -S -E -H -u mauser -- bash -c "
"'/tmp/foo.sh key1=value1 key2=value2'")
self.assertEqual(command, expected)
# same user, positional args, no named args
kwargs = copy.deepcopy(self._base_kwargs)
kwargs["sudo"] = False
kwargs["user"] = LOGGED_USER_USERNAME
kwargs["named_args"] = {}
kwargs["positional_args"] = [
"ein", "zwei", "drei", "mamma mia", "foo\nbar"
]
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
self.assertEqual(command,
"/tmp/foo.sh ein zwei drei 'mamma mia' 'foo\nbar'")
# different user, named args, positional args
kwargs = copy.deepcopy(self._base_kwargs)
kwargs["sudo"] = False
kwargs["user"] = "******"
kwargs["named_args"] = {}
kwargs["positional_args"] = ["ein", "zwei", "drei", "mamma mia"]
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
ex = ("sudo -E -H -u mauser -- "
"bash -c '/tmp/foo.sh ein zwei drei '\"'\"'mamma mia'\"'\"''")
self.assertEqual(command, ex)
# same user, positional and named args
kwargs = copy.deepcopy(self._base_kwargs)
kwargs["sudo"] = False
kwargs["user"] = LOGGED_USER_USERNAME
kwargs["named_args"] = OrderedDict([("key1", "value1"),
("key2", "value2"),
("key3", "value 3")])
kwargs["positional_args"] = ["ein", "zwei", "drei"]
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
exp = "/tmp/foo.sh key1=value1 key2=value2 key3='value 3' ein zwei drei"
self.assertEqual(command, exp)
# different user, positional and named args
kwargs = copy.deepcopy(self._base_kwargs)
kwargs["sudo"] = False
kwargs["user"] = "******"
kwargs["named_args"] = OrderedDict([("key1", "value1"),
("key2", "value2"),
("key3", "value 3")])
kwargs["positional_args"] = ["ein", "zwei", "drei"]
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
expected = (
"sudo -E -H -u mauser -- bash -c '/tmp/foo.sh key1=value1 key2=value2 "
"key3='\"'\"'value 3'\"'\"' ein zwei drei'")
self.assertEqual(command, expected)
def test_command_construction_correct_default_parameter_values_are_used(self):
runner_parameters = {}
action_db_parameters = {
'project': {
'type': 'string',
'default': 'st2',
'position': 0,
},
'version': {
'type': 'string',
'position': 1,
'required': True
},
'fork': {
'type': 'string',
'position': 2,
'default': 'StackStorm',
},
'branch': {
'type': 'string',
'position': 3,
'default': 'master',
},
'update_mistral': {
'type': 'boolean',
'position': 4,
'default': False
},
'update_changelog': {
'type': 'boolean',
'position': 5,
'default': False
},
'local_repo': {
'type': 'string',
'position': 6,
}
}
context = {}
action_db = ActionDB(pack='dummy', name='action')
runner = LocalShellScriptRunner('id')
runner.runner_parameters = {}
runner.action = action_db
# 1. All default values used
live_action_db_parameters = {
'project': 'st2flow',
'version': '3.0.0',
'fork': 'StackStorm',
'local_repo': '/tmp/repo'
}
runner_params, action_params = param_utils.render_final_params(runner_parameters,
action_db_parameters,
live_action_db_parameters,
context)
self.assertDictEqual(action_params, {
'project': 'st2flow',
'version': '3.0.0',
'fork': 'StackStorm',
'branch': 'master', # default value used
'update_mistral': False, # default value used
'update_changelog': False, # default value used
'local_repo': '/tmp/repo'
})
action_db.parameters = action_db_parameters
positional_args, named_args = runner._get_script_args(action_params)
named_args = runner._transform_named_args(named_args)
shell_script_action = ShellScriptAction(name='dummy', action_exec_id='dummy',
script_local_path_abs='/tmp/local.sh',
named_args=named_args,
positional_args=positional_args)
command_string = shell_script_action.get_full_command_string()
expected = '/tmp/local.sh st2flow 3.0.0 StackStorm master 0 0 /tmp/repo'
self.assertEqual(command_string, expected)
# 2. Some default values used
live_action_db_parameters = {
'project': 'st2web',
'version': '3.1.0',
'fork': 'StackStorm1',
'update_changelog': True,
'local_repo': '/tmp/repob'
}
runner_params, action_params = param_utils.render_final_params(runner_parameters,
action_db_parameters,
live_action_db_parameters,
context)
self.assertDictEqual(action_params, {
'project': 'st2web',
'version': '3.1.0',
'fork': 'StackStorm1',
'branch': 'master', # default value used
'update_mistral': False, # default value used
'update_changelog': True, # default value used
'local_repo': '/tmp/repob'
})
action_db.parameters = action_db_parameters
positional_args, named_args = runner._get_script_args(action_params)
named_args = runner._transform_named_args(named_args)
shell_script_action = ShellScriptAction(name='dummy', action_exec_id='dummy',
script_local_path_abs='/tmp/local.sh',
named_args=named_args,
positional_args=positional_args)
command_string = shell_script_action.get_full_command_string()
expected = '/tmp/local.sh st2web 3.1.0 StackStorm1 master 0 1 /tmp/repob'
self.assertEqual(command_string, expected)
# 3. None is specified for a boolean parameter, should use a default
live_action_db_parameters = {
'project': 'st2rbac',
'version': '3.2.0',
'fork': 'StackStorm2',
'update_changelog': None,
'local_repo': '/tmp/repoc'
}
runner_params, action_params = param_utils.render_final_params(runner_parameters,
action_db_parameters,
live_action_db_parameters,
context)
self.assertDictEqual(action_params, {
'project': 'st2rbac',
'version': '3.2.0',
'fork': 'StackStorm2',
'branch': 'master', # default value used
'update_mistral': False, # default value used
'update_changelog': False, # default value used
'local_repo': '/tmp/repoc'
})
action_db.parameters = action_db_parameters
positional_args, named_args = runner._get_script_args(action_params)
named_args = runner._transform_named_args(named_args)
shell_script_action = ShellScriptAction(name='dummy', action_exec_id='dummy',
script_local_path_abs='/tmp/local.sh',
named_args=named_args,
positional_args=positional_args)
command_string = shell_script_action.get_full_command_string()
expected = '/tmp/local.sh st2rbac 3.2.0 StackStorm2 master 0 0 /tmp/repoc'
self.assertEqual(command_string, expected)
def test_command_construction_correct_default_parameter_values_are_used(self):
runner_parameters = {}
action_db_parameters = {
'project': {
'type': 'string',
'default': 'st2',
'position': 0,
},
'version': {
'type': 'string',
'position': 1,
'required': True
},
'fork': {
'type': 'string',
'position': 2,
'default': 'StackStorm',
},
'branch': {
'type': 'string',
'position': 3,
'default': 'master',
},
'update_changelog': {
'type': 'boolean',
'position': 4,
'default': False
},
'local_repo': {
'type': 'string',
'position': 5,
}
}
context = {}
action_db = ActionDB(pack='dummy', name='action')
runner = LocalShellScriptRunner('id')
runner.runner_parameters = {}
runner.action = action_db
# 1. All default values used
live_action_db_parameters = {
'project': 'st2flow',
'version': '3.0.0',
'fork': 'StackStorm',
'local_repo': '/tmp/repo'
}
runner_params, action_params = param_utils.render_final_params(runner_parameters,
action_db_parameters,
live_action_db_parameters,
context)
self.assertDictEqual(action_params, {
'project': 'st2flow',
'version': '3.0.0',
'fork': 'StackStorm',
'branch': 'master', # default value used
'update_changelog': False, # default value used
'local_repo': '/tmp/repo'
})
action_db.parameters = action_db_parameters
positional_args, named_args = runner._get_script_args(action_params)
named_args = runner._transform_named_args(named_args)
shell_script_action = ShellScriptAction(name='dummy', action_exec_id='dummy',
script_local_path_abs='/tmp/local.sh',
named_args=named_args,
positional_args=positional_args)
command_string = shell_script_action.get_full_command_string()
expected = '/tmp/local.sh st2flow 3.0.0 StackStorm master 0 /tmp/repo'
self.assertEqual(command_string, expected)
# 2. Some default values used
live_action_db_parameters = {
'project': 'st2web',
'version': '3.1.0',
'fork': 'StackStorm1',
'update_changelog': True,
'local_repo': '/tmp/repob'
}
runner_params, action_params = param_utils.render_final_params(runner_parameters,
action_db_parameters,
live_action_db_parameters,
context)
self.assertDictEqual(action_params, {
'project': 'st2web',
'version': '3.1.0',
'fork': 'StackStorm1',
'branch': 'master', # default value used
'update_changelog': True, # default value used
'local_repo': '/tmp/repob'
})
action_db.parameters = action_db_parameters
positional_args, named_args = runner._get_script_args(action_params)
named_args = runner._transform_named_args(named_args)
shell_script_action = ShellScriptAction(name='dummy', action_exec_id='dummy',
script_local_path_abs='/tmp/local.sh',
named_args=named_args,
positional_args=positional_args)
command_string = shell_script_action.get_full_command_string()
expected = '/tmp/local.sh st2web 3.1.0 StackStorm1 master 1 /tmp/repob'
self.assertEqual(command_string, expected)
# 3. None is specified for a boolean parameter, should use a default
live_action_db_parameters = {
'project': 'st2rbac',
'version': '3.2.0',
'fork': 'StackStorm2',
'update_changelog': None,
'local_repo': '/tmp/repoc'
}
runner_params, action_params = param_utils.render_final_params(runner_parameters,
action_db_parameters,
live_action_db_parameters,
context)
self.assertDictEqual(action_params, {
'project': 'st2rbac',
'version': '3.2.0',
'fork': 'StackStorm2',
'branch': 'master', # default value used
'update_changelog': False, # default value used
'local_repo': '/tmp/repoc'
})
action_db.parameters = action_db_parameters
positional_args, named_args = runner._get_script_args(action_params)
named_args = runner._transform_named_args(named_args)
shell_script_action = ShellScriptAction(name='dummy', action_exec_id='dummy',
script_local_path_abs='/tmp/local.sh',
named_args=named_args,
positional_args=positional_args)
command_string = shell_script_action.get_full_command_string()
expected = '/tmp/local.sh st2rbac 3.2.0 StackStorm2 master 0 /tmp/repoc'
self.assertEqual(command_string, expected)
def test_command_construction_with_parameters(self):
# same user, named args, no positional args
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = LOGGED_USER_USERNAME
kwargs['named_args'] = OrderedDict([
('key1', 'value1'),
('key2', 'value2')
])
kwargs['positional_args'] = []
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
self.assertEqual(command, '/tmp/foo.sh key1=value1 key2=value2')
# same user, named args, no positional args, sudo password
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = True
kwargs['sudo_password'] = '******'
kwargs['user'] = LOGGED_USER_USERNAME
kwargs['named_args'] = OrderedDict([
('key1', 'value1'),
('key2', 'value2')
])
kwargs['positional_args'] = []
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
expected = ('sudo -S -E -- bash -c '
'\'/tmp/foo.sh key1=value1 key2=value2\'')
self.assertEqual(command, expected)
# different user, named args, no positional args
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = '******'
kwargs['named_args'] = OrderedDict([
('key1', 'value1'),
('key2', 'value2')
])
kwargs['positional_args'] = []
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
expected = 'sudo -E -H -u mauser -- bash -c \'/tmp/foo.sh key1=value1 key2=value2\''
self.assertEqual(command, expected)
# different user, named args, no positional args, sudo password
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['sudo_password'] = '******'
kwargs['user'] = '******'
kwargs['named_args'] = OrderedDict([
('key1', 'value1'),
('key2', 'value2')
])
kwargs['positional_args'] = []
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
expected = ('sudo -S -E -H -u mauser -- bash -c '
'\'/tmp/foo.sh key1=value1 key2=value2\'')
self.assertEqual(command, expected)
# same user, positional args, no named args
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = LOGGED_USER_USERNAME
kwargs['named_args'] = {}
kwargs['positional_args'] = ['ein', 'zwei', 'drei', 'mamma mia', 'foo\nbar']
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
self.assertEqual(command, '/tmp/foo.sh ein zwei drei \'mamma mia\' \'foo\nbar\'')
# different user, named args, positional args
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = '******'
kwargs['named_args'] = {}
kwargs['positional_args'] = ['ein', 'zwei', 'drei', 'mamma mia']
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
ex = ('sudo -E -H -u mauser -- '
'bash -c \'/tmp/foo.sh ein zwei drei \'"\'"\'mamma mia\'"\'"\'\'')
self.assertEqual(command, ex)
# same user, positional and named args
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = LOGGED_USER_USERNAME
kwargs['named_args'] = OrderedDict([
('key1', 'value1'),
('key2', 'value2'),
('key3', 'value 3')
])
kwargs['positional_args'] = ['ein', 'zwei', 'drei']
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
exp = '/tmp/foo.sh key1=value1 key2=value2 key3=\'value 3\' ein zwei drei'
self.assertEqual(command, exp)
# different user, positional and named args
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = '******'
kwargs['named_args'] = OrderedDict([
('key1', 'value1'),
('key2', 'value2'),
('key3', 'value 3')
])
kwargs['positional_args'] = ['ein', 'zwei', 'drei']
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
expected = ('sudo -E -H -u mauser -- bash -c \'/tmp/foo.sh key1=value1 key2=value2 '
'key3=\'"\'"\'value 3\'"\'"\' ein zwei drei\'')
self.assertEqual(command, expected)
def test_command_construction_with_parameters(self):
# same user, named args, no positional args
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = LOGGED_USER_USERNAME
kwargs['named_args'] = {'key1': 'value1', 'key2': 'value2'}
kwargs['positional_args'] = []
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
self.assertEqual(command, '/tmp/foo.sh key2=value2 key1=value1')
# different user, named args, no positional args
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = '******'
kwargs['named_args'] = {'key1': 'value1', 'key2': 'value2'}
kwargs['positional_args'] = ''
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
expected = 'sudo -E -u mauser -- bash -c \'/tmp/foo.sh key2=value2 key1=value1\''
self.assertEqual(command, expected)
# same user, positional args, no named args
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = LOGGED_USER_USERNAME
kwargs['named_args'] = {}
kwargs['positional_args'] = 'ein zwei drei'
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
self.assertEqual(command, '/tmp/foo.sh ein zwei drei')
# different user, named args, positional args
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = '******'
kwargs['named_args'] = {}
kwargs['positional_args'] = 'ein zwei drei'
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
expected = 'sudo -E -u mauser -- bash -c \'/tmp/foo.sh ein zwei drei\''
self.assertEqual(command, expected)
# same user, positional and named args
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = LOGGED_USER_USERNAME
kwargs['named_args'] = {'key1': 'value1', 'key2': 'value2'}
kwargs['positional_args'] = 'ein zwei drei'
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
self.assertEqual(command, '/tmp/foo.sh key2=value2 key1=value1 ein zwei drei')
# different user, positional and named args
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = '******'
kwargs['named_args'] = {'key1': 'value1', 'key2': 'value2'}
kwargs['positional_args'] = 'ein zwei drei'
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
expected = ('sudo -E -u mauser -- bash -c \'/tmp/foo.sh key2=value2 '
'key1=value1 ein zwei drei\'')
self.assertEqual(command, expected)
def test_command_construction_with_parameters(self):
# same user, named args, no positional args
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = LOGGED_USER_USERNAME
kwargs['named_args'] = {'key1': 'value1', 'key2': 'value2'}
kwargs['positional_args'] = []
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
self.assertEqual(command, '/tmp/foo.sh key2=value2 key1=value1')
# different user, named args, no positional args
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = '******'
kwargs['named_args'] = {'key1': 'value1', 'key2': 'value2'}
kwargs['positional_args'] = ''
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
expected = 'sudo -E -u mauser -- bash -c \'/tmp/foo.sh key2=value2 key1=value1\''
self.assertEqual(command, expected)
# same user, positional args, no named args
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = LOGGED_USER_USERNAME
kwargs['named_args'] = {}
kwargs['positional_args'] = 'ein zwei drei'
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
self.assertEqual(command, '/tmp/foo.sh ein zwei drei')
# different user, named args, positional args
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = '******'
kwargs['named_args'] = {}
kwargs['positional_args'] = 'ein zwei drei'
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
expected = 'sudo -E -u mauser -- bash -c \'/tmp/foo.sh ein zwei drei\''
self.assertEqual(command, expected)
# same user, positional and named args
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = LOGGED_USER_USERNAME
kwargs['named_args'] = {'key1': 'value1', 'key2': 'value2'}
kwargs['positional_args'] = 'ein zwei drei'
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
self.assertEqual(command,
'/tmp/foo.sh key2=value2 key1=value1 ein zwei drei')
# different user, positional and named args
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = '******'
kwargs['named_args'] = {'key1': 'value1', 'key2': 'value2'}
kwargs['positional_args'] = 'ein zwei drei'
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
expected = ('sudo -E -u mauser -- bash -c \'/tmp/foo.sh key2=value2 '
'key1=value1 ein zwei drei\'')
self.assertEqual(command, expected)
def test_command_construction_correct_default_parameter_values_are_used(
self):
runner_parameters = {}
action_db_parameters = {
"project": {
"type": "string",
"default": "st2",
"position": 0,
},
"version": {
"type": "string",
"position": 1,
"required": True
},
"fork": {
"type": "string",
"position": 2,
"default": "StackStorm",
},
"branch": {
"type": "string",
"position": 3,
"default": "master",
},
"update_changelog": {
"type": "boolean",
"position": 4,
"default": False
},
"local_repo": {
"type": "string",
"position": 5,
},
}
context = {}
action_db = ActionDB(pack="dummy", name="action")
runner = LocalShellScriptRunner("id")
runner.runner_parameters = {}
runner.action = action_db
# 1. All default values used
live_action_db_parameters = {
"project": "st2flow",
"version": "3.0.0",
"fork": "StackStorm",
"local_repo": "/tmp/repo",
}
runner_params, action_params = param_utils.render_final_params(
runner_parameters, action_db_parameters, live_action_db_parameters,
context)
self.assertDictEqual(
action_params,
{
"project": "st2flow",
"version": "3.0.0",
"fork": "StackStorm",
"branch": "master", # default value used
"update_changelog": False, # default value used
"local_repo": "/tmp/repo",
},
)
action_db.parameters = action_db_parameters
positional_args, named_args = runner._get_script_args(action_params)
named_args = runner._transform_named_args(named_args)
shell_script_action = ShellScriptAction(
name="dummy",
action_exec_id="dummy",
script_local_path_abs="/tmp/local.sh",
named_args=named_args,
positional_args=positional_args,
)
command_string = shell_script_action.get_full_command_string()
expected = "/tmp/local.sh st2flow 3.0.0 StackStorm master 0 /tmp/repo"
self.assertEqual(command_string, expected)
# 2. Some default values used
live_action_db_parameters = {
"project": "st2web",
"version": "3.1.0",
"fork": "StackStorm1",
"update_changelog": True,
"local_repo": "/tmp/repob",
}
runner_params, action_params = param_utils.render_final_params(
runner_parameters, action_db_parameters, live_action_db_parameters,
context)
self.assertDictEqual(
action_params,
{
"project": "st2web",
"version": "3.1.0",
"fork": "StackStorm1",
"branch": "master", # default value used
"update_changelog": True, # default value used
"local_repo": "/tmp/repob",
},
)
action_db.parameters = action_db_parameters
positional_args, named_args = runner._get_script_args(action_params)
named_args = runner._transform_named_args(named_args)
shell_script_action = ShellScriptAction(
name="dummy",
action_exec_id="dummy",
script_local_path_abs="/tmp/local.sh",
named_args=named_args,
positional_args=positional_args,
)
command_string = shell_script_action.get_full_command_string()
expected = "/tmp/local.sh st2web 3.1.0 StackStorm1 master 1 /tmp/repob"
self.assertEqual(command_string, expected)
# 3. None is specified for a boolean parameter, should use a default
live_action_db_parameters = {
"project": "st2rbac",
"version": "3.2.0",
"fork": "StackStorm2",
"update_changelog": None,
"local_repo": "/tmp/repoc",
}
runner_params, action_params = param_utils.render_final_params(
runner_parameters, action_db_parameters, live_action_db_parameters,
context)
self.assertDictEqual(
action_params,
{
"project": "st2rbac",
"version": "3.2.0",
"fork": "StackStorm2",
"branch": "master", # default value used
"update_changelog": False, # default value used
"local_repo": "/tmp/repoc",
},
)
action_db.parameters = action_db_parameters
positional_args, named_args = runner._get_script_args(action_params)
named_args = runner._transform_named_args(named_args)
shell_script_action = ShellScriptAction(
name="dummy",
action_exec_id="dummy",
script_local_path_abs="/tmp/local.sh",
named_args=named_args,
positional_args=positional_args,
)
command_string = shell_script_action.get_full_command_string()
expected = "/tmp/local.sh st2rbac 3.2.0 StackStorm2 master 0 /tmp/repoc"
self.assertEqual(command_string, expected)
def test_command_construction_with_parameters(self):
# same user, named args, no positional args
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = LOGGED_USER_USERNAME
kwargs['named_args'] = OrderedDict([
('key1', 'value1'),
('key2', 'value2')
])
kwargs['positional_args'] = []
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
self.assertEqual(command, '/tmp/foo.sh key1=value1 key2=value2')
# same user, named args, no positional args, sudo password
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = True
kwargs['sudo_password'] = '******'
kwargs['user'] = LOGGED_USER_USERNAME
kwargs['named_args'] = OrderedDict([
('key1', 'value1'),
('key2', 'value2')
])
kwargs['positional_args'] = []
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
expected = ('sudo -S -E -- bash -c '
'\'/tmp/foo.sh key1=value1 key2=value2\'')
self.assertEqual(command, expected)
# different user, named args, no positional args
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = '******'
kwargs['named_args'] = OrderedDict([
('key1', 'value1'),
('key2', 'value2')
])
kwargs['positional_args'] = []
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
expected = 'sudo -E -H -u mauser -- bash -c \'/tmp/foo.sh key1=value1 key2=value2\''
self.assertEqual(command, expected)
# different user, named args, no positional args, sudo password
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['sudo_password'] = '******'
kwargs['user'] = '******'
kwargs['named_args'] = OrderedDict([
('key1', 'value1'),
('key2', 'value2')
])
kwargs['positional_args'] = []
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
expected = ('sudo -S -E -H -u mauser -- bash -c '
'\'/tmp/foo.sh key1=value1 key2=value2\'')
self.assertEqual(command, expected)
# same user, positional args, no named args
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = LOGGED_USER_USERNAME
kwargs['named_args'] = {}
kwargs['positional_args'] = ['ein', 'zwei', 'drei', 'mamma mia', 'foo\nbar']
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
self.assertEqual(command, '/tmp/foo.sh ein zwei drei \'mamma mia\' \'foo\nbar\'')
# different user, named args, positional args
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = '******'
kwargs['named_args'] = {}
kwargs['positional_args'] = ['ein', 'zwei', 'drei', 'mamma mia']
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
ex = ('sudo -E -H -u mauser -- '
'bash -c \'/tmp/foo.sh ein zwei drei \'"\'"\'mamma mia\'"\'"\'\'')
self.assertEqual(command, ex)
# same user, positional and named args
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = LOGGED_USER_USERNAME
kwargs['named_args'] = OrderedDict([
('key1', 'value1'),
('key2', 'value2'),
('key3', 'value 3')
])
kwargs['positional_args'] = ['ein', 'zwei', 'drei']
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
exp = '/tmp/foo.sh key1=value1 key2=value2 key3=\'value 3\' ein zwei drei'
self.assertEqual(command, exp)
# different user, positional and named args
kwargs = copy.deepcopy(self._base_kwargs)
kwargs['sudo'] = False
kwargs['user'] = '******'
kwargs['named_args'] = OrderedDict([
('key1', 'value1'),
('key2', 'value2'),
('key3', 'value 3')
])
kwargs['positional_args'] = ['ein', 'zwei', 'drei']
action = ShellScriptAction(**kwargs)
command = action.get_full_command_string()
expected = ('sudo -E -H -u mauser -- bash -c \'/tmp/foo.sh key1=value1 key2=value2 '
'key3=\'"\'"\'value 3\'"\'"\' ein zwei drei\'')
self.assertEqual(command, expected)
