| from pychecker2 import main |
| from pychecker2 import Options |
| |
| import unittest |
| |
| class WarningTester(unittest.TestCase): |
| def __init__(self, arg): |
| unittest.TestCase.__init__(self, arg) |
| self.options = Options.Options() |
| self.checklist = main.create_checklist(self.options) |
| self.argv = [] |
| |
| def check_warning(self, w, expected_line, expected_type, *expected_args): |
| warn_line, warn_type, warn_args = w |
| try: |
| self.assertEqual(warn_type, expected_type) |
| self.assertEqual(warn_line, expected_line) |
| self.assertEqual(len(warn_args), len(expected_args)) |
| for i in range(len(expected_args)): |
| self.assertEqual(warn_args[i], expected_args[i]) |
| except AssertionError: # help w/debugging |
| print warn_line, warn_type, warn_args |
| print expected_line, expected_type, expected_args |
| raise |
| |
| def check_file(self, data): |
| import tempfile, os |
| fname = tempfile.mktemp() |
| fp = open(fname, 'wb') |
| try: |
| fp.write(data) |
| fp.close() |
| f, = self.options.process_options(self.argv + [fname]) |
| self.checklist.check_file(f) |
| finally: |
| fp.close() |
| os.unlink(fname) |
| return f |
| |
| def warning_file(self, f, line, warning, *args): |
| assert len(f.warnings) == 1, "Not just one warning:" + `f.warnings` |
| self.check_warning(f.warnings[0], line, warning, *args) |
| |
| def warning(self, test, line, warning, *args): |
| test += '\n' |
| f = self.check_file(test) |
| self.warning_file(f, line, warning, *args) |
| |
| def silent(self, test): |
| f = self.check_file(test) |
| if f.warnings: |
| print f.warnings |
| self.assertEqual(len(f.warnings), 0) |
| |
| def setUp(self): |
| self.argv = [] |
| |
