packaging/test_release.py

Rama main · réplica de lectura del repositorio original.

"""Regression checks for publication boundaries and command installation."""
import contextlib
import io
import zipfile
import json
from pathlib import Path
import tarfile
import tempfile
import unittest

import release


class ReleaseTests(unittest.TestCase):
    def test_exact_assets_and_corruption(self):
        with tempfile.TemporaryDirectory() as temp:
            path = Path(temp)
            names = release.expected_assets("11.5.2")
            self.assertEqual(len(names), 34)
            self.assertEqual(sum(n.endswith(".deb") for n in names), 6)
            self.assertEqual(sum(n.endswith(".rpm") for n in names), 6)
            for name in names - {"croc_v11.5.2_checksums.txt"}:
                if name.endswith(".zip"):
                    with zipfile.ZipFile(path / name, "w") as archive:
                        archive.writestr("croc.exe", b"croc")
                        archive.writestr("croc-update.exe", b"updater")
                elif name.endswith(".tar.gz") and not name.endswith("_src.tar.gz") and not name.startswith("croc-web_"):
                    with tarfile.open(path / name, "w:gz") as archive:
                        for binary in ("croc", "croc-update"):
                            entry = tarfile.TarInfo(binary)
                            entry.size = 4
                            entry.mode = 0o755
                            archive.addfile(entry, io.BytesIO(b"test"))
                else:
                    (path / name).write_bytes(b"test asset")
            with contextlib.redirect_stdout(io.StringIO()):
                release.check_assets(path, "11.5.2", write=True)
                release.check_assets(path, "11.5.2")
            (path / "croc_11.5.2-1_armel.deb").write_bytes(b"corrupt")
            with self.assertRaisesRegex(ValueError, "checksums"):
                release.check_assets(path, "11.5.2")
            (path / "croc_11.5.2-1_armel.deb").unlink()
            with self.assertRaisesRegex(ValueError, "missing=.*armel"):
                release.check_assets(path, "11.5.2", write=True)

    def test_unexpected_assets_block_publication(self):
        with tempfile.TemporaryDirectory() as temp:
            path = Path(temp)
            for name in release.expected_assets("11.5.2"):
                (path / name).touch()
            (path / "croc_11.5.2-1_armv5.deb").touch()
            with self.assertRaisesRegex(ValueError, "unexpected=.*armv5"):
                release.check_assets(path, "11.5.2", write=True)

    def test_remote_release_must_be_draft_with_exact_assets(self):
        with tempfile.TemporaryDirectory() as temp:
            path = Path(temp) / "release.json"
            names = sorted(release.expected_assets("11.5.2"))
            metadata = {"isDraft": True, "assets": [{"name": name} for name in names]}
            path.write_text(json.dumps(metadata))
            release.check_draft(path, "11.5.2")
            metadata["isDraft"] = False
            path.write_text(json.dumps(metadata))
            with self.assertRaises(ValueError):
                release.check_draft(path, "11.5.2")
            metadata["isDraft"] = True
            metadata["assets"][-1] = metadata["assets"][0]
            path.write_text(json.dumps(metadata))
            with self.assertRaises(ValueError):
                release.check_draft(path, "11.5.2")

    def test_archive_must_contain_one_regular_binary(self):
        with tempfile.TemporaryDirectory() as temp:
            path = Path(temp)
            for name in ("../croc", "croc"):
                with tarfile.open(path / "archive.tar", "w") as archive:
                    entry = tarfile.TarInfo(name)
                    entry.type = tarfile.SYMTYPE
                    entry.linkname = "/usr/bin/croc"
                    archive.addfile(entry)
                with self.assertRaisesRegex(ValueError, "regular croc"):
                    release.unpack_binary(path / "archive.tar", path / "binary")

    def test_standalone_archive_requires_both_binaries(self):
        with tempfile.TemporaryDirectory() as temp:
            archive = Path(temp) / "croc.tar.gz"
            with tarfile.open(archive, "w:gz") as stream:
                entry = tarfile.TarInfo("croc")
                entry.size = 4
                stream.addfile(entry, io.BytesIO(b"test"))
            with self.assertRaisesRegex(ValueError, "croc-update"):
                release.check_cli_archive(archive)

    def test_stable_versions_only(self):
        for bad in ("v01.2.3", "11.5.2-rc1", "11.5", "../11.5.2"):
            with self.assertRaises(ValueError):
                release.version(bad)


if __name__ == "__main__":
    unittest.main()