tenant.openagents/omega

1from __future__ import annotations
2
3import argparse
4import unittest
5from unittest.mock import patch
6
7from zed_eval import source
8from zed_eval.builds import prepare_build_request, resolve_source
9
10
11class ResolveRemoteRefTests(unittest.TestCase):
12    def test_full_sha_needs_no_network(self) -> None:
13        sha = "a" * 40
14        with patch("zed_eval.source.git_output") as git_output:
15            self.assertEqual(source.resolve_remote_ref("url", sha), sha)
16            git_output.assert_not_called()
17
18    def test_annotated_tag_prefers_peeled_commit(self) -> None:
19        ls_remote = (
20            "1111111111111111111111111111111111111111\trefs/tags/v0.210.0\n"
21            "2222222222222222222222222222222222222222\trefs/tags/v0.210.0^{}"
22        )
23        with patch("zed_eval.source.git_output", return_value=ls_remote):
24            self.assertEqual(source.resolve_remote_ref("url", "v0.210.0"), "2" * 40)
25
26    def test_unresolvable_ref_raises(self) -> None:
27        with patch("zed_eval.source.git_output", return_value=""):
28            with self.assertRaises(ValueError):
29                source.resolve_remote_ref("url", "missing")
30
31
32class BuildIdDedupTests(unittest.TestCase):
33    """A clean tag and the SHA it resolves to must yield the *same* build id, so
34    a build done by one teammate is reused by another."""
35
36    def _build_id_for(self, from_source: str, resolved_sha: str) -> str:
37        args = argparse.Namespace(
38            from_source=from_source,
39            base_sha=None,
40            clean_source=False,
41            zed_version=None,
42            repo_url=source.DEFAULT_REPO_URL,
43        )
44        with patch("zed_eval.source.resolve_remote_ref", return_value=resolved_sha):
45            base_sha, clean, label, pre_resolved = resolve_source(args)
46        request = prepare_build_request(
47            base_sha=base_sha,
48            patch_path=None,
49            build_id=None,
50            allow_untracked=False,
51            require_clean=False,
52            repo_url=source.DEFAULT_REPO_URL,
53            clean_source=clean,
54            source_label=label,
55            pre_resolved_base_sha=pre_resolved,
56        )
57        return request["build_id"]
58
59    def test_tag_and_sha_share_build_id(self) -> None:
60        sha = "c" * 40
61        from_tag = self._build_id_for("v0.210.0", sha)
62        from_sha = self._build_id_for(sha, sha)
63        self.assertEqual(from_tag, from_sha)
64        self.assertTrue(from_tag.startswith("bld-"))
65
66
67if __name__ == "__main__":
68    unittest.main()
69