import os
import pytest
import testinfra.utils.ansible_runner


testinfra_hosts = testinfra.utils.ansible_runner.AnsibleRunner(
    os.environ['MOLECULE_INVENTORY_FILE']).get_hosts('all')


@pytest.mark.parametrize("dir", [
    "/opt/promtail",
    "/etc/promtail",
    "/etc/promtail/file_sd",
    "/var/lib/promtail",
])
def test_directories(host, dir):
    d = host.file(dir)
    assert d.is_directory
    assert d.exists


@pytest.mark.parametrize("files", [
    "/etc/systemd/system/promtail.service",
    "/usr/local/bin/promtail",
    "/etc/promtail/promtail.yml"
])
def test_files(host, files):
    f = host.file(files)
    assert f.exists
    assert f.is_file


def test_user(host):
    assert host.group("promtail").exists
    assert host.user("promtail").exists


def test_service(host):
    s = host.service("promtail")
    assert s.is_running


def test_http_socket(host):
    s = host.socket("tcp://0.0.0.0:9080")
    assert s.is_listening


def test_grpc_socket(host):
    s = host.socket("tcp://0.0.0.0:9095")
    assert s.is_listening


def test_version(host):
    result = host.run("/usr/local/bin/promtail --version")
    assert result.rc == 0