testinfra_hosts = ["prometheus"]

"""
Test functionality of defining exporters and default fallbacks
"""


def test_check_hosts_added_simple(host):
    t1 = host.file("/opt/simple_target1.yml")
    t2 = host.file("/opt/simple_target2.yml")
    t3 = host.file("/opt/simple_target3.yml")
    t4 = host.file("/opt/simple_target4.yml")

    assert t1.exists
    assert t2.exists
    assert t3.exists
    assert t4.exists

    assert t1.content_string == "    - application\n"

    assert t2.content_string == "    - test1\n    - test2\n"

    assert t3.content_string == "    - application_AA\n"

    assert t4.content_string == "    - exporter_without_id\n"


"""
Test prefix functionality
"""


def test_check_hosts_added_prefix(host):
    t1 = host.file("/opt/prefix_target1.yml")
    t2 = host.file("/opt/prefix_target2.yml")
    t3 = host.file("/opt/prefix/prefix_target3.yml")

    assert t1.exists
    assert t2.exists
    assert t3.exists

    assert t1.content_string == "    - application\n"

    assert t2.content_string == "    - application\n"

    assert t3.content_string == "    - application\n"


"""
Test hook functionality
"""


def test_check_hosts_added_hooks(host):
    t1 = host.file("/opt/hook_target.yml")
    t2 = host.file("/opt/hook1")
    t3 = host.file("/opt/hook2")

    assert t1.exists
    assert t2.exists
    assert t3.exists

    assert (
        t1.content_string == "    - application\n    - application2\n"
        or t1.content_string == "    - application2\n    - application\n"
    )

    assert t3.content_string == "hello\nhello\n"


"""
Test lineinfile strategy parameters
"""


def test_check_host_added_lineinfile(host):
    t1 = host.file("/opt/lineinfile.yml")

    assert t1.user == "prometheus"
    assert t1.group == "prometheus"
    assert t1.mode == 0o600

    assert (
        t1.content_string == "- labels:\n"
        "    my: label\n"
        "  targets:\n"
        "  - existing:9100\n"
        "  - application:9100\n"
    )