import os

import testinfra.utils.ansible_runner


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


def test_ssh_connectivity(host):
    """
    Test if SSH server is reachable.
    """

    with host.sudo():

        ping = host.run('hping3 -S -p 22 -c 1 %s', 'parameters-mandatory')
        assert ping.rc == 0

        ping = host.run('hping3 -S -p 22 -c 1 %s', 'parameters-optional')
        assert ping.rc == 0


def test_http_connectivity(host):
    """
    Test if HTTP server is reachable.
    """

    with host.sudo():

        ping = host.run('hping3 -S -p 80 -c 1 %s', 'parameters-mandatory')
        assert ping.rc == 0

        ping = host.run('hping3 -S -p 80 -c 1 %s', 'parameters-optional')
        assert ping.rc == 0