grm/tests/unit/test_executor.py

"""Unit tests for AnsibleExecutor."""

import os
from pathlib import Path
from unittest.mock import MagicMock, mock_open, patch

import pytest

from gitea_runner_manager.exceptions import AnsibleError
from gitea_runner_manager.executor import AnsibleExecutor


def _mock_popen_process(returncode: int = 0) -> MagicMock:
    """Create a mock Popen process that yields no stdout and exits with given code."""
    proc = MagicMock()
    proc.stdout = iter([])
    proc.wait.return_value = returncode
    proc.returncode = returncode
    return proc


class TestAnsibleExecutorRun:
    def test_run_success(self, tmp_path: Path, capsys: pytest.CaptureFixture[str]) -> None:
        executor = AnsibleExecutor(log_dir=tmp_path)
        with patch("subprocess.Popen") as mock_popen, patch("builtins.open", mock_open()):
            mock_popen.return_value = _mock_popen_process(returncode=0)
            executor.run(["echo", "hello"], description="Test run")

        captured = capsys.readouterr()
        assert "[GRM] Test run" in captured.out
        assert "[GRM] Done" in captured.out

    def test_run_failure(self, tmp_path: Path) -> None:
        executor = AnsibleExecutor(log_dir=tmp_path)
        with patch("subprocess.Popen") as mock_popen, patch("builtins.open", mock_open()):
            mock_popen.return_value = _mock_popen_process(returncode=1)
            with pytest.raises(AnsibleError, match="exit code 1"):
                executor.run(["false"], description="Test run")

    def test_run_default_description(self, tmp_path: Path, capsys: pytest.CaptureFixture[str]) -> None:
        executor = AnsibleExecutor(log_dir=tmp_path)
        with patch("subprocess.Popen") as mock_popen, patch("builtins.open", mock_open()):
            mock_popen.return_value = _mock_popen_process(returncode=0)
            executor.run(["echo", "hello"])

        captured = capsys.readouterr()
        assert "[GRM] Running Ansible playbook" in captured.out

    def test_run_with_stdout_lines(self, tmp_path: Path) -> None:
        executor = AnsibleExecutor(log_dir=tmp_path)
        proc = MagicMock()
        proc.stdout = iter(["line1\n", "line2\n"])
        proc.wait.return_value = 0
        proc.returncode = 0
        mock_file = mock_open()

        with patch("subprocess.Popen", return_value=proc), patch("builtins.open", mock_file):
            executor.run(["echo", "hello"], description="Test run")
            handle = mock_file()
            handle.write.assert_any_call("line1\n")
            handle.write.assert_any_call("line2\n")

    def test_run_extracts_status(self, tmp_path: Path, capsys: pytest.CaptureFixture[str]) -> None:
        executor = AnsibleExecutor(log_dir=tmp_path)
        log_content = (
            "TASK [gitea-runner : Report runner status]\n"
            "ok: [127.0.0.1] => {\n"
            '    "msg": "Runner \'127.0.0.1\' is installed and running."\n'
            "}\n"
        )
        with patch("subprocess.Popen") as mock_popen, patch("builtins.open", mock_open(read_data=log_content)):
            mock_popen.return_value = _mock_popen_process(returncode=0)
            executor.run(["echo", "hello"], description="Test run")

        captured = capsys.readouterr()
        assert "Runner '127.0.0.1' is installed and running" in captured.out


class TestAnsibleExecutorExtractStatus:
    def test_extract_status_found(self, tmp_path: Path) -> None:
        executor = AnsibleExecutor()
        log_content = (
            "TASK [gitea-runner : Report runner status]\n"
            "ok: [127.0.0.1] => {\n"
            '    "msg": "Runner \'127.0.0.1\' is installed and running."\n'
            "}\n"
        )
        log_file = tmp_path / "test.log"
        log_file.write_text(log_content)
        status = executor._extract_status(log_file)
        assert status is not None
        assert "Runner '127.0.0.1' is installed and running" in status

    def test_extract_status_not_found(self, tmp_path: Path) -> None:
        executor = AnsibleExecutor()
        log_file = tmp_path / "test.log"
        log_file.write_text('TASK [other]\nok: [host] => {\n    "msg": "something else"\n}\n')
        status = executor._extract_status(log_file)
        assert status is None

    def test_extract_status_exception(self, tmp_path: Path) -> None:
        executor = AnsibleExecutor()
        log_file = tmp_path / "test.log"
        log_file.write_text("incomplete")
        log_file.chmod(0o000)
        try:
            status = executor._extract_status(log_file)
            assert status is None
        finally:
            log_file.chmod(0o644)


class TestAnsibleExecutorPrepareLog:
    def test_prepare_log(self, tmp_path: Path) -> None:
        executor = AnsibleExecutor(log_dir=tmp_path)
        log_file = executor._prepare_log(["ansible-playbook", "test.yml"])
        assert log_file.parent == tmp_path
        assert log_file.name.startswith("ansible-")
        content = log_file.read_text()
        assert "=== GRM Ansible Run:" in content
        assert "Command: ansible-playbook test.yml" in content

    def test_prepare_log_creates_directories(self, tmp_path: Path) -> None:
        nested = tmp_path / "a" / "b" / "c"
        executor = AnsibleExecutor(log_dir=nested)
        log_file = executor._prepare_log(["echo"])
        assert nested.exists()
        assert log_file.exists()


class TestAnsibleExecutorTranslation:
    def test_run_translated(self, tmp_path: Path, capsys: pytest.CaptureFixture[str]) -> None:
        executor = AnsibleExecutor(log_dir=tmp_path)
        with (
            patch.dict(os.environ, {"GRM_LANG": "de"}),
            patch("subprocess.Popen") as mock_popen,
            patch("builtins.open", mock_open()),
        ):
            mock_popen.return_value = _mock_popen_process(returncode=0)
            executor.run(["echo", "hello"])  # use default description to test translation

        captured = capsys.readouterr()
        assert "Ansible-Playbook wird ausgeführt" in captured.out
        assert "Fertig" in captured.out

    def test_unsupported_lang_fallback(self, tmp_path: Path, capsys: pytest.CaptureFixture[str]) -> None:
        executor = AnsibleExecutor(log_dir=tmp_path)
        with (
            patch.dict(os.environ, {"GRM_LANG": "xx"}),
            patch("subprocess.Popen") as mock_popen,
            patch("builtins.open", mock_open()),
        ):
            mock_popen.return_value = _mock_popen_process(returncode=0)
            executor.run(["echo", "hello"])

        captured = capsys.readouterr()
        assert "[GRM] Running Ansible playbook" in captured.out  # English fallback