mirror of
https://github.com/home-assistant/core.git
synced 2026-05-24 01:10:24 +01:00
Robert Resch
b724e52408
Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>
91 lines
2.6 KiB
Python
91 lines
2.6 KiB
Python
"""Tests for script.check_requirements.__main__ (CLI entry point)."""
|
|
|
|
import json
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from script.check_requirements import __main__ as main_mod
|
|
from script.check_requirements.pypi import ProvenanceResult, PypiPackageInfo
|
|
|
|
|
|
def test_main_writes_artifact(
|
|
tmp_path: Path,
|
|
monkeypatch: pytest.MonkeyPatch,
|
|
capsys: pytest.CaptureFixture[str],
|
|
) -> None:
|
|
"""The CLI parses args, runs checks, and writes a JSON artifact."""
|
|
diff_file = tmp_path / "diff.patch"
|
|
diff_file.write_text(
|
|
"diff --git a/requirements_all.txt b/requirements_all.txt\n"
|
|
"--- a/requirements_all.txt\n"
|
|
"+++ b/requirements_all.txt\n"
|
|
"@@ -1 +1 @@\n"
|
|
"-pkg==1.0.0\n"
|
|
"+pkg==1.1.0\n",
|
|
encoding="utf-8",
|
|
)
|
|
output_file = tmp_path / "results.json"
|
|
|
|
monkeypatch.setattr(
|
|
"script.check_requirements.runner.fetch_package_info",
|
|
lambda name, version: PypiPackageInfo(
|
|
project_urls={"Source": "https://github.com/example/pkg"},
|
|
repo_url="https://github.com/example/pkg",
|
|
file_provenance_urls=["whatever"],
|
|
found=True,
|
|
),
|
|
)
|
|
monkeypatch.setattr(
|
|
"script.check_requirements.runner.check_provenance",
|
|
lambda info: ProvenanceResult(
|
|
has_attestation=True,
|
|
publisher_kind="GitHub",
|
|
recognized_publisher=True,
|
|
detail="ok",
|
|
),
|
|
)
|
|
|
|
exit_code = main_mod.main(
|
|
[
|
|
"--pr-number",
|
|
"42",
|
|
"--diff",
|
|
str(diff_file),
|
|
"--output",
|
|
str(output_file),
|
|
]
|
|
)
|
|
assert exit_code == 0
|
|
|
|
payload = json.loads(output_file.read_text(encoding="utf-8"))
|
|
assert payload["pr_number"] == 42
|
|
assert payload["packages"][0]["name"] == "pkg"
|
|
|
|
captured = capsys.readouterr()
|
|
assert "check_requirements: 1 package change(s)" in captured.err
|
|
|
|
|
|
def test_main_missing_diff_file_exits(
|
|
tmp_path: Path,
|
|
capsys: pytest.CaptureFixture[str],
|
|
) -> None:
|
|
"""A missing --diff file triggers argparse's error path (SystemExit)."""
|
|
output_file = tmp_path / "results.json"
|
|
missing_diff = tmp_path / "does-not-exist.patch"
|
|
|
|
with pytest.raises(SystemExit) as excinfo:
|
|
main_mod.main(
|
|
[
|
|
"--pr-number",
|
|
"1",
|
|
"--diff",
|
|
str(missing_diff),
|
|
"--output",
|
|
str(output_file),
|
|
]
|
|
)
|
|
assert excinfo.value.code == 2 # argparse error exit
|
|
captured = capsys.readouterr()
|
|
assert "not found" in captured.err
|