# HG changeset patch # Parent f5e20abea8712ade4f64de403e67fab9f2fe6bf9 diff -r f5e20abea871 Lib/site.py --- a/Lib/site.py Sun May 15 23:52:50 2016 -0400 +++ b/Lib/site.py Mon May 16 04:07:59 2016 +0000 @@ -527,8 +527,9 @@ known_paths = removeduppaths() if ENABLE_USER_SITE is None: ENABLE_USER_SITE = check_enableusersite() - known_paths = addusersitepackages(known_paths) - known_paths = addsitepackages(known_paths) + if os.environ.get("_PYTHONNOSITEPACKAGES") is None: + known_paths = addusersitepackages(known_paths) + known_paths = addsitepackages(known_paths) if sys.platform == 'os2emx': setBEGINLIBPATH() setquit() diff -r f5e20abea871 Lib/test/regrtest.py --- a/Lib/test/regrtest.py Sun May 15 23:52:50 2016 -0400 +++ b/Lib/test/regrtest.py Mon May 16 04:07:59 2016 +0000 @@ -169,6 +169,7 @@ import imp import platform import sysconfig +from subprocess import Popen, PIPE # Some times __path__ and __file__ are not absolute (e.g. while running from @@ -464,6 +465,50 @@ test_support.use_resources = use_resources save_modules = set(sys.modules) + opt_args = test_support.args_from_interpreter_flags() + base_cmd = [sys.executable] + opt_args + ['-m', 'test.regrtest'] + debug_output_pat = re.compile(r"\[\d+ refs\]$") + + def get_args_tuple(test, verbose, quiet, huntrleaks, use_resources, pgo): + return ( + (test, verbose, quiet), + dict(huntrleaks=huntrleaks, use_resources=use_resources, + pgo=pgo) + ) + + def _runtest(test, verbose, quiet, huntrleaks=False, + use_resources=None, pgo=False): + if test == "test_site": + args_tuple = get_args_tuple(test, verbose, quiet, huntrleaks, + use_resources, pgo=pgo) + env = os.environ.copy() + try: + del env["_PYTHONNOSITEPACKAGES"] + except KeyError: + pass + popen = Popen(base_cmd + ['--slaveargs', json.dumps(args_tuple)], + stdout=PIPE, stderr=PIPE, + universal_newlines=True, + close_fds=(os.name != 'nt'), + env=env) + stdout, stderr = popen.communicate() + # Strip last refcount output line if it exists, since it + # comes from the shutdown of the interpreter in the subcommand. + stderr = debug_output_pat.sub("", stderr) + stdout, _, result = stdout.strip().rpartition("\n") + result = json.loads(result) + if stdout: + print stdout + if stderr: + print >>sys.stderr, stderr + if result[0] == INTERRUPTED: + assert result[1] == 'KeyboardInterrupt' + raise KeyboardInterrupt + return result + else: + return runtest(test, verbose, quiet, huntrleaks=False, + use_resources=None, pgo=pgo) + def accumulate_result(test, result): ok, test_time = result test_times.append((test_time, test)) @@ -501,20 +546,13 @@ print "Multiprocess option requires thread support" sys.exit(2) from Queue import Queue - from subprocess import Popen, PIPE - debug_output_pat = re.compile(r"\[\d+ refs\]$") output = Queue() def tests_and_args(): for test in tests: - args_tuple = ( - (test, verbose, quiet), - dict(huntrleaks=huntrleaks, use_resources=use_resources, - pgo=pgo) - ) + args_tuple = get_args_tuple(test, verbose, quiet, huntrleaks, + use_resources, pgo=pgo) yield (test, args_tuple) pending = tests_and_args() - opt_args = test_support.args_from_interpreter_flags() - base_cmd = [sys.executable] + opt_args + ['-m', 'test.regrtest'] # required to spawn a new process with PGO flag on/off if pgo: base_cmd = base_cmd + ['--pgo'] @@ -587,16 +625,16 @@ if trace: # If we're tracing code coverage, then we don't exit with status # if on a false return value from main. - tracer.runctx('runtest(test, verbose, quiet)', + tracer.runctx('_runtest(test, verbose, quiet)', globals=globals(), locals=vars()) else: try: - result = runtest(test, verbose, quiet, huntrleaks, None, pgo) + result = _runtest(test, verbose, quiet, huntrleaks, None, pgo) accumulate_result(test, result) if verbose3 and result[0] == FAILED: if not pgo: print "Re-running test %r in verbose mode" % test - runtest(test, True, quiet, huntrleaks, None, pgo) + _runtest(test, True, quiet, huntrleaks, None, pgo) except KeyboardInterrupt: interrupted = True break @@ -664,7 +702,7 @@ sys.stdout.flush() try: test_support.verbose = True - ok = runtest(test, True, quiet, huntrleaks, None, pgo) + ok = _runtest(test, True, quiet, huntrleaks, None, pgo) except KeyboardInterrupt: # print a newline separate from the ^C print @@ -893,8 +931,9 @@ for name, get, restore in self.resource_info(): current = get() original = saved_values.pop(name) - # Check for changes to the resource's value - if current != original: + # Check for changes to the resource's value. test_site is always run + # in a subprocess and is allowed to change os.environ and sys.path. + if current != original and self.testname != "test_site": self.changed = True restore(original) if not self.quiet and not self.pgo: diff -r f5e20abea871 Lib/test/test_site.py --- a/Lib/test/test_site.py Sun May 15 23:52:50 2016 -0400 +++ b/Lib/test/test_site.py Mon May 16 04:07:59 2016 +0000 @@ -8,6 +8,7 @@ from test.test_support import run_unittest, TESTFN, EnvironmentVarGuard from test.test_support import captured_output import __builtin__ +import imp import os import sys import re @@ -21,6 +22,9 @@ # already. if "site" in sys.modules: import site + if "_PYTHONNOSITEPACKAGES" in os.environ: + del os.environ["_PYTHONNOSITEPACKAGES"] + imp.reload(site) else: raise unittest.SkipTest("importation of site.py suppressed") diff -r f5e20abea871 Makefile.pre.in --- a/Makefile.pre.in Sun May 15 23:52:50 2016 -0400 +++ b/Makefile.pre.in Mon May 16 04:07:59 2016 +0000 @@ -862,7 +862,7 @@ TESTOPTS= -l $(EXTRATESTOPTS) TESTPROG= $(srcdir)/Lib/test/regrtest.py -TESTPYTHON= $(RUNSHARED) ./$(BUILDPYTHON) -Wd -3 -E -tt $(TESTPYTHONOPTS) +TESTPYTHON= _PYTHONNOSITEPACKAGES=1 $(RUNSHARED) ./$(BUILDPYTHON) -Wd -3 -E -tt $(TESTPYTHONOPTS) test: all platform -find $(srcdir)/Lib -name '*.py[co]' -print | xargs rm -f -$(TESTPYTHON) $(TESTPROG) $(TESTOPTS)