def testDumpWiki(self): attachment_dir = self.setUpFolder(mock=tests.MOCK_ALWAYS_REAL) LocalFolder(tests.ZIM_DATADIR).file('zim.png').copyto(attachment_dir.file('test.png')) notebook = self.setUpNotebook() notebook.get_attachments_dir = lambda *a: attachment_dir pageview = setUpPageView(notebook, text='{{./test.png?type=equation}}') pageview.textview.get_buffer().set_modified(True) tree = pageview.get_parsetree() text = WikiDumper().dump(tree) self.assertEquals(text, ['{{./test.png?type=equation}}\n'])
def testModelFromData(self): notebook = self.setUpNotebook() page = notebook.get_page(Path('Test')) model = self.otype.model_from_data(notebook, page, {}, TABLE_WIKI_TEXT) builder = OldParseTreeBuilder() # XXX builder.start('zim-tree') self.otype.dump(builder, model) builder.end('zim-tree') tree = ParseTree(builder.close()) #self.assertEquals(list(tree.iter_tokens()), TABLE_TOKENS) -- XXX should work but doesn;t :( self.assertEquals(''.join(WikiDumper().dump(tree)), TABLE_WIKI_TEXT[1:-1])
def testModelFromElement(self): tree = WikiParser().parse(TABLE_WIKI_TEXT) element = tree._etree.getroot().find('table') self.assertIsNotNone(element) model = self.otype.model_from_element(element.attrib, element) builder = OldParseTreeBuilder() # XXX builder.start('zim-tree') self.otype.dump(builder, model) builder.end('zim-tree') tree = ParseTree(builder.close()) #self.assertEquals(list(tree.iter_tokens()), TABLE_TOKENS) -- XXX should work but doesn;t :( self.assertEquals(''.join(WikiDumper().dump(tree)), TABLE_WIKI_TEXT[1:-1])