"""Tests for systemd service and timer configuration"""

import pytest

testinfra_hosts = [
    "borg-client",
    "borg-client-2",
    "borg-client-multi",
    "borg-client-nonroot",
    "borg-client-multi-keys",
]

CLIENT_CONFIGS = {
    "borg-client": {
        "user": "root",
        "server": "borg-server",
        "schedule": "*-*-* 02:00:00",
        "success_exit_status": False,
    },
    "borg-client-2": {
        "user": "root",
        "server": "borg-server-2",
        "schedule": "*-*-* 03:00:00",
        "success_exit_status": True,
    },
    "borg-client-nonroot": {
        "user": "backupuser",
        "server": "borg-server",
        "schedule": "*-*-* 02:00:00",
        "success_exit_status": False,
    },
}

MULTI_INSTANCE_CONFIGS = {
    "borg-client-multi": {
        "user": "root",
        "repos": ["configs", "home-data"],
        "schedules": {
            "configs": "*-*-* 02:00:00",
            "home-data": "*-*-* 04:00:00",
        },
    },
    "borg-client-multi-keys": {
        "user": "root",
        "repos": ["configs-keys", "home-data-keys"],
        "schedules": {
            "configs-keys": "*-*-* 02:00:00",
            "home-data-keys": "*-*-* 04:00:00",
        },
    },
}


@pytest.fixture
def config(host):
    hostname = host.backend.hostname
    if hostname in CLIENT_CONFIGS:
        return {"type": "single", **CLIENT_CONFIGS[hostname]}
    return {"type": "multi", **MULTI_INSTANCE_CONFIGS[hostname]}


class TestSystemdServiceFile:
    def test_service_file_exists(self, host, config):
        if config["type"] == "single":
            service = host.file(
                f"/etc/systemd/system/borg_backup@{config['server']}.service"
            )
            assert service.exists
            assert service.user == "root"
            assert service.group == "root"
            assert service.mode == 0o644
        else:
            for repo in config["repos"]:
                service = host.file(f"/etc/systemd/system/borg_backup@{repo}.service")
                assert service.exists
                assert service.user == "root"
                assert service.group == "root"

    def test_service_content(self, host, config):
        if config["type"] == "single":
            service = host.file(
                f"/etc/systemd/system/borg_backup@{config['server']}.service"
            )
            assert service.contains("[Unit]")
            assert service.contains("[Service]")
            assert service.contains("[Install]")
            assert service.contains("Type=oneshot")
            assert service.contains("ExecStart=/usr/local/bin/run_borg_backup")
        else:
            for repo in config["repos"]:
                service = host.file(f"/etc/systemd/system/borg_backup@{repo}.service")
                assert service.contains("[Unit]")
                assert service.contains("[Service]")
                assert service.contains("[Install]")
                assert service.contains("Type=oneshot")

    def test_service_user(self, host, config):
        if config["type"] == "single":
            service = host.file(
                f"/etc/systemd/system/borg_backup@{config['server']}.service"
            )
            assert service.contains(f"User={config['user']}")
            assert service.contains(f"Group={config['user']}")
        else:
            for repo in config["repos"]:
                service = host.file(f"/etc/systemd/system/borg_backup@{repo}.service")
                assert service.contains(f"User={config['user']}")
                assert service.contains(f"Group={config['user']}")

    def test_success_exit_status(self, host, config):
        if config["type"] != "single":
            return
        service = host.file(
            f"/etc/systemd/system/borg_backup@{config['server']}.service"
        )
        assert service.exists

        if config["success_exit_status"]:
            assert service.contains("SuccessExitStatus=1 TEMPFAIL")
        else:
            assert not service.contains("SuccessExitStatus=")


class TestSystemdTimerFile:
    def test_timer_file_exists(self, host, config):
        if config["type"] == "single":
            timer = host.file(
                f"/etc/systemd/system/borg_backup@{config['server']}.timer"
            )
            assert timer.exists
            assert timer.user == "root"
            assert timer.group == "root"
            assert timer.mode == 0o644
        else:
            for repo in config["repos"]:
                timer = host.file(f"/etc/systemd/system/borg_backup@{repo}.timer")
                assert timer.exists

    def test_timer_content(self, host, config):
        if config["type"] == "single":
            timer = host.file(
                f"/etc/systemd/system/borg_backup@{config['server']}.timer"
            )
            assert timer.contains("[Unit]")
            assert timer.contains("[Timer]")
            assert timer.contains("[Install]")
            assert timer.contains("OnCalendar=")
            assert timer.contains("AccuracySec=")
        else:
            for repo in config["repos"]:
                timer = host.file(f"/etc/systemd/system/borg_backup@{repo}.timer")
                assert timer.contains("[Unit]")
                assert timer.contains("[Timer]")
                assert timer.contains("[Install]")
                assert timer.contains("OnCalendar=")
                assert timer.contains("AccuracySec=")

    def test_timer_schedule(self, host, config):
        if config["type"] == "single":
            timer = host.file(
                f"/etc/systemd/system/borg_backup@{config['server']}.timer"
            )
            assert f"OnCalendar={config['schedule']}" in timer.content_string
        else:
            for repo in config["repos"]:
                timer = host.file(f"/etc/systemd/system/borg_backup@{repo}.timer")
                assert f"OnCalendar={config['schedules'][repo]}" in timer.content_string


class TestSystemdState:
    def test_timer_enabled(self, host, config):
        if config["type"] == "single":
            timer_name = f"borg_backup@{config['server']}.timer"
            c = host.run(f"systemctl is-enabled {timer_name}")
            assert c.rc == 0
            assert c.stdout.strip() == "enabled"
        else:
            for repo in config["repos"]:
                c = host.run(f"systemctl is-enabled borg_backup@{repo}.timer")
                assert c.rc == 0
                assert c.stdout.strip() == "enabled"

    def test_timer_active(self, host, config):
        if config["type"] == "single":
            timer_name = f"borg_backup@{config['server']}.timer"
            c = host.run(f"systemctl is-active {timer_name}")
            assert c.rc == 0
        else:
            for repo in config["repos"]:
                c = host.run(f"systemctl is-active borg_backup@{repo}.timer")
                assert c.rc == 0

    def test_daemon_reload_ok(self, host, config):
        c = host.run("systemctl daemon-reload")
        assert c.rc == 0