def setUp(self):
self.session = mock.Mock()
self.argument = CliInputJSONArgument(self.session)
# Create the various forms the data could come in. The two main forms
# are as a string and or as a path to a file.
self.input_json = '{"A": "foo", "B": "bar"}'
# Make a temporary file
self.temp_dir = tempfile.mkdtemp()
self.temp_file = os.path.join(self.temp_dir, 'foo.json')
with open(self.temp_file, 'w') as f:
f.write(self.input_json)
def setUp(self):
self.session = mock.Mock()
self.argument = CliInputJSONArgument(self.session)
# Create the various forms the data could come in. The two main forms
# are as a string and or as a path to a file.
self.input_json = '{"A": "foo", "B": "bar"}'
# Make a temporary file
self.temp_dir = tempfile.mkdtemp()
self.temp_file = os.path.join(self.temp_dir, 'foo.json')
with open(self.temp_file, 'w') as f:
f.write(self.input_json)
class TestCliInputJSONArgument(unittest.TestCase):
def setUp(self):
self.session = mock.Mock()
self.argument = CliInputJSONArgument(self.session)
# Create the various forms the data could come in. The two main forms
# are as a string and or as a path to a file.
self.input_json = '{"A": "foo", "B": "bar"}'
# Make a temporary file
self.temp_dir = tempfile.mkdtemp()
self.temp_file = os.path.join(self.temp_dir, 'foo.json')
with open(self.temp_file, 'w') as f:
f.write(self.input_json)
def tearDown(self):
shutil.rmtree(self.temp_dir)
def test_register_argument_action(self):
register_args = self.session.register.call_args_list
self.assertEqual(register_args[0][0][0], 'calling-command.*')
self.assertEqual(register_args[0][0][1],
self.argument.add_to_call_parameters)
def test_add_to_call_parameters_no_file(self):
parsed_args = mock.Mock()
# Make the value a JSON string
parsed_args.cli_input_json = self.input_json
call_parameters = {}
self.argument.add_to_call_parameters(
service_operation=None, call_parameters=call_parameters,
parsed_args=parsed_args, parsed_globals=None
)
self.assertEqual(call_parameters, {'A': 'foo', 'B': 'bar'})
def test_add_to_call_parameters_with_file(self):
parsed_args = mock.Mock()
# Make the value a file with JSON located inside.
parsed_args.cli_input_json = 'file://' + self.temp_file
call_parameters = {}
self.argument.add_to_call_parameters(
service_operation=None, call_parameters=call_parameters,
parsed_args=parsed_args, parsed_globals=None
)
self.assertEqual(call_parameters, {'A': 'foo', 'B': 'bar'})
def test_add_to_call_parameters_bad_json(self):
parsed_args = mock.Mock()
# Create a bad JSON input
parsed_args.cli_input_json = self.input_json + ','
call_parameters = {}
with self.assertRaises(ParamError):
self.argument.add_to_call_parameters(
service_operation=None, call_parameters=call_parameters,
parsed_args=parsed_args, parsed_globals=None
)
def test_add_to_call_parameters_no_clobber(self):
parsed_args = mock.Mock()
parsed_args.cli_input_json = self.input_json
# The value for ``A`` should not be clobbered by the input JSON
call_parameters = {'A': 'baz'}
self.argument.add_to_call_parameters(
service_operation=None, call_parameters=call_parameters,
parsed_args=parsed_args, parsed_globals=None
)
self.assertEqual(call_parameters, {'A': 'baz', 'B': 'bar'})
def test_no_add_to_call_parameters(self):
parsed_args = mock.Mock()
parsed_args.cli_input_json = None
call_parameters = {'A': 'baz'}
self.argument.add_to_call_parameters(
service_operation=None, call_parameters=call_parameters,
parsed_args=parsed_args, parsed_globals=None
)
# Nothing should have been added to the call parameters because
# ``cli_input_json`` is not in the ``parsed_args``
self.assertEqual(call_parameters, {'A': 'baz'})
class TestCliInputJSONArgument(unittest.TestCase):
def setUp(self):
self.session = mock.Mock()
self.argument = CliInputJSONArgument(self.session)
# Create the various forms the data could come in. The two main forms
# are as a string and or as a path to a file.
self.input_json = '{"A": "foo", "B": "bar"}'
# Make a temporary file
self.temp_dir = tempfile.mkdtemp()
self.temp_file = os.path.join(self.temp_dir, 'foo.json')
with open(self.temp_file, 'w') as f:
f.write(self.input_json)
def tearDown(self):
shutil.rmtree(self.temp_dir)
def test_register_argument_action(self):
register_args = self.session.register.call_args_list
self.assertEqual(register_args[0][0][0], 'calling-command')
self.assertEqual(register_args[0][0][1],
self.argument.add_to_call_parameters)
def test_add_to_call_parameters_no_file(self):
parsed_args = mock.Mock()
# Make the value a JSON string
parsed_args.cli_input_json = self.input_json
call_parameters = {}
self.argument.add_to_call_parameters(
service_operation=None, call_parameters=call_parameters,
parsed_args=parsed_args, parsed_globals=None
)
self.assertEqual(call_parameters, {'A': 'foo', 'B': 'bar'})
def test_add_to_call_parameters_with_file(self):
parsed_args = mock.Mock()
# Make the value a file with JSON located inside.
parsed_args.cli_input_json = 'file://' + self.temp_file
call_parameters = {}
self.argument.add_to_call_parameters(
service_operation=None, call_parameters=call_parameters,
parsed_args=parsed_args, parsed_globals=None
)
self.assertEqual(call_parameters, {'A': 'foo', 'B': 'bar'})
def test_add_to_call_parameters_bad_json(self):
parsed_args = mock.Mock()
# Create a bad JSON input
parsed_args.cli_input_json = self.input_json + ','
call_parameters = {}
with self.assertRaises(ParamError):
self.argument.add_to_call_parameters(
service_operation=None, call_parameters=call_parameters,
parsed_args=parsed_args, parsed_globals=None
)
def test_add_to_call_parameters_no_clobber(self):
parsed_args = mock.Mock()
parsed_args.cli_input_json = self.input_json
# The value for ``A`` should not be clobbered by the input JSON
call_parameters = {'A': 'baz'}
self.argument.add_to_call_parameters(
service_operation=None, call_parameters=call_parameters,
parsed_args=parsed_args, parsed_globals=None
)
self.assertEqual(call_parameters, {'A': 'baz', 'B': 'bar'})
def test_no_add_to_call_parameters(self):
parsed_args = mock.Mock()
parsed_args.cli_input_json = None
call_parameters = {'A': 'baz'}
self.argument.add_to_call_parameters(
service_operation=None, call_parameters=call_parameters,
parsed_args=parsed_args, parsed_globals=None
)
# Nothing should have been added to the call parameters because
# ``cli_input_json`` is not in the ``parsed_args``
self.assertEqual(call_parameters, {'A': 'baz'})
