A Python parser for MediaWiki wikicode https://mwparserfromhell.readthedocs.io/
Você não pode selecionar mais de 25 tópicos Os tópicos devem começar com uma letra ou um número, podem incluir traços ('-') e podem ter até 35 caracteres.

test_utils.py 3.0 KiB

11 anos atrás
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667
  1. # -*- coding: utf-8 -*-
  2. #
  3. # Copyright (C) 2012-2013 Ben Kurtovic <ben.kurtovic@verizon.net>
  4. #
  5. # Permission is hereby granted, free of charge, to any person obtaining a copy
  6. # of this software and associated documentation files (the "Software"), to deal
  7. # in the Software without restriction, including without limitation the rights
  8. # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  9. # copies of the Software, and to permit persons to whom the Software is
  10. # furnished to do so, subject to the following conditions:
  11. #
  12. # The above copyright notice and this permission notice shall be included in
  13. # all copies or substantial portions of the Software.
  14. #
  15. # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16. # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17. # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  18. # AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19. # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  20. # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  21. # SOFTWARE.
  22. from __future__ import unicode_literals
  23. import unittest
  24. from mwparserfromhell.nodes import Template, Text
  25. from mwparserfromhell.smart_list import SmartList
  26. from mwparserfromhell.utils import parse_anything
  27. from mwparserfromhell.wikicode import Wikicode
  28. from ._test_tree_equality import TreeEqualityTestCase
  29. class TestUtils(TreeEqualityTestCase):
  30. """Tests for the utils module, which provides parse_anything()."""
  31. def test_parse_anything_valid(self):
  32. """tests for valid input to utils.parse_anything()"""
  33. wrap = lambda L: Wikicode(SmartList(L))
  34. textify = lambda L: wrap([Text(item) for item in L])
  35. tests = [
  36. (wrap([Text("foobar")]), textify(["foobar"])),
  37. (Template(wrap([Text("spam")])),
  38. wrap([Template(textify(["spam"]))])),
  39. ("fóóbar", textify(["fóóbar"])),
  40. (b"foob\xc3\xa1r", textify(["foobár"])),
  41. (123, textify(["123"])),
  42. (True, textify(["True"])),
  43. (None, wrap([])),
  44. ([Text("foo"), Text("bar"), Text("baz")],
  45. textify(["foo", "bar", "baz"])),
  46. ([wrap([Text("foo")]), Text("bar"), "baz", 123, 456],
  47. textify(["foo", "bar", "baz", "123", "456"])),
  48. ([[[([[((("foo",),),)], "bar"],)]]], textify(["foo", "bar"]))
  49. ]
  50. for test, valid in tests:
  51. self.assertWikicodeEqual(valid, parse_anything(test))
  52. def test_parse_anything_invalid(self):
  53. """tests for invalid input to utils.parse_anything()"""
  54. self.assertRaises(ValueError, parse_anything, Ellipsis)
  55. self.assertRaises(ValueError, parse_anything, object)
  56. self.assertRaises(ValueError, parse_anything, object())
  57. self.assertRaises(ValueError, parse_anything, type)
  58. self.assertRaises(ValueError, parse_anything, ["foo", [object]])
  59. if __name__ == "__main__":
  60. unittest.main(verbosity=2)