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 23 """Unit test for xml.py.""" 24 25 import unittest 26 import xml 27 28 29 class MockErrorHandler(object): 30 def __init__(self, handle_style_error): 31 self.turned_off_filtering = False 32 self._handle_style_error = handle_style_error 33 34 def turn_off_line_filtering(self): 35 self.turned_off_filtering = True 36 37 def __call__(self, line_number, category, confidence, message): 38 self._handle_style_error(self, line_number, category, confidence, message) 39 return True 40 41 42 class XMLCheckerTest(unittest.TestCase): 43 """Tests XMLChecker class.""" 44 45 def assert_no_error(self, xml_data): 46 def handle_style_error(mock_error_handler, line_number, category, confidence, message): 47 self.fail('Unexpected error: %d %s %d %s' % (line_number, category, confidence, message)) 48 49 error_handler = MockErrorHandler(handle_style_error) 50 checker = xml.XMLChecker('foo.xml', error_handler) 51 checker.check(xml_data.split('\n')) 52 self.assertTrue(error_handler.turned_off_filtering) 53 54 def assert_error(self, expected_line_number, expected_category, xml_data): 55 def handle_style_error(mock_error_handler, line_number, category, confidence, message): 56 mock_error_handler.had_error = True 57 self.assertEqual(expected_line_number, line_number) 58 self.assertEqual(expected_category, category) 59 60 error_handler = MockErrorHandler(handle_style_error) 61 error_handler.had_error = False 62 63 checker = xml.XMLChecker('foo.xml', error_handler) 64 checker.check(xml_data.split('\n')) 65 self.assertTrue(error_handler.had_error) 66 self.assertTrue(error_handler.turned_off_filtering) 67 68 def mock_handle_style_error(self): 69 pass 70 71 def test_conflict_marker(self): 72 self.assert_error(1, 'xml/syntax', '<<<<<<< HEAD\n<foo>\n</foo>\n') 73 74 def test_extra_closing_tag(self): 75 self.assert_error(3, 'xml/syntax', '<foo>\n</foo>\n</foo>\n') 76 77 def test_init(self): 78 error_handler = MockErrorHandler(self.mock_handle_style_error) 79 checker = xml.XMLChecker('foo.xml', error_handler) 80 self.assertEqual(checker._handle_style_error, error_handler) 81 82 def test_missing_closing_tag(self): 83 self.assert_error(3, 'xml/syntax', '<foo>\n<bar>\n</foo>\n') 84 85 def test_no_error(self): 86 self.assert_no_error('<foo>\n</foo>') 87