DEVX-181: ci: retrigger validation

tests/unit/test_molecule_lease.py

@@ -0,0 +1,121 @@
+"""Unit tests for devx.molecule.lease."""
+
+import os
+from unittest.mock import MagicMock, mock_open, patch
+
+from click.testing import CliRunner
+
+from devx.molecule.lease import (
+    DEFAULT_TTL_SECONDS,
+    LEASE_UNTIL_ENV,
+    OWNER_ENV,
+    emit_github_env,
+    lease_env,
+    lease_owner,
+    main,
+)
+
+
+class TestLeaseOwner:
+    def test_gitea_env_preferred(self) -> None:
+        with patch.dict(
+            os.environ,
+            {"GITEA_RUN_ID": "123", "GITEA_JOB": "tests", "GITHUB_RUN_ID": "999"},
+        ):
+            assert lease_owner() == "run-123-tests"
+
+    def test_github_env_fallback(self) -> None:
+        env = {"GITHUB_RUN_ID": "777", "GITHUB_JOB": "molecule"}
+        with patch.dict(os.environ, env, clear=True):
+            assert lease_owner() == "run-777-molecule"
+
+    def test_job_defaults_when_missing(self) -> None:
+        with patch.dict(os.environ, {"GITHUB_RUN_ID": "5"}, clear=True):
+            assert lease_owner() == "run-5-job"
+
+    def test_local_fallback(self) -> None:
+        with (
+            patch.dict(os.environ, {}, clear=True),
+            patch("devx.molecule.lease.socket.gethostname", return_value="devbox"),
+        ):
+            assert lease_owner() == "local-devbox"
+
+
+class TestLeaseEnv:
+    def test_arithmetic(self) -> None:
+        env = lease_env(ttl=3600, now=1000)
+        assert env[LEASE_UNTIL_ENV] == str(1000 + 3600)
+        assert OWNER_ENV in env
+
+    def test_default_now_uses_time(self) -> None:
+        with patch("devx.molecule.lease.time.time", return_value=2000.9):
+            env = lease_env(ttl=60)
+        assert env[LEASE_UNTIL_ENV] == str(2000 + 60)
+
+    def test_default_ttl(self) -> None:
+        env = lease_env(now=0)
+        assert env[LEASE_UNTIL_ENV] == str(DEFAULT_TTL_SECONDS)
+
+
+class TestEmitGithubEnv:
+    def test_writes_lines(self) -> None:
+        m = mock_open()
+        with (
+            patch.dict(os.environ, {"GITHUB_ENV": "/tmp/env"}, clear=False),
+            patch("builtins.open", m),
+        ):
+            written = emit_github_env({"A": "1", "B": "2"})
+        assert written == 2
+        m.assert_called_once_with("/tmp/env", "a", encoding="utf-8")
+        m().write.assert_any_call("A=1\n")
+        m().write.assert_any_call("B=2\n")
+
+    def test_noop_without_env(self) -> None:
+        with patch.dict(os.environ, {}, clear=True):
+            assert emit_github_env({"A": "1"}) == 0
+
+
+class TestCli:
+    def test_prints_and_exports(self) -> None:
+        runner = CliRunner()
+        m = mock_open()
+        with (
+            patch.dict(os.environ, {"GITHUB_ENV": "/tmp/env", "GITHUB_RUN_ID": "42"}, clear=True),
+            patch("builtins.open", m),
+            patch("devx.molecule.lease.time.time", return_value=100.0),
+        ):
+            result = runner.invoke(main, ["--ttl", "600"])
+        assert result.exit_code == 0
+        assert f"{LEASE_UNTIL_ENV}=700" in result.output
+        assert f"{OWNER_ENV}=run-42-job" in result.output
+        assert "Exported 2 lease vars" in result.output
+
+    def test_prints_without_github_env(self) -> None:
+        runner = CliRunner()
+        with (
+            patch.dict(os.environ, {"GITHUB_RUN_ID": "9", "GITHUB_JOB": "tests"}, clear=True),
+            patch("devx.molecule.lease.time.time", return_value=50.0),
+        ):
+            result = runner.invoke(main)
+        assert result.exit_code == 0
+        assert f"{LEASE_UNTIL_ENV}={50 + DEFAULT_TTL_SECONDS}" in result.output
+        assert f"{OWNER_ENV}=run-9-tests" in result.output
+        assert "GITHUB_ENV" not in result.output
+
+
+class TestStartDockerIntegration:
+    """start_docker.main() must emit lease env on success (REQ-2)."""
+
+    @patch("devx.molecule.start_docker.start_docker_daemon", return_value=True)
+    def test_emits_lease_on_success(self, _mock: MagicMock) -> None:
+        from devx.molecule.start_docker import main as start_docker_main
+
+        runner = CliRunner()
+        m = mock_open()
+        with (
+            patch.dict(os.environ, {"GITHUB_ENV": "/tmp/env", "GITHUB_RUN_ID": "7"}, clear=True),
+            patch("builtins.open", m),
+        ):
+            result = runner.invoke(start_docker_main, ["--timeout", "1"])
+        assert result.exit_code == 0
+        assert "Exported 2 lease vars" in result.output