def setUp(self): self.context = TopologyContext( config={}, topology=mock_protobuf.get_mock_topology(), task_to_component={}, my_task_id="task_id", metrics_collector=None, topo_pex_path="path.to.pex")
def set_topology_context(self, metrics_collector): """Sets a new topology context""" Log.debug("Setting topology context") cluster_config = self.get_topology_config() cluster_config.update(self._get_dict_from_config(self.my_component.config)) task_to_component_map = self._get_task_to_comp_map() self.context = TopologyContext(cluster_config, self.pplan.topology, task_to_component_map, self.my_task_id, metrics_collector, self.topology_pex_abs_path)
class TopologyContextTest(unittest.TestCase): def setUp(self): self.context = TopologyContext( config={}, topology=mock_protobuf.get_mock_topology(), task_to_component={}, my_task_id="task_id", metrics_collector=None, topo_pex_path="path.to.pex") def test_task_hook(self): task_hook = mock_generator.MockTaskHook() self.assertFalse(self.context.hook_exists) self.context.add_task_hook(task_hook) self.assertTrue(self.context.hook_exists) self.context.invoke_hook_prepare() self.context.invoke_hook_emit(None, None, None) self.assertTrue(task_hook.emit_called) self.context.invoke_hook_spout_ack(None, 0.1) self.assertTrue(task_hook.spout_ack_called) self.context.invoke_hook_spout_fail(None, 0.1) self.assertTrue(task_hook.spout_fail_called) self.context.invoke_hook_bolt_execute(None, 0.1) self.assertTrue(task_hook.bolt_exec_called) self.context.invoke_hook_bolt_ack(None, 0.1) self.assertTrue(task_hook.bolt_ack_called) self.context.invoke_hook_bolt_fail(None, 0.1) self.assertTrue(task_hook.bolt_fail_called)
def setUp(self): self.context = TopologyContext(config={}, topology=mock_protobuf.get_mock_topology(), task_to_component={}, my_task_id="task_id", metrics_collector=None, topo_pex_path="path.to.pex")
class TopologyContextTest(unittest.TestCase): def setUp(self): self.context = TopologyContext(config={}, topology=mock_protobuf.get_mock_topology(), task_to_component={}, my_task_id="task_id", metrics_collector=None, topo_pex_path="path.to.pex") def test_task_hook(self): task_hook = mock_generator.MockTaskHook() self.assertFalse(self.context.hook_exists) self.context.add_task_hook(task_hook) self.assertTrue(self.context.hook_exists) self.context.invoke_hook_prepare() self.context.invoke_hook_emit(None, None, None) self.assertTrue(task_hook.emit_called) self.context.invoke_hook_spout_ack(None, 0.1) self.assertTrue(task_hook.spout_ack_called) self.context.invoke_hook_spout_fail(None, 0.1) self.assertTrue(task_hook.spout_fail_called) self.context.invoke_hook_bolt_execute(None, 0.1) self.assertTrue(task_hook.bolt_exec_called) self.context.invoke_hook_bolt_ack(None, 0.1) self.assertTrue(task_hook.bolt_ack_called) self.context.invoke_hook_bolt_fail(None, 0.1) self.assertTrue(task_hook.bolt_fail_called)