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)
