"""Tests for systemd success exit status functionality"""
import pytest

testinfra_hosts = ['borg-client', 'borg-client-success-exit-status']


def test_systemd_service_has_correct_success_exit_status(host):
    service = host.file('/etc/systemd/system/borg_backup@borg-server.service')
    assert service.exists

    if host.backend.get_hostname() == 'borg-client-success-exit-status':
        assert service.contains('SuccessExitStatus=1 TEMPFAIL')
    else:
        assert not service.contains('SuccessExitStatus=')


def test_systemd_timer_is_enabled(host):
    timer_name = 'borg_backup@borg-server.timer'
    c = host.run(f"systemctl is-enabled {timer_name}")
    assert c.rc == 0


def test_systemd_timer_is_active(host):
    timer_name = 'borg_backup@borg-server.timer'
    c = host.run(f"systemctl is-active {timer_name}")
    assert c.rc == 0


@pytest.mark.parametrize('exit_status', [1, 'TEMPFAIL'])
def test_success_exit_status_values(host, exit_status):
    if host.backend.get_hostname() == 'borg-client-success-exit-status':
        service = host.file('/etc/systemd/system/borg_backup@borg-server.service')
        assert service.contains(str(exit_status))