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)