def from_state(
cls,
state_data,
action_graph,
output_path,
context,
run_node,
):
"""Restore a JobRun from a serialized state."""
pool_repo = node.NodePoolRepository.get_instance()
run_node = pool_repo.get_node(state_data.get('node_name'), run_node)
job_name = state_data['job_name']
job_run = cls(
job_name,
state_data['run_num'],
state_data['run_time'],
run_node,
action_graph=action_graph,
manual=state_data.get('manual', False),
output_path=output_path,
base_context=context,
)
action_runs = ActionRunFactory.action_run_collection_from_state(
job_run,
state_data['runs'],
state_data['cleanup_run'],
)
job_run.action_runs = action_runs
return job_run
def from_state(cls, state_data, action_graph, output_path, context,
run_node):
"""Restore a JobRun from a serialized state."""
node_pools = node.NodePoolStore.get_instance()
if state_data.get('node_name'):
run_node = node_pools.get(state_data['node_name'])
# TODO: remove in 0.6
if 'job_name' not in state_data:
# This is only to support old state files.
job_name = state_data['id'].split('.')[0]
else:
job_name = state_data['job_name']
job_run = cls(
job_name,
state_data['run_num'],
state_data['run_time'],
run_node,
action_graph=action_graph,
manual=state_data.get('manual', False),
output_path=output_path,
base_context=context
)
action_runs = ActionRunFactory.action_run_collection_from_state(
job_run, state_data['runs'], state_data['cleanup_run'])
job_run.action_runs = action_runs
return job_run
def test_action_run_collection_from_state(self):
state_data = [self.action_state_data]
cleanup_action_state_data = {
'job_run_id': 'job_run_id',
'action_name': 'cleanup',
'state': 'succeeded',
'run_time': self.run_time,
'start_time': None,
'end_time': None,
'command': 'do cleanup',
'node_name': 'anode',
'action_runner': {
'status_path': '/tmp/foo',
'exec_path': '/bin/foo'
}
}
collection = ActionRunFactory.action_run_collection_from_state(
self.job_run,
state_data,
cleanup_action_state_data,
)
assert_equal(collection.action_graph, self.action_graph)
assert_length(collection.run_map, 2)
assert_equal(collection.run_map['act1'].action_name, 'act1')
assert_equal(collection.run_map['cleanup'].action_name, 'cleanup')
def from_state(
cls,
state_data,
action_graph,
output_path,
context,
run_node,
):
"""Restore a JobRun from a serialized state."""
pool_repo = node.NodePoolRepository.get_instance()
run_node = pool_repo.get_node(state_data.get('node_name'), run_node)
job_name = state_data['job_name']
job_run = cls(
job_name,
state_data['run_num'],
state_data['run_time'],
run_node,
action_graph=action_graph,
manual=state_data.get('manual', False),
output_path=output_path,
base_context=context,
)
action_runs = ActionRunFactory.action_run_collection_from_state(
job_run,
state_data['runs'],
state_data['cleanup_run'],
)
job_run.action_runs = action_runs
return job_run
def test_action_run_collection_from_state(self):
state_data = [self.action_state_data]
cleanup_action_state_data = {
'job_run_id': 'job_run_id',
'action_name': 'cleanup',
'state': 'succeeded',
'run_time': self.run_time,
'start_time': None,
'end_time': None,
'command': 'do cleanup',
'node_name': 'anode'
}
collection = ActionRunFactory.action_run_collection_from_state(
self.job_run, state_data, cleanup_action_state_data)
assert_equal(collection.action_graph, self.action_graph)
assert_length(collection.run_map, 2)
assert_equal(collection.run_map['act1'].action_name, 'act1')
assert_equal(collection.run_map['cleanup'].action_name, 'cleanup')
