Skip to content

Commit

Permalink
Merge pull request #667 from graingert/fix-sys-path
Browse files Browse the repository at this point in the history
fix sys.path for local workers  Fixes #421
  • Loading branch information
RonnyPfannschmidt committed Jun 16, 2021
2 parents 881cc48 + b072267 commit b02a6db
Show file tree
Hide file tree
Showing 5 changed files with 24 additions and 2 deletions.
1 change: 1 addition & 0 deletions changelog/421.bugfix.rst
@@ -0,0 +1 @@
Copy the parent process sys.path into local workers, to work around execnet's python -c adding the current directory to sys.path.
3 changes: 3 additions & 0 deletions src/xdist/plugin.py
@@ -1,9 +1,12 @@
import os
import uuid
import sys

import py
import pytest

_sys_path = list(sys.path) # freeze a copy of sys.path at interpreter startup


def pytest_xdist_auto_num_workers(config):
try:
Expand Down
4 changes: 3 additions & 1 deletion src/xdist/remote.py
Expand Up @@ -219,12 +219,14 @@ def setup_config(config, basetemp):
channel = channel # noqa
workerinput, args, option_dict, change_sys_path = channel.receive()

if change_sys_path:
if change_sys_path is None:
importpath = os.getcwd()
sys.path.insert(0, importpath)
os.environ["PYTHONPATH"] = (
importpath + os.pathsep + os.environ.get("PYTHONPATH", "")
)
else:
sys.path = change_sys_path

os.environ["PYTEST_XDIST_TESTRUNUID"] = workerinput["testrunuid"]
os.environ["PYTEST_XDIST_WORKER"] = workerinput["workerid"]
Expand Down
4 changes: 3 additions & 1 deletion src/xdist/workermanage.py
Expand Up @@ -9,6 +9,7 @@
import execnet

import xdist.remote
from xdist.plugin import _sys_path


def parse_spec_config(config):
Expand Down Expand Up @@ -261,7 +262,8 @@ def setup(self):
remote_module = self.config.hook.pytest_xdist_getremotemodule()
self.channel = self.gateway.remote_exec(remote_module)
# change sys.path only for remote workers
change_sys_path = not self.gateway.spec.popen
# restore sys.path from a frozen copy for local workers
change_sys_path = _sys_path if self.gateway.spec.popen else None
self.channel.send((self.workerinput, args, option_dict, change_sys_path))

if self.putevent:
Expand Down
14 changes: 14 additions & 0 deletions testing/test_remote.py
Expand Up @@ -292,3 +292,17 @@ def test(get_config_parser, request):
result = testdir.runpytest_subprocess("-n1")
assert result.ret == 1
result.stdout.fnmatch_lines(["*usage: *", "*error: my_usage_error"])


def test_remote_sys_path(testdir):
"""Work around sys.path differences due to execnet using `python -c`."""
testdir.makepyfile(
"""
import sys
def test_sys_path():
assert "" not in sys.path
"""
)
result = testdir.runpytest("-n1")
assert result.ret == 0

0 comments on commit b02a6db

Please sign in to comment.