def testHook(self): reg_map = RegisterMap({StrongMacro : Register()}) reg_map.add_hooks([StrongMacroHook]) s = '((silne "long argument"))' o = parse(s, reg_map, document_root = True) self.assertEquals(DummyNode, o.children[0].__class__) self.assertEquals(StrongNode, o.children[1].__class__) self.assertEquals("argument replaced by hook", o.children[1].children[0].content)
def testHookAddedToExistings(self): reg_map = RegisterMap({StrongMacro : Register()}) reg_map.add_hooks([StrongMacroHook]) self.assertEquals(set([StrongMacroHook]), reg_map.hooks[StrongMacro])