xml_unittest.py revision f05b935882198ccf7d81675736e3aeb089c5113a 
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
27import unittest
28
29import xml
30
31
32class 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
71if __name__ == '__main__':
72    unittest.main()
73