def test_get_shell_cmds_funcs_with_params(self):
self.service.config = {
'generic_command_config': {
'shell_commands': [
{
'name': 'test_1',
'command': 'echo {} {}',
'allow_params': True
},
]
}
}
params = {
"shell_params": ["Hello", "world!"]
}
table = get_shell_commands_from_config(self.service.config)
func_1 = table['test_1']
result_1 = asyncio.get_event_loop().run_until_complete(
func_1(params)
)
self.assertEqual(
result_1,
{'stdout': 'Hello world!\n', 'stderr': '', 'returncode': 0}
)
pass
def test_get_shell_cmds_funcs(self):
self.service.config = {
'generic_command_config': {
'shell_commands': [
{
'name': 'test_1',
'command': 'echo'
},
],
},
}
table = get_shell_commands_from_config(self.service.config)
func_1 = table['test_1']
result_1 = asyncio.get_event_loop().run_until_complete(
func_1(mock.Mock()), )
self.assertEqual(
result_1,
{
'stdout': '\n',
'stderr': '',
'returncode': 0
},
)
def test_get_shell_cmds_missing_fields(self):
self.service.config = {
'generic_command_config': {
'shell_commands': [
{'name': 'test_1'},
{'command': '/bin/false'},
]
}
}
table = get_shell_commands_from_config(self.service.config)
self.assertEqual(table, {})
def test_get_shell_cmds(self):
self.service.config = {
'generic_command_config': {
'shell_commands': [
{'name': 'test_1', 'command': '/bin/true'},
{'name': 'test_2', 'command': '/bin/false'},
]
}
}
table = get_shell_commands_from_config(self.service.config)
self.assertIn('test_1', table)
self.assertIn('test_2', table)
def test_get_shell_cmds_nonexistent_func(self):
self.service.config = {
'generic_command_config': {
'shell_commands': [
{'name': 'test_1', 'command': 'nonexistent/command foo'},
]
}
}
table = get_shell_commands_from_config(self.service.config)
func_1 = table['test_1']
try:
asyncio.get_event_loop().run_until_complete(
func_1(mock.Mock())
)
self.fail('An exception should have been raised')
except FileNotFoundError:
pass
def test_get_shell_cmds_missing_shell_commands_list(self):
self.service.config = {'generic_command_config': {}}
table = get_shell_commands_from_config(self.service.config)
self.assertEqual(table, {})
def test_get_shell_cmds_missing_config(self):
self.service.config = {}
table = get_shell_commands_from_config(self.service.config)
self.assertEqual(table, {})
