#!/usr/bin/python # Copyright (C) 2009 Google Inc. All rights reserved. # # Redistribution and use in source and binary forms, with or without # modification, are permitted provided that the following conditions are # met: # # * Redistributions of source code must retain the above copyright # notice, this list of conditions and the following disclaimer. # * Redistributions in binary form must reproduce the above # copyright notice, this list of conditions and the following disclaimer # in the documentation and/or other materials provided with the # distribution. # * Neither the name of Google Inc. nor the names of its # contributors may be used to endorse or promote products derived from # this software without specific prior written permission. # # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. """Unit test for text_style.py.""" import unittest import text as text_style from text import TextChecker class TextStyleTestCase(unittest.TestCase): """TestCase for text_style.py""" def assertNoError(self, lines): """Asserts that the specified lines has no errors.""" self.had_error = False def error_for_test(line_number, category, confidence, message): """Records if an error occurs.""" self.had_error = True text_style.process_file_data('', lines, error_for_test) self.assert_(not self.had_error, '%s should not have any errors.' % lines) def assertError(self, lines, expected_line_number): """Asserts that the specified lines has an error.""" self.had_error = False def error_for_test(line_number, category, confidence, message): """Checks if the expected error occurs.""" self.assertEquals(expected_line_number, line_number) self.assertEquals('whitespace/tab', category) self.had_error = True text_style.process_file_data('', lines, error_for_test) self.assert_(self.had_error, '%s should have an error [whitespace/tab].' % lines) def test_no_error(self): """Tests for no error cases.""" self.assertNoError(['']) self.assertNoError(['abc def', 'ggg']) def test_error(self): """Tests for error cases.""" self.assertError(['2009-12-16\tKent Tamura\t'], 1) self.assertError(['2009-12-16 Kent Tamura ', '', '\tReviewed by NOBODY.'], 3) class TextCheckerTest(unittest.TestCase): """Tests TextChecker class.""" def mock_handle_style_error(self): pass def test_init(self): """Test __init__ constructor.""" checker = TextChecker("foo.txt", self.mock_handle_style_error) self.assertEquals(checker.file_path, "foo.txt") self.assertEquals(checker.handle_style_error, self.mock_handle_style_error) if __name__ == '__main__': unittest.main()