Function test_check

tests/test_pex_builder.py:522–607

(tmpdir)

    reason="The cat binary is required for this test.",
)
def test_check(tmpdir):
    # type: (Any) -> None

    def assert_perform_check_zip64_handling(
        zipapp,  # type: str
        test_run=True,  # type: bool
    ):
        # type: (...) -> None
        assert Check.NONE.perform_check(Layout.ZIPAPP, zipapp) is None
        assert Check.ERROR.perform_check(Layout.PACKED, zipapp) is None
        assert Check.ERROR.perform_check(Layout.LOOSE, zipapp) is None

        # Python 3.13 newly supports ZIP64 in `zipimport.zipimporter`.
        if sys.version_info >= (3, 13):
            assert Check.WARN.perform_check(Layout.ZIPAPP, zipapp) is True
            assert Check.ERROR.perform_check(Layout.ZIPAPP, zipapp) is True
            if test_run:
                subprocess.check_call(args=[sys.executable, zipapp])
        else:
            expected_error_message_re = r"The PEX zip at {path} is not a valid zipapp: ".format(
                path=re.escape(zipapp)
            )
            with pytest.warns(PEXWarning, match=expected_error_message_re):
                assert Check.WARN.perform_check(Layout.ZIPAPP, zipapp) is False
            with pytest.raises(InvalidZipAppError, match=expected_error_message_re):
                Check.ERROR.perform_check(Layout.ZIPAPP, zipapp)

            if test_run:
                assert subprocess.call(args=[sys.executable, zipapp]) != 0

    @contextmanager
    def write_zipapp(path):
        # type: (str) -> Iterator[ZipFile]
        with open_zip(path, "w") as zip_file:
            yield zip_file
            zip_file.writestr("__main__.py", "print('BOOTED')")

    # N.B.: This test creates very large files in /tmp; so to relieve possible memory pressure in
    # tmpfs-based /tmp filesystems, we unlink files as soon as we don't need them instead of waiting
    # for tmpdir fixture cleanup at the end of the test.

    file_too_big = os.path.join(str(tmpdir), "file_too_big.py")
    with open(file_too_big, "w") as fp:
        fp.write("\n")
    accum = file_too_big + ".accum"
    for _ in range(32):
        safe_rename(file_too_big, accum)
        with open(file_too_big, "wb") as dest:
            subprocess.check_call(args=["cat", accum, accum], stdout=dest.fileno())
    os.unlink(accum)
    assert os.path.getsize(file_too_big) > 0xFFFFFFFF, (
        "ZIP64 must be used if file sizes are bigger than 0xFFFFFFFF per "
        "https://pkware.cachefly.net/webdocs/casestudies/APPNOTE.TXT 4.3.9.2"
    )

    zipapp_too_big = os.path.join(str(tmpdir), "too-big.pyz")
    with write_zipapp(zipapp_too_big) as zf:
        zf.write(file_too_big, os.path.basename(file_too_big))