def test_register_node__is_not_end_event(self):
node_cls = MagicMock()
FlowNodeClsFactory.register_node('key', node_cls)
self.assertEqual(FlowNodeClsFactory.get_node_cls('key'), node_cls)
flow.post_new_end_event_register.send.assert_not_called()
FlowNodeClsFactory.nodes_cls.pop('key')
def test_register_node__with_end_event(self):
class TestEnd(EndEvent):
pass
FlowNodeClsFactory.register_node('key', TestEnd)
self.assertEqual(FlowNodeClsFactory.get_node_cls('key'), TestEnd)
flow.post_new_end_event_register.send.assert_called_once_with(
sender=EndEvent, node_type='key', node_cls=TestEnd)
FlowNodeClsFactory.nodes_cls.pop('key')