xml_unittest.py 3.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788
  1. # Copyright (C) 2010 Apple Inc. All rights reserved.
  2. #
  3. # Redistribution and use in source and binary forms, with or without
  4. # modification, are permitted provided that the following conditions
  5. # are met:
  6. # 1. Redistributions of source code must retain the above copyright
  7. # notice, this list of conditions and the following disclaimer.
  8. # 2. Redistributions in binary form must reproduce the above copyright
  9. # notice, this list of conditions and the following disclaimer in the
  10. # documentation and/or other materials provided with the distribution.
  11. #
  12. # THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS'' AND
  13. # ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  14. # WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  15. # DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS BE LIABLE FOR
  16. # ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  17. # DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
  18. # SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
  19. # CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
  20. # OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  21. # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  22. """Unit test for xml.py."""
  23. import unittest2 as unittest
  24. import xml
  25. class MockErrorHandler(object):
  26. def __init__(self, handle_style_error):
  27. self.turned_off_filtering = False
  28. self._handle_style_error = handle_style_error
  29. def turn_off_line_filtering(self):
  30. self.turned_off_filtering = True
  31. def __call__(self, line_number, category, confidence, message):
  32. self._handle_style_error(self, line_number, category, confidence, message)
  33. return True
  34. class XMLCheckerTest(unittest.TestCase):
  35. """Tests XMLChecker class."""
  36. def assert_no_error(self, xml_data):
  37. def handle_style_error(mock_error_handler, line_number, category, confidence, message):
  38. self.fail('Unexpected error: %d %s %d %s' % (line_number, category, confidence, message))
  39. error_handler = MockErrorHandler(handle_style_error)
  40. checker = xml.XMLChecker('foo.xml', error_handler)
  41. checker.check(xml_data.split('\n'))
  42. self.assertTrue(error_handler.turned_off_filtering)
  43. def assert_error(self, expected_line_number, expected_category, xml_data):
  44. def handle_style_error(mock_error_handler, line_number, category, confidence, message):
  45. mock_error_handler.had_error = True
  46. self.assertEqual(expected_line_number, line_number)
  47. self.assertEqual(expected_category, category)
  48. error_handler = MockErrorHandler(handle_style_error)
  49. error_handler.had_error = False
  50. checker = xml.XMLChecker('foo.xml', error_handler)
  51. checker.check(xml_data.split('\n'))
  52. self.assertTrue(error_handler.had_error)
  53. self.assertTrue(error_handler.turned_off_filtering)
  54. def mock_handle_style_error(self):
  55. pass
  56. def test_conflict_marker(self):
  57. self.assert_error(1, 'xml/syntax', '<<<<<<< HEAD\n<foo>\n</foo>\n')
  58. def test_extra_closing_tag(self):
  59. self.assert_error(3, 'xml/syntax', '<foo>\n</foo>\n</foo>\n')
  60. def test_init(self):
  61. error_handler = MockErrorHandler(self.mock_handle_style_error)
  62. checker = xml.XMLChecker('foo.xml', error_handler)
  63. self.assertEqual(checker._handle_style_error, error_handler)
  64. def test_missing_closing_tag(self):
  65. self.assert_error(3, 'xml/syntax', '<foo>\n<bar>\n</foo>\n')
  66. def test_no_error(self):
  67. self.assert_no_error('<foo>\n</foo>')