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)