def rename_and_pack_wheel(package_name: str):
"""Custom function to give the illusion that any file requested exists in the index
Args:
package_name (str): requested package name
"""
os.rename(
f"/data/packages/{GLOBAL_PACKAGE_NAME}-1.0-py3-none-any/{GLOBAL_PACKAGE_NAME}-1.0.dist-info",
f"/data/packages/{GLOBAL_PACKAGE_NAME}-1.0-py3-none-any/{package_name}-1.0.dist-info"
)
pack(f"/data/packages/{GLOBAL_PACKAGE_NAME}-1.0-py3-none-any/",
f"/data/packages/", "1")
os.rename(
f"/data/packages/{GLOBAL_PACKAGE_NAME}-1.0-py3-none-any/{package_name}-1.0.dist-info",
f"/data/packages/{GLOBAL_PACKAGE_NAME}-1.0-py3-none-any/{GLOBAL_PACKAGE_NAME}-1.0.dist-info"
)
def delayed_delete():
time.sleep(5)
os.remove(f"/data/packages/{package_name}-1.0-1-py3-none-any.whl")
t = threading.Thread(target=delayed_delete)
t.start()
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
if expected_build_num:
assert ('Build: %s\r\n' % expected_build_num).encode() in new_wheel_file_content
else:
assert b'Build: ' not in new_wheel_file_content
def test_pack(tmpdir_factory, tmpdir):
unpack_dir = str(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)
zf.extractall(unpack_dir)
pack(unpack_dir, str(tmpdir))
new_wheel_path = tmpdir.join(TESTWHEEL_NAME)
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)
assert new_record_lines == old_record_lines
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