109 lines
4.5 KiB
Python
109 lines
4.5 KiB
Python
"""Unit tests for runner_manager module."""
|
|
|
|
from pathlib import Path
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
from gitea_runner_manager.api_client import GiteaAPIClient
|
|
from gitea_runner_manager.exceptions import AnsibleError
|
|
from gitea_runner_manager.runner_manager import RunnerManager
|
|
|
|
|
|
class TestRunnerManager:
|
|
def test_list_runners(self) -> None:
|
|
api = MagicMock(spec=GiteaAPIClient)
|
|
api.get_runners.return_value = [{"id": 1, "name": "r1"}]
|
|
manager = RunnerManager(api)
|
|
runners = manager.list_runners()
|
|
assert runners == [{"id": 1, "name": "r1"}]
|
|
|
|
def test_generate_token(self) -> None:
|
|
api = MagicMock(spec=GiteaAPIClient)
|
|
api.create_registration_token.return_value = "token123"
|
|
manager = RunnerManager(api)
|
|
token = manager.generate_token()
|
|
assert token == "token123"
|
|
|
|
def test_install_without_name_or_token(self) -> None:
|
|
api = MagicMock(spec=GiteaAPIClient)
|
|
api.base_url = "https://git.example.com"
|
|
api.create_registration_token.return_value = "auto-token"
|
|
manager = RunnerManager(api)
|
|
|
|
with patch("subprocess.run") as mock_run:
|
|
mock_run.return_value = MagicMock(returncode=0)
|
|
manager.install("192.168.1.10", "ubuntu")
|
|
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=auto-token" in cmd_str
|
|
assert "runner_name=192.168.1.10" in cmd_str
|
|
assert "gitea_url=https://git.example.com" in cmd_str
|
|
|
|
def test_install_with_name_and_token(self) -> None:
|
|
api = MagicMock(spec=GiteaAPIClient)
|
|
api.base_url = "https://git.example.com"
|
|
manager = RunnerManager(api)
|
|
|
|
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")
|
|
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
|
|
api.create_registration_token.assert_not_called()
|
|
|
|
def test_install_playbook_not_found(self) -> None:
|
|
api = MagicMock(spec=GiteaAPIClient)
|
|
api.base_url = "https://git.example.com"
|
|
manager = RunnerManager(api)
|
|
with patch.object(Path, "exists", return_value=False):
|
|
with pytest.raises(AnsibleError, match="Playbook not found"):
|
|
manager.install("host", "user")
|
|
|
|
def test_install_ansible_failure(self) -> None:
|
|
api = MagicMock(spec=GiteaAPIClient)
|
|
api.base_url = "https://git.example.com"
|
|
api.create_registration_token.return_value = "tok"
|
|
manager = RunnerManager(api)
|
|
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")
|
|
|
|
def test_update(self) -> None:
|
|
api = MagicMock(spec=GiteaAPIClient)
|
|
manager = RunnerManager(api)
|
|
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 "act_runner_version=v0.2.0" in cmd_str
|
|
|
|
def test_update_playbook_not_found(self) -> None:
|
|
api = MagicMock(spec=GiteaAPIClient)
|
|
manager = RunnerManager(api)
|
|
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:
|
|
api = MagicMock(spec=GiteaAPIClient)
|
|
manager = RunnerManager(api)
|
|
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")
|