Spaces:
Running
Running
File size: 4,826 Bytes
412a75b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 | import importlib.util
import subprocess
import sys
from pathlib import Path
PROJECT_ROOT = Path(__file__).resolve().parents[1]
MODULE_PATH = PROJECT_ROOT / ".github" / "scripts" / "docker_release_plan.py"
def load_module():
spec = importlib.util.spec_from_file_location("docker_release_plan", MODULE_PATH)
module = importlib.util.module_from_spec(spec)
assert spec.loader is not None
sys.modules[spec.name] = module
spec.loader.exec_module(module)
return module
def git(repo: Path, *args: str) -> str:
result = subprocess.run(
["git", *args],
cwd=repo,
check=True,
capture_output=True,
text=True,
)
return result.stdout.strip()
def commit_file(repo: Path, name: str, content: str, message: str) -> str:
(repo / name).write_text(content, encoding="utf-8")
git(repo, "add", name)
git(repo, "commit", "-m", message)
return git(repo, "rev-parse", "HEAD")
def seed_remote_refs(repo: Path, *branches: str) -> None:
for branch in branches:
git(repo, "update-ref", f"refs/remotes/origin/{branch}", git(repo, "rev-parse", branch))
def test_docker_publish_workflow_tracks_branch_promotions():
workflow_path = PROJECT_ROOT / ".github" / "workflows" / "docker-publish.yml"
content = workflow_path.read_text(encoding="utf-8")
assert 'branches:\n - "testing"\n - "main"' in content
assert 'tags:\n - "v*"' in content
assert "workflow_dispatch:" in content
assert "inputs:" in content
assert "tag:" in content
assert 'ref: ${{ matrix.source_tag }}' in content
assert "SOURCE_REF_TYPE: ${{ github.ref_type }}" in content
assert "BEFORE_SHA: ${{ github.event_name == 'push' && github.event.before || '' }}" in content
def test_plan_branch_push_builds_when_tag_reaches_allowed_branch(monkeypatch, tmp_path: Path):
release_plan = load_module()
git(tmp_path, "init", "-b", "main")
git(tmp_path, "config", "user.name", "Test User")
git(tmp_path, "config", "user.email", "test@example.com")
commit_file(tmp_path, "README.md", "base\n", "base")
git(tmp_path, "tag", "v1.6")
git(tmp_path, "branch", "testing")
git(tmp_path, "checkout", "-b", "development")
git(tmp_path, "checkout", "main")
git(tmp_path, "merge", "--ff-only", "development")
git(tmp_path, "checkout", "development")
commit_file(tmp_path, "feature.txt", "release\n", "release v1.7")
git(tmp_path, "tag", "v1.7")
testing_before = git(tmp_path, "rev-parse", "testing")
git(tmp_path, "checkout", "testing")
git(tmp_path, "merge", "--no-ff", "development", "-m", "promote v1.7 to testing")
git(tmp_path, "checkout", "main")
git(tmp_path, "merge", "--no-ff", "development", "-m", "promote v1.7 to main")
seed_remote_refs(tmp_path, "testing", "main")
monkeypatch.chdir(tmp_path)
monkeypatch.setenv("ALLOWED_BRANCHES", "testing main")
monkeypatch.setenv("MAIN_BRANCH", "main")
monkeypatch.setenv("DOCKER_IMAGE_REPO", "example/agent-zero")
monkeypatch.setenv("RELEASE_TAG_REGEX", r"^v([0-9]+)\.([0-9]+)$")
monkeypatch.setenv("MIN_RELEASE_MAJOR", "1")
monkeypatch.setenv("MIN_RELEASE_MINOR", "0")
monkeypatch.setenv("EVENT_NAME", "push")
monkeypatch.setenv("SOURCE_REF_TYPE", "branch")
monkeypatch.setenv("MANUAL_TAG", "")
monkeypatch.setenv("AFTER_SHA", git(tmp_path, "rev-parse", "testing"))
monkeypatch.setenv("SOURCE_REF_NAME", "testing")
monkeypatch.setenv("BEFORE_SHA", testing_before)
config = release_plan.load_config()
branch_states = release_plan.collect_branch_states(config)
testing_candidates, testing_notes = release_plan.plan_branch_push(config, branch_states)
assert testing_notes == []
assert len(testing_candidates) == 1
assert testing_candidates[0].branch == "testing"
assert testing_candidates[0].source_tag == "v1.7"
assert testing_candidates[0].mode == "push_promoted_tag"
assert testing_candidates[0].publish_version is False
assert testing_candidates[0].publish_branch_tag is True
monkeypatch.setenv("SOURCE_REF_NAME", "main")
monkeypatch.setenv("BEFORE_SHA", git(tmp_path, "rev-list", "--max-parents=0", "HEAD"))
monkeypatch.setenv("AFTER_SHA", git(tmp_path, "rev-parse", "main"))
config = release_plan.load_config()
branch_states = release_plan.collect_branch_states(config)
main_candidates, main_notes = release_plan.plan_branch_push(config, branch_states)
assert main_notes == []
assert len(main_candidates) == 1
assert main_candidates[0].branch == "main"
assert main_candidates[0].source_tag == "v1.7"
assert main_candidates[0].mode == "push_promoted_tag"
assert main_candidates[0].publish_version is True
assert main_candidates[0].publish_branch_tag is True
|