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])
