# -*- coding: utf-8 -*- # # Copyright (C) 2012-2013 Ben Kurtovic # # Permission is hereby granted, free of charge, to any person obtaining a copy # of this software and associated documentation files (the "Software"), to deal # in the Software without restriction, including without limitation the rights # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell # copies of the Software, and to permit persons to whom the Software is # furnished to do so, subject to the following conditions: # # The above copyright notice and this permission notice shall be included in # all copies or substantial portions of the Software. # # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE # AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE # SOFTWARE. from __future__ import unicode_literals import unittest from mwparserfromhell.compat import range from mwparserfromhell.nodes import Template, Text, Wikilink from mwparserfromhell.nodes.extras import Parameter from mwparserfromhell.parser import Parser from mwparserfromhell.smart_list import SmartList from mwparserfromhell.wikicode import Wikicode class TestParser(unittest.TestCase): """Tests for the Parser class itself, which tokenizes and builds nodes.""" def assertNodesEqual(self, expected, actual): """Assert that two Nodes are the same type and have the same data.""" self.assertIs(type(expected), type(actual)) if isinstance(expected, Text): self.assertEqual(expected.value, actual.value) elif isinstance(expected, Template): self.assertWikicodeEqual(expected.name, actual.name) length = len(expected.params) self.assertEqual(length, len(actual.params)) for i in range(length): exp_param = expected.params[i] act_param = actual.params[i] self.assertWikicodeEqual(exp_param.name, act_param.name) self.assertWikicodeEqual(exp_param.value, act_param.value) self.assertIs(exp_param.showkey, act_param.showkey) elif isinstance(expected, Wikilink): self.assertWikicodeEqual(expected.title, actual.title) if expected.text is not None: self.assertWikicodeEqual(expected.text, actual.text) else: self.assertIs(None, actual.text) def assertWikicodeEqual(self, expected, actual): """Assert that two Wikicode objects have the same data.""" self.assertIsInstance(actual, Wikicode) length = len(expected.nodes) self.assertEqual(length, len(actual.nodes)) for i in range(length): self.assertNodesEqual(expected.get(i), actual.get(i)) def test_parser(self): """integration test for parsing overall""" text = "this is text; {{this|is=a|template={{with|[[links]]|in}}it}}" wrap = lambda L: Wikicode(SmartList(L)) expected = wrap([ Text("this is text; "), Template(wrap([Text("this")]), [ Parameter(wrap([Text("is")]), wrap([Text("a")])), Parameter(wrap([Text("template")]), wrap([ Template(wrap([Text("with")]), [ Parameter(wrap([Text("1")]), wrap([Wikilink(wrap([Text("links")]))]), showkey=False), Parameter(wrap([Text("2")]), wrap([Text("in")]), showkey=False) ]), Text("it") ])) ]) ]) actual = Parser(text).parse() self.assertWikicodeEqual(expected, actual) if __name__ == "__main__": unittest.main(verbosity=2)