summaryrefslogtreecommitdiffstats
path: root/Lib/test
diff options
context:
space:
mode:
authorR. David Murray <rdmurray@bitdance.com>2010-12-16 19:08:51 (GMT)
committerR. David Murray <rdmurray@bitdance.com>2010-12-16 19:08:51 (GMT)
commit5317e9cd8dc4bf1f02eb9144acb119673d537387 (patch)
tree7b89da90a5d01a62d749b7f976438d14bd762fa6 /Lib/test
parent20f0fb68aa24876abf768292b33cda138237ff6c (diff)
downloadcpython-5317e9cd8dc4bf1f02eb9144acb119673d537387.zip
cpython-5317e9cd8dc4bf1f02eb9144acb119673d537387.tar.gz
cpython-5317e9cd8dc4bf1f02eb9144acb119673d537387.tar.bz2
#10719: restore messages generated on invalid compileall args
Before the introduction of filename arguments to compileall it gave semi useful messages about not being able to 'list' names that weren't valid directories. This fix restores that behavior. In addition to the test for this case, the patch also adds a test for the default behavior of compileall when no arguments are provided, and fixes a bug in one of the previously added tests.
Diffstat (limited to 'Lib/test')
-rw-r--r--Lib/test/test_compileall.py25
1 files changed, 17 insertions, 8 deletions
diff --git a/Lib/test/test_compileall.py b/Lib/test/test_compileall.py
index cd08a8e..3f333a5 100644
--- a/Lib/test/test_compileall.py
+++ b/Lib/test/test_compileall.py
@@ -124,14 +124,15 @@ class EncodingTest(unittest.TestCase):
class CommandLineTests(unittest.TestCase):
"""Test compileall's CLI."""
- def assertRunOK(self, *args):
- rc, out, err = script_helper.assert_python_ok('-m', 'compileall', *args)
+ def assertRunOK(self, *args, **env_vars):
+ rc, out, err = script_helper.assert_python_ok(
+ '-m', 'compileall', *args, **env_vars)
self.assertEqual(b'', err)
return out
- def assertRunNotOK(self, *args):
+ def assertRunNotOK(self, *args, **env_vars):
rc, out, err = script_helper.assert_python_failure(
- '-m', 'compileall', *args)
+ '-m', 'compileall', *args, **env_vars)
return rc, out, err
def assertCompiled(self, fn):
@@ -149,12 +150,17 @@ class CommandLineTests(unittest.TestCase):
# Create the __init__.py and a package module.
self.initfn = script_helper.make_script(self.pkgdir, '__init__', '')
self.barfn = script_helper.make_script(self.pkgdir, 'bar', '')
- sys.path.insert(0, self.directory)
def _cleanup(self):
support.rmtree(self.directory)
- assert sys.path[0] == self.directory, 'Missing path'
- del sys.path[0]
+
+ def test_no_args_compiles_path(self):
+ # Note that -l is implied for the no args case.
+ bazfn = script_helper.make_script(self.directory, 'baz', '')
+ self.assertRunOK(PYTHONPATH=self.directory)
+ self.assertCompiled(bazfn)
+ self.assertNotCompiled(self.initfn)
+ self.assertNotCompiled(self.barfn)
# Ensure that the default behavior of compileall's CLI is to create
# PEP 3147 pyc/pyo files.
@@ -322,11 +328,14 @@ class CommandLineTests(unittest.TestCase):
bingfn = script_helper.make_script(self.pkgdir, 'bing', 'syntax(error')
rc, out, err = self.assertRunNotOK('nosuchfile', self.initfn,
bingfn, self.barfn)
- self.assertRegex(b'rror', err)
+ self.assertRegex(out, b'rror')
self.assertNotCompiled(bingfn)
self.assertCompiled(self.initfn)
self.assertCompiled(self.barfn)
+ def test_invalid_arg_produces_message(self):
+ out = self.assertRunOK('badfilename')
+ self.assertRegex(out, b"Can't list badfilename")
def test_main():