-
-
Notifications
You must be signed in to change notification settings - Fork 2.7k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Improve handling of pyargs #3010
Conversation
7ccc72d
to
4e5344d
Compare
|
||
# The structure of the test directory is now: | ||
# . | ||
# ├── local |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I tried to keep close to the actual issue, and kept the virtualenv structure.
4e5344d
to
df6885d
Compare
@@ -645,6 +647,69 @@ def join_pythonpath(*dirs): | |||
"*1 passed*" | |||
]) | |||
|
|||
@pytest.mark.skipif(not hasattr(os, "symlink"), reason="requires symlinks") |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This assumes if the python/os supports symlinks, so does the filesystem the temporary directory is created. This is not a given, but as this is in testing only, and generally this is the case (both in CI and locally) in most workflows, I think this is good enough.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I agree 😁
df6885d
to
b62fd79
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Great work! Thanks for debugging and studying the underlying issue, it is quite a ride.
I made a few comments, please take a look at them when you have the chance.
_pytest/main.py
Outdated
return None | ||
return pkgutil.ImpLoader(fullname, file, filename, etc) | ||
|
||
pkgutil.ImpImporter.find_module = find_module_patched |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I'm a little concerned about patching this globally like it is done here... who knows what we might break by patching this.
I would rather be conservative and patch this only while we are calling find_loader
in the lines below this one. I think a nice way to implement this would be by using a contextmanager
:
try:
with _patched_find_module():
loader = pkgutil.find_loader(x)
except ImportError:
And _patched_find_module
can be implemented like so:
@contextlib.contextmanager
def _patched_find_module():
""" ... """
if six.PY2: # python 3.4+ uses importlib instead
def find_module_patched(self, fullname, path=None):
subname = fullname.split(".")[-1]
if subname != fullname and self.path is None:
return None
if self.path is None:
path = None
else:
path = [self.path]
try:
file, filename, etc = pkgutil.imp.find_module(subname,
path)
except ImportError:
return None
return pkgutil.ImpLoader(fullname, file, filename, etc)
old_find_module = pkgutil.ImpImporter.find_module
pkgutil.ImpImporter.find_module = find_module_patched
try:
yield
finally:
pkgutil.ImpImporter.find_module = old_find_module
else:
yield
We can then even use _patched_find_module()
's docstring to better explain what's going on.
@@ -645,6 +647,69 @@ def join_pythonpath(*dirs): | |||
"*1 passed*" | |||
]) | |||
|
|||
@pytest.mark.skipif(not hasattr(os, "symlink"), reason="requires symlinks") |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I agree 😁
testing/acceptance_test.py
Outdated
# The structure of the test directory is now: | ||
# . | ||
# ├── local | ||
# │ └── lib -> ../world |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I think this comment is incorrect, there's no world
directory being created
testing/acceptance_test.py
Outdated
# └── bar | ||
# ├── __init__.py | ||
# ├── conftest.py | ||
# └── test_world.py |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Ditto here, the code above is creating test_bar.py
testing/acceptance_test.py
Outdated
# └── test_world.py | ||
|
||
def join_pythonpath(*dirs): | ||
cur = py.std.os.environ.get('PYTHONPATH') |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
You can use os.environ
directly, we are planning on dropping py.std
usage eventually
except ImportError: | ||
return x | ||
if loader is None: | ||
return x | ||
# This method is sometimes invoked when AssertionRewritingHook, which | ||
# does not define a get_filename method, is already in place: | ||
try: | ||
path = loader.get_filename(x) | ||
with _patched_find_module(): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
I'm not 100% sure why it is needed here, but I do know the tests fail if you don't have patching in both places.
b462110
to
9703053
Compare
9703053
to
3ca1e4b
Compare
_pytest/main.py
Outdated
@@ -728,17 +730,57 @@ def _tryconvertpyarg(self, x): | |||
"""Convert a dotted module name to path. | |||
|
|||
""" | |||
import pkgutil | |||
@contextlib.contextmanager | |||
def _patched_find_module(): |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This is probably better in the global namespace than a local function because it clutters the code quite a bit. Sorry I wasn't clear about that before.
@RonnyPfannschmidt I would love a second set of eyes on this. |
i did a quick skim and it looks ok, im not fluent with the details either, i wonder if we should look into using importlib |
On python3.4+ we are basically using importlib: pkgutil.find_loader is a thin wrapper around importlib.util.find_spec there. We could use importlib.util.find_spec directly in python3 as I did in this commit, however this is largely equivalent to using pkgutil.find_loader when you check the source so I think at least until python2 support is dropped, it is better to stick with using pkgutil.find_loader. |
@cryvate thanks for elaborating 👍 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks @cryvate!
@RonnyPfannschmidt we can merge IMO unless you want more time to study the PR.
i really dont have the time ^^ |
|
||
monkeypatch.setenv('PYTHONPATH', join_pythonpath(*search_path)) | ||
for p in search_path: | ||
monkeypatch.syspath_prepend(p) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@blueyed
Though git blame will assign the 'blame' to me, actually this part of the code was a copy-paste from the test_cmdline_python_namespace_package
test so I'd pass on the 'blame' to this PR. Regardless, I'd think that the different order (the opposite way around if I'm reading it correctly) is accidental here as well as in the other test.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@cryvate
Thanks and sorry for not looking good enough.
as well as in the other test
Which one do you mean?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@blueyed
Not a problem at all! I meant test_cmdline_python_namespace_package
where I copy-pasted from.
The conversion of pyargs to paths is improved here:
Note that this might not fix the issue in python 3.0-3, as
pkgutil.find_loader
still worked like in python2.7 but as it is not supported I did not look into it.I couldn't just use the backport of importlib, as it does not have importlib.find_loader or importlib.util.find_spec which is what is needed. The current (3.4+) pkgutil.find_loader uses importlib.util.find_spec under the hood, instead of parts of pkgutil.