118 lines
4.1 KiB
Python
118 lines
4.1 KiB
Python
import os
|
|
from pathlib import WindowsPath, PureWindowsPath
|
|
|
|
import pytest
|
|
|
|
from pypiserver.pkg_helpers import guess_pkgname_and_version, is_listed_path
|
|
|
|
files = [
|
|
("pytz-2012b.tar.bz2", "pytz", "2012b"),
|
|
("pytz-2012b.tgz", "pytz", "2012b"),
|
|
("pytz-2012b.ZIP", "pytz", "2012b"),
|
|
("pytz-2012a.zip", "pytz", "2012a"),
|
|
("pytz-2012b.tar.xz", "pytz", "2012b"),
|
|
("gevent-1.0b1.win32-py2.6.exe", "gevent", "1.0b1"),
|
|
("gevent-1.0b1.win32-py2.7.msi", "gevent", "1.0b1"),
|
|
("greenlet-0.3.4-py3.1-win-amd64.egg", "greenlet", "0.3.4"),
|
|
("greenlet-0.3.4.win-amd64-py3.2.exe", "greenlet", "0.3.4"),
|
|
("greenlet-0.3.4-py3.2-win32.egg", "greenlet", "0.3.4"),
|
|
("greenlet-0.3.4-py2.7-linux-x86_64.egg", "greenlet", "0.3.4"),
|
|
("pep8-0.6.0.zip", "pep8", "0.6.0"),
|
|
("ABC12-34_V1X-1.2.3.zip", "ABC12", "34_V1X-1.2.3"),
|
|
("A100-200-XYZ-1.2.3.zip", "A100-200-XYZ", "1.2.3"),
|
|
("flup-1.0.3.dev-20110405.tar.gz", "flup", "1.0.3.dev-20110405"),
|
|
("package-1.0.0-alpha.1.zip", "package", "1.0.0-alpha.1"),
|
|
("package-1.3.7+build.11.e0f985a.zip", "package", "1.3.7+build.11.e0f985a"),
|
|
("package-v1-8.1.301.ga0df26f.zip", "package-v1", "8.1.301.ga0df26f"),
|
|
("package-v1.1-8.1.301.ga0df26f.zip", "package-v1.1", "8.1.301.ga0df26f"),
|
|
("package-2013.02.17.dev123.zip", "package", "2013.02.17.dev123"),
|
|
("package-20000101.zip", "package", "20000101"),
|
|
("flup-123-1.0.3.dev-20110405.tar.gz", "flup-123", "1.0.3.dev-20110405"),
|
|
("package-123-1.0.0-alpha.1.zip", "package-123", "1.0.0-alpha.1"),
|
|
(
|
|
"package-123-1.3.7+build.11.e0f985a.zip",
|
|
"package-123",
|
|
"1.3.7+build.11.e0f985a",
|
|
),
|
|
("package-123-v1.1_3-8.1.zip", "package-123-v1.1_3", "8.1"),
|
|
("package-123-2013.02.17.dev123.zip", "package-123", "2013.02.17.dev123"),
|
|
("package-123-20000101.zip", "package-123", "20000101"),
|
|
(
|
|
"pyelasticsearch-0.5-brainbot-1-20130712.zip",
|
|
"pyelasticsearch",
|
|
"0.5-brainbot-1-20130712",
|
|
),
|
|
("pywin32-217-cp27-none-win32.whl", "pywin32", "217"),
|
|
("pywin32-217-55-cp27-none-win32.whl", "pywin32", "217-55"),
|
|
("pywin32-217.1-cp27-none-win32.whl", "pywin32", "217.1"),
|
|
("package.zip", "package", ""),
|
|
(
|
|
"package-name-0.0.1.dev0.linux-x86_64.tar.gz",
|
|
"package-name",
|
|
"0.0.1.dev0",
|
|
),
|
|
(
|
|
"package-name-0.0.1.dev0.macosx-10.10-intel.tar.gz",
|
|
"package-name",
|
|
"0.0.1.dev0",
|
|
),
|
|
(
|
|
"package-name-0.0.1.alpha.1.win-amd64-py3.2.exe",
|
|
"package-name",
|
|
"0.0.1.alpha.1",
|
|
),
|
|
("pkg-3!1.0-0.1.tgz", "pkg", "3!1.0-0.1"), # TO BE FIXED
|
|
("pkg-3!1+.0-0.1.tgz", "pkg", "3!1+.0-0.1"), # TO BE FIXED
|
|
("pkg.zip", "pkg", ""),
|
|
("foo/pkg.zip", "pkg", ""),
|
|
("foo/pkg-1b.zip", "pkg", "1b"),
|
|
("foo/pywin32-217.1-cp27-none-win32.whl", "pywin32", "217.1"),
|
|
(
|
|
"package-name-0.0.1.alpha.1.win-amd64-py3.2.exe",
|
|
"package-name",
|
|
"0.0.1.alpha.1",
|
|
),
|
|
]
|
|
|
|
|
|
def _capitalize_ext(fpath):
|
|
f, e = os.path.splitext(fpath)
|
|
if e != ".whl":
|
|
e = e.upper()
|
|
return f + e
|
|
|
|
|
|
@pytest.mark.parametrize(("filename", "pkgname", "version"), files)
|
|
def test_guess_pkgname_and_version(filename, pkgname, version):
|
|
exp = (pkgname, version)
|
|
assert guess_pkgname_and_version(filename) == exp
|
|
assert guess_pkgname_and_version(_capitalize_ext(filename)) == exp
|
|
|
|
|
|
@pytest.mark.parametrize(("filename", "pkgname", "version"), files)
|
|
def test_guess_pkgname_and_version_asc(filename, pkgname, version):
|
|
exp = (pkgname, version)
|
|
filename = f"{filename}.asc"
|
|
assert guess_pkgname_and_version(filename) == exp
|
|
|
|
|
|
invalid_files = ["some_file", "some_file.ext", "some_wheel.whl"]
|
|
|
|
|
|
@pytest.mark.parametrize("filename", invalid_files)
|
|
def test_guess_pkgname_and_version_invalid_files(filename):
|
|
assert guess_pkgname_and_version(filename) is None
|
|
|
|
|
|
paths = [
|
|
("/some/path", True),
|
|
(PureWindowsPath(r"c:\some\windows\path"), True),
|
|
("/.hidden", False),
|
|
(PureWindowsPath(r"c:\.hidden\windows\path"), False),
|
|
]
|
|
|
|
|
|
@pytest.mark.parametrize(("pathname", "allowed"), paths)
|
|
def test_allowed_path_check(pathname, allowed):
|
|
assert is_listed_path(pathname) == allowed
|