class ParallelIdentifierReferTest(unittest.TestCase):
def setUp(self):
self.pi = ParallelIdentifier([(6, 2), (8, 3)])
def test_refers_to_true(self):
operation = mock.Mock()
operation.operation_id = 6
self.assertTrue(self.pi.refers_to(operation))
def test_refers_to_false(self):
operation = mock.Mock()
operation.operation_id = 2
self.assertFalse(self.pi.refers_to(operation))
def _execute(self, parsed_arguments):
net = rom.get_object(self.storage, parsed_arguments.net_key)
parallel_id = ParallelIdentifier.deserialize(parsed_arguments.parallel_id)
_execute_deferred = defer.Deferred()
inputs_file = NamedTemporaryFile()
outputs_file = NamedTemporaryFile()
write_inputs(inputs_file, net=net, parallel_id=parallel_id, operation_id=parsed_arguments.operation_id)
cmdline = copy.copy(self.perl_wrapper)
cmdline_builder = CMDLINE_BUILDERS[parsed_arguments.action_type]
cmdline.extend(cmdline_builder(parsed_arguments.method, parsed_arguments.action_id, inputs_file, outputs_file))
LOG.info("Executing (%s): %s", socket.gethostname(), " ".join(cmdline))
logannotator = LogAnnotator(cmdline)
deferred = logannotator.start()
deferred.addCallback(
self._finish,
parsed_arguments=parsed_arguments,
inputs_file=inputs_file,
outputs_file=outputs_file,
parallel_id=parallel_id,
net=net,
_execute_deferred=_execute_deferred,
)
deferred.addErrback(self._exit)
return _execute_deferred
def test_child_identifier_doesnt_modify_parent(self):
pi = ParallelIdentifier([])
child_identifier = pi.child_identifier(6, 2)
self.assertEqual([], list(pi))
def test_child_identifier(self):
pi = ParallelIdentifier([])
self.assertEqual([(6, 2)], list(pi.child_identifier(6, 2)))
def setUp(self):
self.pi = ParallelIdentifier([(6, 2), (8, 3)])
