niceSOFT/python3-wheel: The official binary distribution format for Python

tests/cli/test_pack.py
62 строки2 KB
Alex Grönholm
Fixed wheel pack duplicating WHEEL contents on build number change
15 авг 2021, 23:58
15 авг 2021, 23:583f1a73a
import os
from textwrap import dedent
from zipfile import ZipFile

import pytest

from wheel.cli.pack import pack

THISDIR = os.path.dirname(__file__)
TESTWHEEL_NAME = 'test-1.0-py2.py3-none-any.whl'
TESTWHEEL_PATH = os.path.join(THISDIR, '..', 'testdata', TESTWHEEL_NAME)


@pytest.mark.filterwarnings('error:Duplicate name')
@pytest.mark.parametrize('build_tag_arg, existing_build_tag, filename', [
    (None, None, 'test-1.0-py2.py3-none-any.whl'),
    ('2b', None, 'test-1.0-2b-py2.py3-none-any.whl'),
    (None, '3', 'test-1.0-3-py2.py3-none-any.whl'),
    ('', '3', 'test-1.0-py2.py3-none-any.whl'),
], ids=['nobuildnum', 'newbuildarg', 'oldbuildnum', 'erasebuildnum'])
def test_pack(tmpdir_factory, tmpdir, build_tag_arg, existing_build_tag, filename):
    unpack_dir = tmpdir_factory.mktemp('wheeldir')
    with ZipFile(TESTWHEEL_PATH) as zf:
        old_record = zf.read('test-1.0.dist-info/RECORD')
        old_record_lines = sorted(line.rstrip() for line in old_record.split(b'\n')
                                  if line and not line.startswith(b'test-1.0.dist-info/WHEEL,'))
        zf.extractall(str(unpack_dir))

    if existing_build_tag:
        # Add the build number to WHEEL
        wheel_file_path = unpack_dir.join('test-1.0.dist-info').join('WHEEL')
        wheel_file_content = wheel_file_path.read_binary()
        assert b'Build' not in wheel_file_content
        wheel_file_content += b'Build: 3\r\n'
        wheel_file_path.write_binary(wheel_file_content)

    pack(str(unpack_dir), str(tmpdir), build_tag_arg)
    new_wheel_path = tmpdir.join(filename)
    assert new_wheel_path.isfile()

    with ZipFile(str(new_wheel_path)) as zf:
        new_record = zf.read('test-1.0.dist-info/RECORD')
        new_record_lines = sorted(line.rstrip() for line in new_record.split(b'\n')
                                  if line and not line.startswith(b'test-1.0.dist-info/WHEEL,'))

        new_wheel_file_content = zf.read('test-1.0.dist-info/WHEEL')

    assert new_record_lines == old_record_lines

    expected_build_num = build_tag_arg or existing_build_tag
    expected_wheel_content = dedent("""\
        Wheel-Version: 1.0
        Generator: bdist_wheel (0.30.0)
        Root-Is-Purelib: false
        Tag: py2-none-any
        Tag: py3-none-any
    """.replace('\n', '\r\n'))
    if expected_build_num:
        expected_wheel_content += 'Build: %s\r\n' % expected_build_num

    expected_wheel_content = expected_wheel_content.encode('ascii')
    assert new_wheel_file_content == expected_wheel_content