| # -*- coding: utf-8 -*- |
| import sys |
| import codecs |
| import logging |
| import os |
| import re |
| import shutil |
| import StringIO |
| import sys |
| import tempfile |
| import unittest |
| |
| from lib2to3 import main |
| |
| |
| TEST_DATA_DIR = os.path.join(os.path.dirname(__file__), "data") |
| PY2_TEST_MODULE = os.path.join(TEST_DATA_DIR, "py2_test_grammar.py") |
| |
| |
| class TestMain(unittest.TestCase): |
| |
| if not hasattr(unittest.TestCase, 'assertNotRegex'): |
| # This method was only introduced in 3.2. |
| def assertNotRegex(self, text, regexp, msg=None): |
| import re |
| if not hasattr(regexp, 'search'): |
| regexp = re.compile(regexp) |
| if regexp.search(text): |
| self.fail("regexp %s MATCHED text %r" % (regexp.pattern, text)) |
| |
| def setUp(self): |
| self.temp_dir = None # tearDown() will rmtree this directory if set. |
| |
| def tearDown(self): |
| # Clean up logging configuration down by main. |
| del logging.root.handlers[:] |
| if self.temp_dir: |
| shutil.rmtree(self.temp_dir) |
| |
| def run_2to3_capture(self, args, in_capture, out_capture, err_capture): |
| save_stdin = sys.stdin |
| save_stdout = sys.stdout |
| save_stderr = sys.stderr |
| sys.stdin = in_capture |
| sys.stdout = out_capture |
| sys.stderr = err_capture |
| try: |
| return main.main("lib2to3.fixes", args) |
| finally: |
| sys.stdin = save_stdin |
| sys.stdout = save_stdout |
| sys.stderr = save_stderr |
| |
| def test_unencodable_diff(self): |
| input_stream = StringIO.StringIO(u"print 'nothing'\nprint u'über'\n") |
| out = StringIO.StringIO() |
| out_enc = codecs.getwriter("ascii")(out) |
| err = StringIO.StringIO() |
| ret = self.run_2to3_capture(["-"], input_stream, out_enc, err) |
| self.assertEqual(ret, 0) |
| output = out.getvalue() |
| self.assertTrue("-print 'nothing'" in output) |
| self.assertTrue("WARNING: couldn't encode <stdin>'s diff for " |
| "your terminal" in err.getvalue()) |
| |
| def setup_test_source_trees(self): |
| """Setup a test source tree and output destination tree.""" |
| self.temp_dir = tempfile.mkdtemp() # tearDown() cleans this up. |
| self.py2_src_dir = os.path.join(self.temp_dir, "python2_project") |
| self.py3_dest_dir = os.path.join(self.temp_dir, "python3_project") |
| os.mkdir(self.py2_src_dir) |
| os.mkdir(self.py3_dest_dir) |
| # Turn it into a package with a few files. |
| self.setup_files = [] |
| open(os.path.join(self.py2_src_dir, "__init__.py"), "w").close() |
| self.setup_files.append("__init__.py") |
| shutil.copy(PY2_TEST_MODULE, self.py2_src_dir) |
| self.setup_files.append(os.path.basename(PY2_TEST_MODULE)) |
| self.trivial_py2_file = os.path.join(self.py2_src_dir, "trivial.py") |
| self.init_py2_file = os.path.join(self.py2_src_dir, "__init__.py") |
| with open(self.trivial_py2_file, "w") as trivial: |
| trivial.write("print 'I need a simple conversion.'") |
| self.setup_files.append("trivial.py") |
| |
| def test_filename_changing_on_output_single_dir(self): |
| """2to3 a single directory with a new output dir and suffix.""" |
| self.setup_test_source_trees() |
| out = StringIO.StringIO() |
| err = StringIO.StringIO() |
| suffix = "TEST" |
| ret = self.run_2to3_capture( |
| ["-n", "--add-suffix", suffix, "--write-unchanged-files", |
| "--no-diffs", "--output-dir", |
| self.py3_dest_dir, self.py2_src_dir], |
| StringIO.StringIO(""), out, err) |
| self.assertEqual(ret, 0) |
| stderr = err.getvalue() |
| self.assertIn(" implies -w.", stderr) |
| self.assertIn( |
| "Output in %r will mirror the input directory %r layout" % ( |
| self.py3_dest_dir, self.py2_src_dir), stderr) |
| self.assertEqual(set(name+suffix for name in self.setup_files), |
| set(os.listdir(self.py3_dest_dir))) |
| for name in self.setup_files: |
| self.assertIn("Writing converted %s to %s" % ( |
| os.path.join(self.py2_src_dir, name), |
| os.path.join(self.py3_dest_dir, name+suffix)), stderr) |
| sep = re.escape(os.sep) |
| self.assertRegexpMatches( |
| stderr, r"No changes to .*/__init__\.py".replace("/", sep)) |
| self.assertNotRegex( |
| stderr, r"No changes to .*/trivial\.py".replace("/", sep)) |
| |
| def test_filename_changing_on_output_two_files(self): |
| """2to3 two files in one directory with a new output dir.""" |
| self.setup_test_source_trees() |
| err = StringIO.StringIO() |
| py2_files = [self.trivial_py2_file, self.init_py2_file] |
| expected_files = set(os.path.basename(name) for name in py2_files) |
| ret = self.run_2to3_capture( |
| ["-n", "-w", "--write-unchanged-files", |
| "--no-diffs", "--output-dir", self.py3_dest_dir] + py2_files, |
| StringIO.StringIO(""), StringIO.StringIO(), err) |
| self.assertEqual(ret, 0) |
| stderr = err.getvalue() |
| self.assertIn( |
| "Output in %r will mirror the input directory %r layout" % ( |
| self.py3_dest_dir, self.py2_src_dir), stderr) |
| self.assertEqual(expected_files, set(os.listdir(self.py3_dest_dir))) |
| |
| def test_filename_changing_on_output_single_file(self): |
| """2to3 a single file with a new output dir.""" |
| self.setup_test_source_trees() |
| err = StringIO.StringIO() |
| ret = self.run_2to3_capture( |
| ["-n", "-w", "--no-diffs", "--output-dir", self.py3_dest_dir, |
| self.trivial_py2_file], |
| StringIO.StringIO(""), StringIO.StringIO(), err) |
| self.assertEqual(ret, 0) |
| stderr = err.getvalue() |
| self.assertIn( |
| "Output in %r will mirror the input directory %r layout" % ( |
| self.py3_dest_dir, self.py2_src_dir), stderr) |
| self.assertEqual(set([os.path.basename(self.trivial_py2_file)]), |
| set(os.listdir(self.py3_dest_dir))) |
| |
| |
| if __name__ == '__main__': |
| unittest.main() |
