diff options
Diffstat (limited to 'tests/test_process.py')
| -rw-r--r-- | tests/test_process.py | 33 |
1 files changed, 33 insertions, 0 deletions
diff --git a/tests/test_process.py b/tests/test_process.py index dee2984f..9f7a16f5 100644 --- a/tests/test_process.py +++ b/tests/test_process.py @@ -953,6 +953,39 @@ class EnvironmentTest(CoverageTest): out_py = self.run_command("python -m package") self.assertMultiLineEqual(out_cov, out_py) + def test_coverage_custom_script(self): + # https://github.com/nedbat/coveragepy/issues/678 + # If sys.path[0] isn't the Python default, then coverage.py won't + # fiddle with it. + self.make_file("a/b/c/thing.py", """\ + SOMETHING = "hello-xyzzy" + """) + abc = os.path.abspath("a/b/c") + self.make_file("run_coverage.py", """\ + import sys + sys.path[0:0] = [ + r'{abc}', + '/Users/somebody/temp/something/eggs/something-4.5.1-py2.7-xxx-10.13-x86_64.egg', + ] + + import coverage.cmdline + + if __name__ == '__main__': + sys.exit(coverage.cmdline.main()) + """.format(abc=abc)) + self.make_file("how_is_it.py", """\ + import pprint, sys + pprint.pprint(sys.path) + import thing + print(thing.SOMETHING) + """) + # If this test fails, it will be with "can't import thing". + out = self.run_command("python run_coverage.py run how_is_it.py") + self.assertIn("hello-xyzzy", out) + + out = self.run_command("python -m run_coverage run how_is_it.py") + self.assertIn("hello-xyzzy", out) + class ExcepthookTest(CoverageTest): """Tests of sys.excepthook support.""" |