class FutureOperationTest(unittest.TestCase):
def setUp(self):
self.operation_class = mock.Mock()
self.name = 'my op name'
self.operation_id = 12345
self.input_connections = mock.Mock()
self.output_properties = mock.Mock()
self.log_dir = mock.Mock()
self.parent = mock.Mock()
self.future_operation = FutureOperation(
name=self.name,
operation_class=self.operation_class,
operation_id=self.operation_id,
input_connections=self.input_connections,
output_properties=self.output_properties,
parent=self.parent,
log_dir=self.log_dir)
def test_child_data_no_children(self):
net_key = 'super net key'
self.assertEqual({}, self.future_operation._child_data(net_key))
def test_child_data_one_child(self):
child = FutureOperation(
name='my child name',
operation_id=12346,
parent=self.future_operation,
operation_class=self.operation_class,
input_connections=self.input_connections,
output_properties=self.output_properties,
log_dir=self.log_dir)
net_key = 'super net key'
self.assertEqual({'my child name': (net_key, 12346)},
self.future_operation._child_data(net_key))
def test_init_adds_parent(self):
self.parent.add_child.assert_called_once_with(self.future_operation)
def setUp(self):
self.operation_class = mock.Mock()
self.name = 'my op name'
self.operation_id = 12345
self.input_connections = mock.Mock()
self.output_properties = mock.Mock()
self.log_dir = mock.Mock()
self.parent = mock.Mock()
self.future_operation = FutureOperation(
name=self.name,
operation_class=self.operation_class,
operation_id=self.operation_id,
input_connections=self.input_connections,
output_properties=self.output_properties,
parent=self.parent,
log_dir=self.log_dir)
