def test_args_rollback_to(self):
"""Test parsing args with all valid values for --rollback-to"""
args = [
self.service, self.key, self.env, "--rollback-to", self.value,
"--sr", "{}".format(self.service_registry_file)
]
context = ef_version.handle_args_and_set_context(args)
self.assertEqual(context.rollback_to, self.value)
self.assertEqual(context.env, self.env)
self.assertEqual(context.service_name, self.service_name)
def test_args_get(self):
"""Test parsing args with all valid values for get"""
args = [
self.service, self.key, self.env, "--get", "--sr",
"{}".format(self.service_registry_file)
]
context = ef_version.handle_args_and_set_context(args)
self.assertEqual(context.env, self.env)
self.assertEqual(context.service_name, self.service_name)
self.assertEqual(context.service_registry.filespec,
self.service_registry_file)
def test_args_get_force_env_full_env_not_account_scoped(self):
"""Test parsing args with all valid values for get and add --env_full flag"""
args = [
self.service, self.key, self.env, "--get", "--sr",
"{}".format(self.service_registry_file), "--force_env_full"
]
context = ef_version.handle_args_and_set_context(args)
self.assertEqual(context.env, self.env)
self.assertEqual(context.service_name, self.service_name)
self.assertEqual(context.service_registry.filespec,
self.service_registry_file)
def test_args_set(self):
"""Test parsing args with all valid values for set"""
args = [
self.service, self.key, self.env, "--set", self.value,
"--location", self.location, "--build", self.build_number,
"--commit_hash", self.commit_hash, "--sr",
"{}".format(self.service_registry_file)
]
context = ef_version.handle_args_and_set_context(args)
self.assertEqual(context.build_number, self.build_number)
self.assertEqual(context.commit_hash, self.commit_hash)
self.assertEqual(context.env, self.env)
self.assertEqual(context.location, self.location)
self.assertEqual(context.service_name, self.service_name)
self.assertEqual(context.service_registry.filespec,
self.service_registry_file)
self.assertEqual(context.value, self.value)
def test_args_invalid_env(self):
"""Verify that an invalid environment arg raises an exception"""
args = [self.service, self.key, "invalid_env"]
with self.assertRaises(SystemExit):
ef_version.handle_args_and_set_context(args)
