A Python parser for MediaWiki wikicode https://mwparserfromhell.readthedocs.io/
Nelze vybrat více než 25 témat Téma musí začínat písmenem nebo číslem, může obsahovat pomlčky („-“) a může být dlouhé až 35 znaků.
 
 
 
 

84 řádky
3.5 KiB

  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.compat import str
  25. from mwparserfromhell.nodes import Template
  26. from mwparserfromhell.nodes.extras import Attribute
  27. from ._test_tree_equality import TreeEqualityTestCase, wrap, wraptext
  28. class TestAttribute(TreeEqualityTestCase):
  29. """Test cases for the Attribute node extra."""
  30. def test_unicode(self):
  31. """test Attribute.__unicode__()"""
  32. node = Attribute(wraptext("foo"))
  33. self.assertEqual(" foo", str(node))
  34. node2 = Attribute(wraptext("foo"), wraptext("bar"))
  35. self.assertEqual(' foo="bar"', str(node2))
  36. node3 = Attribute(wraptext("a"), wraptext("b"), False, "", " ", " ")
  37. self.assertEqual("a = b", str(node3))
  38. def test_name(self):
  39. """test getter/setter for the name attribute"""
  40. name = wraptext("id")
  41. node = Attribute(name, wraptext("bar"))
  42. self.assertIs(name, node.name)
  43. node.name = "{{id}}"
  44. self.assertWikicodeEqual(wrap([Template(wraptext("id"))]), node.name)
  45. def test_value(self):
  46. """test getter/setter for the value attribute"""
  47. value = wraptext("foo")
  48. node = Attribute(wraptext("id"), value)
  49. self.assertIs(value, node.value)
  50. node.value = "{{bar}}"
  51. self.assertWikicodeEqual(wrap([Template(wraptext("bar"))]), node.value)
  52. def test_quoted(self):
  53. """test getter/setter for the quoted attribute"""
  54. node1 = Attribute(wraptext("id"), wraptext("foo"), False)
  55. node2 = Attribute(wraptext("id"), wraptext("bar"))
  56. self.assertFalse(node1.quoted)
  57. self.assertTrue(node2.quoted)
  58. node1.quoted = True
  59. node2.quoted = ""
  60. self.assertTrue(node1.quoted)
  61. self.assertFalse(node2.quoted)
  62. def test_padding(self):
  63. """test getter/setter for the padding attributes"""
  64. for pad in ["pad_first", "pad_before_eq", "pad_after_eq"]:
  65. node = Attribute(wraptext("id"), wraptext("foo"), **{pad: "\n"})
  66. self.assertEqual("\n", getattr(node, pad))
  67. setattr(node, pad, " ")
  68. self.assertEqual(" ", getattr(node, pad))
  69. setattr(node, pad, None)
  70. self.assertEqual("", getattr(node, pad))
  71. self.assertRaises(ValueError, setattr, node, pad, True)
  72. if __name__ == "__main__":
  73. unittest.main(verbosity=2)