def test_task_impl(self, mock_exec_cmd):
args = {
'key': 'test_key',
}
mock_exec_cmd.return_value = {'result': {'value': 'expected'}}
# run
task = St2KeyGet()
result = task.task(args)
# assert
self.assertEquals(result, {'result': {'value': 'expected'}})
mock_exec_cmd.assert_called_with(
['st2', 'key', 'get', '--json', 'test_key'], 'get key')
def test_task_impl_convert_false(self, mock_exec_cmd):
args = {
'key': 'test_key',
'convert': False,
}
mock_exec_cmd.return_value = {'result': {'value': '["a", "b", "c"]'}}
# run
task = St2KeyGet()
result = task.task(args)
# assert
self.assertEquals(result, {'result': {'value': '["a", "b", "c"]'}})
mock_exec_cmd.assert_called_with(
['st2', 'key', 'get', '--json', 'test_key'], 'get key')
def test_convert_result_from_json_no_result(self):
res = {'value': '["a", "b", "c"]'}
convert = True
task = St2KeyGet()
result = task.convert_result_from_json(res, convert)
self.assertEquals(result, {'value': '["a", "b", "c"]'})
def test_convert_result_from_json_result_string_no_value(self):
res = {'result': 'string with non-json data'}
convert = True
task = St2KeyGet()
result = task.convert_result_from_json(res, convert)
self.assertEquals(result, {'result': 'string with non-json data'})
def test_convert_result_from_json_no_value(self):
res = {'result': {'blah': 'xxx'}}
convert = True
task = St2KeyGet()
result = task.convert_result_from_json(res, convert)
self.assertEquals(result, {'result': {'blah': 'xxx'}})
def test_convert_result_from_json_invalid_json(self):
res = {'result': {'value': '{"a": "b'}}
convert = True
task = St2KeyGet()
result = task.convert_result_from_json(res, convert)
self.assertEquals(result, {'result': {'value': '{"a": "b'}})
def test_convert_result_from_json_false(self):
res = {'result': {'value': '{"a": "b"}'}}
convert = False
task = St2KeyGet()
result = task.convert_result_from_json(res, convert)
self.assertEquals(result, {'result': {'value': '{"a": "b"}'}})
def test_init(self):
task = St2KeyGet()
self.assertIsInstance(task, St2TaskBase)