def setUp(self):
"""
:return:
"""
self.job_a = Job('a', Templates.base)
self.job_b = Job('b', Templates.base)
self.job_c = Job('c', Templates.base)
self.job_d = Job('d', Templates.base)
self.node_a = Node(self.job_a)
self.node_b = Node(self.job_b)
self.node_c = Node(self.job_c)
self.node_d = Node(self.job_d)
self.node_a.add_child(self.node_b)
self.node_a.add_child(self.node_c)
self.node_d.add_parent(self.node_b)
self.node_d.add_parent(self.node_c)
def test_list_scripts(self):
node = Node(self.job_a,
post_script='script',
post_script_args=' '.join(('arg1', 'arg2', 'arg3')))
expected = 'SCRIPT POST a script arg1 arg2 arg3\n'
actual = node.list_scripts()
msg = 'testing that post script string is formatted correctly'
self.assertEqual(
expected, actual,
'%s\nExpected: %s\nActual: %s\n' % (msg, expected, actual))
def condorpy_node(self):
if not hasattr(self, '_condorpy_node'):
condorpy_node = Node(job=self.job,
pre_script=self.pre_script,
pre_script_args=self.pre_script_args,
post_script=self.post_script,
post_script_args=self.post_script_args,
variables=self.variables,
priority=self.priority,
category=self.category,
retry=self.retry,
pre_skip=self.pre_skip,
abort_dag_on=self.abort_dag_on,
abort_dag_on_return_value=self.abort_dag_on_return_value,
dir=self.dir,
noop=self.noop,
done=self.done
)
self._condorpy_node = condorpy_node
return self._condorpy_node