"""Role testing files using testinfra."""


def test_hosts_file(host):
    """Validate /etc/hosts file."""
    etc_hosts = host.file("/etc/hosts")
    assert etc_hosts.exists
    assert etc_hosts.user == "root"
    assert etc_hosts.group == "root"

def test_source_list_default(host):
    """Validate /etc/apt/sources.list file."""
    etc_apt_sources_list_default = host.file("/etc/apt/sources.list")
    dist_os = host.system_info.distribution
    dist_codename = host.system_info.codename
    assert etc_apt_sources_list_default.exists
    assert etc_apt_sources_list_default.user == "root"
    assert etc_apt_sources_list_default.group == "root"
    assert etc_apt_sources_list_default.mode == 0o644
    if dist_os == "debian":
        assert etc_apt_sources_list_default.contains("deb http://deb.debian.org/debian " + dist_codename + " main contrib")
        assert etc_apt_sources_list_default.contains("deb http://deb.debian.org/debian " + dist_codename + "-updates main contrib")
        assert etc_apt_sources_list_default.contains("deb http://deb.debian.org/debian-security " + dist_codename + "-security main contrib")
        assert etc_apt_sources_list_default.contains("deb http://deb.debian.org/debian " + dist_codename + "-backports main")
    elif dist_os == "ubuntu":
        assert etc_apt_sources_list_default.contains("deb http://fr.archive.ubuntu.com/ubuntu " + dist_codename + " main restricted universe multiverse")
        assert etc_apt_sources_list_default.contains("deb http://fr.archive.ubuntu.com/ubuntu " + dist_codename + "-updates main restricted universe multiverse")
        assert etc_apt_sources_list_default.contains("deb http://fr.archive.ubuntu.com/ubuntu " + dist_codename + "-security main restricted universe multiverse")
        assert etc_apt_sources_list_default.contains("deb http://fr.archive.ubuntu.com/ubuntu " + dist_codename + "-backports main restricted universe multiverse")

def test_source_list_custom(host):
    """Validate /etc/apt/sources.list.d/custom.list file."""
    etc_apt_sources_list_custom = host.file("/etc/apt/sources.list.d").listdir()
    dist_os = host.system_info.distribution
    dist_codename = host.system_info.codename
    for file in etc_apt_sources_list_custom:
        list_file = host.file("/etc/apt/sources.list.d/" + file)
        if list_file.is_file:
            assert list_file.exists
            assert list_file.user == "root"
            assert list_file.group == "root"
            assert list_file.mode == 0o644
            if file == "docker.list":
                assert list_file.contains(r'deb \[signed-by=/usr/share/keyrings/docker-archive-keyring.asc\] https://download.docker.com/linux/' + dist_os + ' ' + dist_codename + ' stable')
            elif file == "hashicorp.list":
                assert list_file.contains(r'deb \[signed-by=/usr/share/keyrings/hashicorp-archive-keyring.asc\] https://apt.releases.hashicorp.com ' + dist_codename + ' main')