import os

import pytest

import testinfra.utils.ansible_runner


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


@pytest.mark.parametrize("platform", [
    "stretch64",
    "buster64"
])
def test_ssh_connectivity(host, platform):
    """
    Test if SSH server is reachable.
    """

    with host.sudo():

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

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


@pytest.mark.parametrize("platform", [
    "stretch64",
    "buster64"
])
def test_http_connectivity(host, platform):
    """
    Test if HTTP server is reachable.
    """

    with host.sudo():

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

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