150 lines
6.1 KiB
Python
150 lines
6.1 KiB
Python
|
# -*- 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()
|