Skip to content

Commit

Permalink
Include subdirectory URL fragment in the cache key
Browse files Browse the repository at this point in the history
  • Loading branch information
sbidoul committed Nov 10, 2019
1 parent 53770a7 commit 169e2e4
Show file tree
Hide file tree
Showing 3 changed files with 42 additions and 7 deletions.
1 change: 1 addition & 0 deletions news/7333.bugfix
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
Include ``subdirectory`` URL fragments in the cache keys.
7 changes: 7 additions & 0 deletions src/pip/_internal/cache.py
Original file line number Diff line number Diff line change
Expand Up @@ -58,6 +58,10 @@ def _get_cache_path_parts(self, link):
key_parts = [link.url_without_fragment]
if link.hash_name is not None and link.hash is not None:
key_parts.append("=".join([link.hash_name, link.hash]))
if link.subdirectory_fragment:
key_parts.append(
"=".join(["subdirectory", link.subdirectory_fragment])
)
key_url = "#".join(key_parts)

# Encode our key url with sha224, we'll use this because it has similar
Expand Down Expand Up @@ -168,11 +172,14 @@ def get(
# type: (...) -> Link
candidates = []

canonical_package_name = canonicalize_name(package_name)
for wheel_name in self._get_candidates(link, package_name):
try:
wheel = Wheel(wheel_name)
except InvalidWheelFilename:
continue
if wheel.name != canonical_package_name:
continue
if not wheel.supported(supported_tags):
# Built for a different python/arch/etc
continue
Expand Down
41 changes: 34 additions & 7 deletions tests/unit/test_cache.py
Original file line number Diff line number Diff line change
@@ -1,13 +1,40 @@
from pip._internal.cache import WheelCache
from pip._internal.models.format_control import FormatControl
from pip._internal.models.link import Link
from pip._internal.utils.compat import expanduser


class TestWheelCache:
def test_expands_path():
wc = WheelCache("~/.foo/", None)
assert wc.cache_dir == expanduser("~/.foo/")

def test_expands_path(self):
wc = WheelCache("~/.foo/", None)
assert wc.cache_dir == expanduser("~/.foo/")

def test_falsey_path_none(self):
wc = WheelCache(False, None)
assert wc.cache_dir is None
def test_falsey_path_none():
wc = WheelCache(False, None)
assert wc.cache_dir is None


def test_subdirectory_fragment():
"""
Test the subdirectory URL fragment is part of the cache key.
"""
wc = WheelCache("~/.foo/", None)
link1 = Link("git+https://g.c/o/r#subdirectory=d1")
link2 = Link("git+https://g.c/o/r#subdirectory=d2")
assert wc.get_path_for_link(link1) != wc.get_path_for_link(link2)


def test_wheel_name_filter(tmpdir):
"""
Test the wheel cache filters on wheel name when several wheels
for different package are stored under the same cache directory.
"""
wc = WheelCache(tmpdir, FormatControl())
link = Link("https://g.c/package.tar.gz")
cache_path = wc.get_path_for_link(link)
cache_path.mkdir(parents=True)
(cache_path / "package-1.0-py3-none-any.whl").touch()
# package matches wheel name
assert wc.get(link, "package", [("py3", "none", "any")]) is not link
# package2 does not match wheel name
assert wc.get(link, "package2", [("py3", "none", "any")]) is link

0 comments on commit 169e2e4

Please sign in to comment.