def _validate_and_populate_local_config(self):
"""Validates local job configuration and populates local task config with default values.
Returns a TaskConfig populated with default values.
"""
resp = self._scheduler.populateJobConfig(self._config.job())
self._check_and_log_response(resp)
return get_populated_task_config(resp)
def _validate_and_populate_local_config(self):
"""Validates local job configuration and populates local task config with default values.
Returns a TaskConfig populated with default values.
"""
resp = self._scheduler.populateJobConfig(self._config.job())
self._check_and_log_response(resp)
return get_populated_task_config(resp)
def execute(self, context):
config = context.get_job_config(context.options.jobspec,
context.options.config_file)
if context.options.rename_from is not None:
cluster = context.options.rename_from.cluster
role = context.options.rename_from.role
env = context.options.rename_from.environment
name = context.options.rename_from.name
else:
cluster = config.cluster()
role = config.role()
env = config.environment()
name = config.name()
api = context.get_api(cluster)
resp = api.query(
api.build_query(role, name, env=env, statuses=ACTIVE_STATES))
context.log_response_and_raise(
resp,
err_code=EXIT_INVALID_PARAMETER,
err_msg="Could not find job to diff against")
if resp.result.scheduleStatusResult.tasks is None:
context.print_err("No tasks found for job %s" %
context.options.jobspec)
return EXIT_COMMAND_FAILURE
else:
remote_tasks = [
t.assignedTask.task
for t in resp.result.scheduleStatusResult.tasks
]
resp = api.populate_job_config(config)
context.log_response_and_raise(resp,
err_code=EXIT_INVALID_CONFIGURATION,
err_msg="Error loading configuration")
# Deepcopy is important here as tasks will be modified for printing.
local_tasks = [
deepcopy(get_populated_task_config(resp))
for _ in range(config.instances())
]
diff_program = os.environ.get("DIFF_VIEWER", "diff")
with NamedTemporaryFile() as local:
self.dump_tasks(local_tasks, local)
with NamedTemporaryFile() as remote:
self.dump_tasks(remote_tasks, remote)
result = subprocess.call(
[diff_program, remote.name, local.name])
# Unlike most commands, diff doesn't return zero on success; it returns
# 1 when a successful diff is non-empty.
if result not in (0, 1):
raise context.CommandError(EXIT_COMMAND_FAILURE,
"Error running diff command")
else:
return EXIT_OK
def execute(self, context):
config = context.get_job_config(context.options.jobspec, context.options.config_file)
if context.options.rename_from is not None:
cluster = context.options.rename_from.cluster
role = context.options.rename_from.role
env = context.options.rename_from.environment
name = context.options.rename_from.name
else:
cluster = config.cluster()
role = config.role()
env = config.environment()
name = config.name()
api = context.get_api(cluster)
resp = api.query(api.build_query(role, name, env=env, statuses=ACTIVE_STATES))
context.log_response_and_raise(resp, err_code=EXIT_INVALID_PARAMETER,
err_msg="Could not find job to diff against")
if resp.result.scheduleStatusResult.tasks is None:
context.print_err("No tasks found for job %s" % context.options.jobspec)
return EXIT_COMMAND_FAILURE
else:
remote_tasks = [t.assignedTask.task for t in resp.result.scheduleStatusResult.tasks]
resp = api.populate_job_config(config)
context.log_response_and_raise(resp, err_code=EXIT_INVALID_CONFIGURATION,
err_msg="Error loading configuration")
# Deepcopy is important here as tasks will be modified for printing.
local_tasks = [deepcopy(get_populated_task_config(resp)) for _ in range(config.instances())]
diff_program = os.environ.get("DIFF_VIEWER", "diff")
with NamedTemporaryFile() as local:
self.dump_tasks(local_tasks, local)
with NamedTemporaryFile() as remote:
self.dump_tasks(remote_tasks, remote)
result = subprocess.call([diff_program, remote.name, local.name])
# Unlike most commands, diff doesn't return zero on success; it returns
# 1 when a successful diff is non-empty.
if result not in (0, 1):
raise context.CommandError(EXIT_COMMAND_FAILURE, "Error running diff command")
else:
return EXIT_OK
def test_get_populated_task_config_deprecated_set(self):
config = TaskConfig()
resp = Response(responseCode=ResponseCode.OK,
result=Result(populateJobResult=PopulateJobResult(
populatedDEPRECATED=set([config]))))
assert config == base.get_populated_task_config(resp)
def test_get_populated_task_config_set(self):
config = TaskConfig()
resp = Response(responseCode=ResponseCode.OK,
result=Result(populateJobResult=PopulateJobResult(
taskConfig=config)))
assert config == base.get_populated_task_config(resp)
def test_get_populated_task_config_deprecated_set(self):
config = TaskConfig()
resp = Response(responseCode=ResponseCode.OK, result=Result(populateJobResult=PopulateJobResult(
populatedDEPRECATED=set([config]))))
assert config == base.get_populated_task_config(resp)
def test_get_populated_task_config_set(self):
config = TaskConfig()
resp = Response(responseCode=ResponseCode.OK, result=Result(populateJobResult=PopulateJobResult(
taskConfig=config)))
assert config == base.get_populated_task_config(resp)