A Python parser for MediaWiki wikicode https://mwparserfromhell.readthedocs.io/
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 

90 lines
3.8 KiB

  1. # -*- coding: utf-8 -*-
  2. #
  3. # Copyright (C) 2012-2014 Ben Kurtovic <ben.kurtovic@gmail.com>
  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"), True, "", " ", " ")
  37. self.assertEqual('a = "b"', str(node3))
  38. node3 = Attribute(wraptext("a"), wraptext("b"), False, "", " ", " ")
  39. self.assertEqual("a = b", str(node3))
  40. node4 = Attribute(wraptext("a"), wrap([]), False, " ", "", " ")
  41. self.assertEqual(" a= ", str(node4))
  42. def test_name(self):
  43. """test getter/setter for the name attribute"""
  44. name = wraptext("id")
  45. node = Attribute(name, wraptext("bar"))
  46. self.assertIs(name, node.name)
  47. node.name = "{{id}}"
  48. self.assertWikicodeEqual(wrap([Template(wraptext("id"))]), node.name)
  49. def test_value(self):
  50. """test getter/setter for the value attribute"""
  51. value = wraptext("foo")
  52. node = Attribute(wraptext("id"), value)
  53. self.assertIs(value, node.value)
  54. node.value = "{{bar}}"
  55. self.assertWikicodeEqual(wrap([Template(wraptext("bar"))]), node.value)
  56. node.value = None
  57. self.assertIs(None, node.value)
  58. def test_quoted(self):
  59. """test getter/setter for the quoted attribute"""
  60. node1 = Attribute(wraptext("id"), wraptext("foo"), False)
  61. node2 = Attribute(wraptext("id"), wraptext("bar"))
  62. self.assertFalse(node1.quoted)
  63. self.assertTrue(node2.quoted)
  64. node1.quoted = True
  65. node2.quoted = ""
  66. self.assertTrue(node1.quoted)
  67. self.assertFalse(node2.quoted)
  68. def test_padding(self):
  69. """test getter/setter for the padding attributes"""
  70. for pad in ["pad_first", "pad_before_eq", "pad_after_eq"]:
  71. node = Attribute(wraptext("id"), wraptext("foo"), **{pad: "\n"})
  72. self.assertEqual("\n", getattr(node, pad))
  73. setattr(node, pad, " ")
  74. self.assertEqual(" ", getattr(node, pad))
  75. setattr(node, pad, None)
  76. self.assertEqual("", getattr(node, pad))
  77. self.assertRaises(ValueError, setattr, node, pad, True)
  78. if __name__ == "__main__":
  79. unittest.main(verbosity=2)