summaryrefslogtreecommitdiffstats
path: root/tests/test_wheel.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_wheel.py')
-rw-r--r--tests/test_wheel.py221
1 files changed, 221 insertions, 0 deletions
diff --git a/tests/test_wheel.py b/tests/test_wheel.py
new file mode 100644
index 0000000..3b88391
--- /dev/null
+++ b/tests/test_wheel.py
@@ -0,0 +1,221 @@
+import configparser
+import os
+from pathlib import Path
+import tempfile
+from unittest import skipIf
+import zipfile
+
+import pytest
+from testpath import assert_isfile, assert_isdir, assert_not_path_exists
+
+from flit.wheel import WheelBuilder, make_wheel_in
+from flit.config import EntryPointsConflict
+
+samples_dir = Path(__file__).parent / 'samples'
+
+
+def unpack(path):
+ z = zipfile.ZipFile(str(path))
+ t = tempfile.TemporaryDirectory()
+ z.extractall(t.name)
+ return t
+
+def test_wheel_module(copy_sample):
+ td = copy_sample('module1_toml')
+ make_wheel_in(td / 'pyproject.toml', td)
+ assert_isfile(td / 'module1-0.1-py2.py3-none-any.whl')
+
+def test_editable_wheel_module(copy_sample):
+ td = copy_sample('module1_toml')
+ make_wheel_in(td / 'pyproject.toml', td, editable=True)
+ whl_file = td / 'module1-0.1-py2.py3-none-any.whl'
+ assert_isfile(whl_file)
+ with unpack(whl_file) as unpacked:
+ pth_path = Path(unpacked, 'module1.pth')
+ assert_isfile(pth_path)
+ assert pth_path.read_text() == str(td)
+ assert_isdir(Path(unpacked, 'module1-0.1.dist-info'))
+
+def test_editable_wheel_has_absolute_pth(copy_sample):
+ td = copy_sample('module1_toml')
+ oldcwd = os.getcwd()
+ os.chdir(str(td))
+ try:
+ make_wheel_in(Path('pyproject.toml'), Path('.'), editable=True)
+ whl_file = 'module1-0.1-py2.py3-none-any.whl'
+ assert_isfile(whl_file)
+ with unpack(whl_file) as unpacked:
+ pth_path = Path(unpacked, 'module1.pth')
+ assert_isfile(pth_path)
+ assert Path(pth_path.read_text()).is_absolute()
+ assert pth_path.read_text() == str(td.resolve())
+ assert_isdir(Path(unpacked, 'module1-0.1.dist-info'))
+ finally:
+ os.chdir(oldcwd)
+
+def test_wheel_package(copy_sample):
+ td = copy_sample('package1')
+ make_wheel_in(td / 'pyproject.toml', td)
+ assert_isfile(td / 'package1-0.1-py2.py3-none-any.whl')
+
+def test_editable_wheel_package(copy_sample):
+ td = copy_sample('package1')
+ make_wheel_in(td / 'pyproject.toml', td, editable=True)
+ whl_file = td / 'package1-0.1-py2.py3-none-any.whl'
+ assert_isfile(whl_file)
+ with unpack(whl_file) as unpacked:
+ pth_path = Path(unpacked, 'package1.pth')
+ assert_isfile(pth_path)
+ assert pth_path.read_text() == str(td)
+ assert_isdir(Path(unpacked, 'package1-0.1.dist-info'))
+
+def test_editable_wheel_namespace_package(copy_sample):
+ td = copy_sample('ns1-pkg')
+ make_wheel_in(td / 'pyproject.toml', td, editable=True)
+ whl_file = td / 'ns1_pkg-0.1-py2.py3-none-any.whl'
+ assert_isfile(whl_file)
+ with unpack(whl_file) as unpacked:
+ pth_path = Path(unpacked, 'ns1.pkg.pth')
+ assert_isfile(pth_path)
+ assert pth_path.read_text() == str(td)
+ assert_isdir(Path(unpacked, 'ns1_pkg-0.1.dist-info'))
+
+def test_wheel_src_module(copy_sample):
+ td = copy_sample('module3')
+ make_wheel_in(td / 'pyproject.toml', td)
+
+ whl_file = td / 'module3-0.1-py2.py3-none-any.whl'
+ assert_isfile(whl_file)
+ with unpack(whl_file) as unpacked:
+ assert_isfile(Path(unpacked, 'module3.py'))
+ assert_isdir(Path(unpacked, 'module3-0.1.dist-info'))
+ assert_isfile(Path(unpacked, 'module3-0.1.dist-info', 'LICENSE'))
+
+def test_editable_wheel_src_module(copy_sample):
+ td = copy_sample('module3')
+ make_wheel_in(td / 'pyproject.toml', td, editable=True)
+ whl_file = td / 'module3-0.1-py2.py3-none-any.whl'
+ assert_isfile(whl_file)
+ with unpack(whl_file) as unpacked:
+ pth_path = Path(unpacked, 'module3.pth')
+ assert_isfile(pth_path)
+ assert pth_path.read_text() == str(td / "src")
+ assert_isdir(Path(unpacked, 'module3-0.1.dist-info'))
+
+def test_wheel_src_package(copy_sample):
+ td = copy_sample('package2')
+ make_wheel_in(td / 'pyproject.toml', td)
+
+ whl_file = td / 'package2-0.1-py2.py3-none-any.whl'
+ assert_isfile(whl_file)
+ with unpack(whl_file) as unpacked:
+ print(os.listdir(unpacked))
+ assert_isfile(Path(unpacked, 'package2', '__init__.py'))
+
+def test_editable_wheel_src_package(copy_sample):
+ td = copy_sample('package2')
+ make_wheel_in(td / 'pyproject.toml', td, editable=True)
+ whl_file = td / 'package2-0.1-py2.py3-none-any.whl'
+ assert_isfile(whl_file)
+ with unpack(whl_file) as unpacked:
+ pth_path = Path(unpacked, 'package2.pth')
+ assert_isfile(pth_path)
+ assert pth_path.read_text() == str(td / "src")
+ assert_isdir(Path(unpacked, 'package2-0.1.dist-info'))
+
+
+def test_wheel_ns_package(copy_sample):
+ td = copy_sample('ns1-pkg')
+ res = make_wheel_in(td / 'pyproject.toml', td)
+ assert res.file == td / 'ns1_pkg-0.1-py2.py3-none-any.whl'
+ assert_isfile(res.file)
+ with unpack(res.file) as td_unpack:
+ assert_isdir(Path(td_unpack, 'ns1_pkg-0.1.dist-info'))
+ assert_isfile(Path(td_unpack, 'ns1', 'pkg', '__init__.py'))
+ assert_not_path_exists(Path(td_unpack, 'ns1', '__init__.py'))
+
+def test_dist_name(copy_sample):
+ td = copy_sample('altdistname')
+ make_wheel_in(td / 'pyproject.toml', td)
+ res = td / 'package_dist1-0.1-py2.py3-none-any.whl'
+ assert_isfile(res)
+ with unpack(res) as td_unpack:
+ assert_isdir(Path(td_unpack, 'package_dist1-0.1.dist-info'))
+
+def test_entry_points(copy_sample):
+ td = copy_sample('entrypoints_valid')
+ make_wheel_in(td / 'pyproject.toml', td)
+ assert_isfile(td / 'package1-0.1-py2.py3-none-any.whl')
+ with unpack(td / 'package1-0.1-py2.py3-none-any.whl') as td_unpack:
+ entry_points = Path(td_unpack, 'package1-0.1.dist-info', 'entry_points.txt')
+ assert_isfile(entry_points)
+ cp = configparser.ConfigParser()
+ cp.read(str(entry_points))
+ assert 'console_scripts' in cp.sections()
+ assert 'myplugins' in cp.sections()
+
+def test_entry_points_conflict(copy_sample):
+ td = copy_sample('entrypoints_conflict')
+ with pytest.raises(EntryPointsConflict):
+ make_wheel_in(td / 'pyproject.toml', td)
+
+def test_wheel_builder():
+ # Slightly lower level interface
+ with tempfile.TemporaryDirectory() as td:
+ target = Path(td, 'sample.whl')
+ with target.open('wb') as f:
+ wb = WheelBuilder.from_ini_path(samples_dir / 'package1' / 'pyproject.toml', f)
+ wb.build()
+
+ assert zipfile.is_zipfile(str(target))
+ assert wb.wheel_filename == 'package1-0.1-py2.py3-none-any.whl'
+
+@skipIf(os.name == 'nt', 'Windows does not preserve necessary permissions')
+def test_permissions_normed(copy_sample):
+ td = copy_sample('module1_toml')
+
+ (td / 'module1.py').chmod(0o620)
+ make_wheel_in(td / 'pyproject.toml', td)
+
+ whl = td / 'module1-0.1-py2.py3-none-any.whl'
+ assert_isfile(whl)
+ with zipfile.ZipFile(str(whl)) as zf:
+ info = zf.getinfo('module1.py')
+ perms = (info.external_attr >> 16) & 0o777
+ assert perms == 0o644, oct(perms)
+ whl.unlink()
+
+ # This time with executable bit set
+ (td / 'module1.py').chmod(0o720)
+ make_wheel_in(td / 'pyproject.toml', td)
+
+ assert_isfile(whl)
+ with zipfile.ZipFile(str(whl)) as zf:
+ info = zf.getinfo('module1.py')
+ perms = (info.external_attr >> 16) & 0o777
+ assert perms == 0o755, oct(perms)
+
+ info = zf.getinfo('module1-0.1.dist-info/METADATA')
+ perms = (info.external_attr >> 16) & 0o777
+ assert perms == 0o644, oct(perms)
+
+def test_compression(tmp_path):
+ info = make_wheel_in(samples_dir / 'module1_toml' / 'pyproject.toml', tmp_path)
+ assert_isfile(info.file)
+ with zipfile.ZipFile(str(info.file)) as zf:
+ for name in [
+ 'module1.py',
+ 'module1-0.1.dist-info/METADATA',
+ ]:
+ assert zf.getinfo(name).compress_type == zipfile.ZIP_DEFLATED
+
+def test_wheel_module_local_version(copy_sample):
+ """Test if a local version specifier is preserved in wheel filename and dist-info dir name"""
+ td = copy_sample('modulewithlocalversion')
+ make_wheel_in(td / 'pyproject.toml', td)
+
+ whl_file = td / 'modulewithlocalversion-0.1.dev0+test-py2.py3-none-any.whl'
+ assert_isfile(whl_file)
+ with unpack(whl_file) as unpacked:
+ assert_isfile(Path(unpacked, 'modulewithlocalversion.py'))
+ assert_isdir(Path(unpacked, 'modulewithlocalversion-0.1.dev0+test.dist-info'))
generated by cgit v1.2.3 (git 2.39.1) at 2024-06-26 11:56:42 +0000