A Python parser for MediaWiki wikicode https://mwparserfromhell.readthedocs.io/
您最多选择25个主题 主题必须以字母或数字开头,可以包含连字符 (-),并且长度不得超过35个字符

test_attribute.py 3.8 KiB

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