Something went wrong. Try again.
My version of https://github.com/vercel-labs/fx
Something went wrong. Try again.
Python
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280from __future__ import annotations
import jsonimport pathlibimport subprocessimport sysimport tempfileimport unittest
from scripts.binary_size import ( BinarySizeError, append_delta_table, build_report, markdown_report, parse_macho_sections,)
REPO_ROOT = pathlib.Path(__file__).resolve().parents[2]WORKFLOW_PATH = REPO_ROOT / ".github" / "workflows" / "binary-size.yml"
class BinarySizeCliTests(unittest.TestCase): def test_empty_delta_table_uses_generic_message(self) -> None: lines: list[str] = []
append_delta_table(lines, "## Segment changes", {})
self.assertEqual(["", "## Segment changes", "", "No changes detected."], lines)
def test_threshold_increase_emits_warning_and_exact_evidence(self) -> None: with tempfile.TemporaryDirectory(prefix="fx-binary-size-") as tmp: root = pathlib.Path(tmp) base_binary = root / "base-fx" head_binary = root / "head-fx" base_binary.write_bytes(b"b" * 100_000) head_binary.write_bytes(b"h" * 152_429) base_sections = root / "base-sections.txt" head_sections = root / "head-sections.txt" base_sections.write_text( "Segment __TEXT: 65536\n" "\tSection __text: 50000\n" "\ttotal 50000\n" "Segment __LINKEDIT: 32768\n" "total 98304\n", encoding="utf-8", ) head_sections.write_text( "Segment __TEXT: 114688\n" "\tSection __text: 102429\n" "\ttotal 102429\n" "Segment __LINKEDIT: 32768\n" "total 147456\n", encoding="utf-8", ) json_path = root / "report.json" markdown_path = root / "report.md" github_output = root / "github-output.txt"
result = subprocess.run( [ sys.executable, "-m", "scripts.binary_size", "--base-binary", str(base_binary), "--head-binary", str(head_binary), "--base-sections", str(base_sections), "--head-sections", str(head_sections), "--base-sha", "a" * 40, "--head-sha", "b" * 40, "--target", "aarch64-macos", "--warning-bytes", "52429", "--output-json", str(json_path), "--output-markdown", str(markdown_path), "--github-output", str(github_output), ], cwd=REPO_ROOT, capture_output=True, text=True, check=False, )
self.assertEqual(0, result.returncode, result.stdout + result.stderr) report = json.loads(json_path.read_text(encoding="utf-8")) self.assertEqual(1, report["schema_version"]) self.assertEqual("warning", report["status"]) self.assertEqual(52_429, report["delta"]["size_bytes"]) self.assertEqual(52_429, report["section_deltas_bytes"]["__TEXT.__text"]) self.assertEqual(49_152, report["segment_deltas_bytes"]["__TEXT"]) self.assertEqual(100_000, report["base"]["size_bytes"]) self.assertEqual(152_429, report["head"]["size_bytes"]) self.assertEqual("a" * 40, report["base"]["source_sha"]) self.assertEqual("b" * 40, report["head"]["source_sha"]) markdown = markdown_path.read_text(encoding="utf-8") self.assertIn("+52,429 bytes", markdown) self.assertIn("52,429 bytes (0.050000 MiB)", markdown) self.assertIn("## Segment changes", markdown) self.assertIn("| `__TEXT` | +49,152 |", markdown) self.assertIn("## Largest section changes", markdown) self.assertIn("| `__TEXT.__text` | +52,429 |", markdown) self.assertEqual( "warning=true\ndelta_bytes=52429\nstatus=warning\n", github_output.read_text(encoding="utf-8"), )
def test_linux_report_attributes_elf_section_growth(self) -> None: with tempfile.TemporaryDirectory(prefix="fx-binary-size-") as tmp: root = pathlib.Path(tmp) base_binary = root / "base-fx" head_binary = root / "head-fx" base_binary.write_bytes(b"b" * 100_000) head_binary.write_bytes(b"h" * 100_100) base_sections = root / "base-sections.txt" head_sections = root / "head-sections.txt" base_sections.write_text( f"{base_binary} :\n" "section size addr\n" ".text 70000 16384\n" ".rodata 20000 86016\n" ".data 1000 106496\n" "Total 91000\n", encoding="utf-8", ) head_sections.write_text( f"{head_binary} :\n" "section size addr\n" ".text 70060 16384\n" ".rodata 20040 86016\n" ".data 1000 106496\n" "Total 91100\n", encoding="utf-8", ) json_path = root / "report.json" markdown_path = root / "report.md"
result = subprocess.run( [ sys.executable, "-m", "scripts.binary_size", "--base-binary", str(base_binary), "--head-binary", str(head_binary), "--base-sections", str(base_sections), "--head-sections", str(head_sections), "--base-sha", "a" * 40, "--head-sha", "b" * 40, "--target", "x86_64-linux", "--output-json", str(json_path), "--output-markdown", str(markdown_path), ], cwd=REPO_ROOT, capture_output=True, text=True, check=False, )
self.assertEqual(0, result.returncode, result.stdout + result.stderr) report = json.loads(json_path.read_text(encoding="utf-8")) self.assertEqual({}, report["segment_deltas_bytes"]) self.assertEqual(60, report["section_deltas_bytes"][".text"]) self.assertEqual(40, report["section_deltas_bytes"][".rodata"]) self.assertIn("| `.text` | +60 |", markdown_path.read_text())
def test_section_parser_rejects_duplicate_architecture_output(self) -> None: with tempfile.TemporaryDirectory(prefix="fx-binary-size-") as tmp: report = pathlib.Path(tmp) / "sections.txt" report.write_text( "Segment __TEXT: 65536\n" "\tSection __text: 50000\n" "Segment __TEXT: 73728\n" "\tSection __text: 60000\n", encoding="utf-8", )
with self.assertRaisesRegex(BinarySizeError, "duplicate segment __TEXT"): parse_macho_sections(report)
def test_decrease_is_informational_and_named_explicitly(self) -> None: with tempfile.TemporaryDirectory(prefix="fx-binary-size-") as tmp: root = pathlib.Path(tmp) base_binary = root / "base-fx" head_binary = root / "head-fx" base_binary.write_bytes(b"b" * 100) head_binary.write_bytes(b"h" * 90) base_sections = root / "base-sections.txt" head_sections = root / "head-sections.txt" base_sections.write_text("Segment __TEXT: 100\n", encoding="utf-8") head_sections.write_text("Segment __TEXT: 90\n", encoding="utf-8")
report = build_report( base_binary=base_binary, head_binary=head_binary, base_sections=base_sections, head_sections=head_sections, base_sha="a" * 40, head_sha="b" * 40, target="aarch64-macos", warning_bytes=52_429, )
delta = report["delta"] self.assertIsInstance(delta, dict) assert isinstance(delta, dict) self.assertEqual("ok", report["status"]) self.assertEqual("decrease", delta.get("direction")) self.assertIn( "The PR binary is smaller than the base binary.", markdown_report(report), )
def test_section_parser_requires_executable_text_segment(self) -> None: with tempfile.TemporaryDirectory(prefix="fx-binary-size-") as tmp: report = pathlib.Path(tmp) / "sections.txt" report.write_text("Segment __DATA: 16384\n", encoding="utf-8")
with self.assertRaisesRegex(BinarySizeError, "missing __TEXT segment"): parse_macho_sections(report)
class BinarySizeWorkflowTests(unittest.TestCase): def test_pr_workflow_compares_all_supported_release_safe_targets(self) -> None: self.assertTrue(WORKFLOW_PATH.is_file(), "binary-size workflow is missing") workflow = WORKFLOW_PATH.read_text(encoding="utf-8")
self.assertIn("pull_request:", workflow) self.assertNotIn("pull_request_target", workflow) self.assertIn("contents: read", workflow) self.assertIn("runs-on: ${{ matrix.runner }}", workflow) for name, target, runner in ( ("linux-x86_64", "x86_64-linux", "ubuntu-24.04"), ("linux-aarch64", "aarch64-linux", "ubuntu-24.04-arm"), ("macos-x86_64", "x86_64-macos", "macos-15-intel"), ("macos-aarch64", "aarch64-macos", "macos-15"), ): self.assertIn(f"name: {name}", workflow) self.assertIn(f"target: {target}", workflow) self.assertIn(f"runner: {runner}", workflow) self.assertIn("fetch-depth: 0", workflow) self.assertIn("github.event.pull_request.base.sha", workflow) self.assertIn('test "$(git rev-parse HEAD)" = "$HEAD_SHA"', workflow) self.assertIn( 'test "$(git -C "$base_worktree" rev-parse HEAD)" = "$BASE_SHA"', workflow, ) self.assertIn("-Dtarget=${{ matrix.target }}", workflow) self.assertIn("-Doptimize=ReleaseSafe", workflow) self.assertGreaterEqual(workflow.count("zig build"), 2) self.assertGreaterEqual(workflow.count("size -m"), 2) self.assertGreaterEqual(workflow.count("size -A -d"), 2) self.assertIn("python3 -m scripts.binary_size", workflow) self.assertIn("$GITHUB_STEP_SUMMARY", workflow) self.assertIn("::warning title=Binary size increase::", workflow) self.assertIn("actions/upload-artifact@v4", workflow) self.assertIn("binary-size-evidence-${{ matrix.name }}", workflow) self.assertIn("binary-size-binaries-${{ matrix.name }}", workflow)
if __name__ == "__main__": unittest.main()