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