"""Unit tests for runner_manager module."""

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

import pytest

from gitea_runner_manager.exceptions import AnsibleError
from gitea_runner_manager.runner_manager import RunnerManager


class TestRunnerManager:
    def test_init(self) -> None:
        manager = RunnerManager()
        assert manager is not None

    def test_install_basic(self) -> None:
        manager = RunnerManager()

        with patch("subprocess.run") as mock_run:
            mock_run.return_value = MagicMock(returncode=0)
            manager.install("192.168.1.10", "ubuntu", token="tok", gitea_url="https://git.example.com")
            mock_run.assert_called_once()
            cmd = mock_run.call_args[0][0]
            cmd_str = " ".join(cmd)
            assert "ansible-playbook" in cmd_str
            assert "192.168.1.10," in cmd_str
            assert "-u" in cmd_str
            assert "ubuntu" in cmd_str
            assert "registration_token=tok" in cmd_str
            assert "runner_name=192.168.1.10" in cmd_str
            assert "gitea_url=https://git.example.com" in cmd_str
            assert "runner_mode=docker" in cmd_str

    def test_install_with_name_and_key(self) -> None:
        manager = RunnerManager()

        with patch("subprocess.run") as mock_run:
            mock_run.return_value = MagicMock(returncode=0)
            manager.install(
                "host1", "root", key="/key", name="my-runner", token="preset", gitea_url="https://git.example.com"
            )
            cmd = mock_run.call_args[0][0]
            cmd_str = " ".join(cmd)
            assert "--private-key" in cmd_str
            assert "/key" in cmd_str
            assert "registration_token=preset" in cmd_str
            assert "runner_name=my-runner" in cmd_str
            assert "runner_mode=docker" in cmd_str
            assert "--ask-become-pass" not in cmd_str

    def test_install_ask_become_pass(self) -> None:
        manager = RunnerManager()

        with patch("subprocess.run") as mock_run:
            mock_run.return_value = MagicMock(returncode=0)
            manager.install("host1", "root", token="tok", gitea_url="https://git.example.com", ask_become_pass=True)
            cmd = mock_run.call_args[0][0]
            cmd_str = " ".join(cmd)
            assert "--ask-become-pass" in cmd_str
            assert "runner_mode=docker" in cmd_str

    def test_install_binary_mode(self) -> None:
        manager = RunnerManager()

        with patch("subprocess.run") as mock_run:
            mock_run.return_value = MagicMock(returncode=0)
            manager.install("host1", "root", token="tok", gitea_url="https://git.example.com", mode="binary")
            cmd = mock_run.call_args[0][0]
            cmd_str = " ".join(cmd)
            assert "runner_mode=binary" in cmd_str

    def test_install_missing_token(self) -> None:
        manager = RunnerManager()
        with pytest.raises(AnsibleError, match="GITEA_REGISTRATION_TOKEN must be set"):
            manager.install("host", "user", gitea_url="https://git.example.com")

    def test_install_playbook_not_found(self) -> None:
        manager = RunnerManager()
        with patch.object(Path, "exists", return_value=False):
            with pytest.raises(AnsibleError, match="Playbook not found"):
                manager.install("host", "user", token="tok")

    def test_install_ansible_failure(self) -> None:
        manager = RunnerManager()
        with patch("subprocess.run") as mock_run:
            mock_run.return_value = MagicMock(returncode=1)
            with pytest.raises(AnsibleError, match="Ansible failed with exit code 1"):
                manager.install("host", "user", token="tok")

    def test_update(self) -> None:
        manager = RunnerManager()
        with patch("subprocess.run") as mock_run:
            mock_run.return_value = MagicMock(returncode=0)
            manager.update("host", "user", key="/key", version="v0.2.0")
            cmd = mock_run.call_args[0][0]
            cmd_str = " ".join(cmd)
            assert "update-runner.yml" in cmd_str
            assert "--private-key" in cmd_str
            assert "/key" in cmd_str
            assert "gitea_runner_version=v0.2.0" in cmd_str
            assert "runner_mode=docker" in cmd_str
            assert "--ask-become-pass" not in cmd_str

    def test_update_ask_become_pass(self) -> None:
        manager = RunnerManager()
        with patch("subprocess.run") as mock_run:
            mock_run.return_value = MagicMock(returncode=0)
            manager.update("host", "user", ask_become_pass=True)
            cmd = mock_run.call_args[0][0]
            cmd_str = " ".join(cmd)
            assert "--ask-become-pass" in cmd_str
            assert "runner_mode=docker" in cmd_str

    def test_update_binary_mode(self) -> None:
        manager = RunnerManager()
        with patch("subprocess.run") as mock_run:
            mock_run.return_value = MagicMock(returncode=0)
            manager.update("host", "user", mode="binary")
            cmd = mock_run.call_args[0][0]
            cmd_str = " ".join(cmd)
            assert "runner_mode=binary" in cmd_str

    def test_update_playbook_not_found(self) -> None:
        manager = RunnerManager()
        with patch.object(Path, "exists", return_value=False):
            with pytest.raises(AnsibleError, match="Playbook not found"):
                manager.update("host", "user")

    def test_update_ansible_failure(self) -> None:
        manager = RunnerManager()
        with patch("subprocess.run") as mock_run:
            mock_run.return_value = MagicMock(returncode=2)
            with pytest.raises(AnsibleError, match="Ansible failed with exit code 2"):
                manager.update("host", "user")