import testinfra.utils.ansible_runner


testinfra_hosts = testinfra.utils.ansible_runner.AnsibleRunner(
    '.molecule/ansible_inventory').get_hosts('mail-server')


def test_connectivity_from_relay(Command, Sudo):
    """
    Tests connectivity towards mail forwarder servers from relay. Connection
    towards parameters-mandatory should fail.
    """

    with Sudo():

        ping = Command('hping3 -S -p 25 -c 1 parameters-mandatory')
        assert ping.rc != 0
        assert "100% packet loss" in ping.stderr

        ping = Command('hping3 -S -p 25 -c 1 parameters-optional')
        assert ping.rc == 0

        ping = Command('hping3 -S -p 25 -c 1 parameters-no-incoming')
        assert "100% packet loss" in ping.stderr
        assert ping.rc != 0


def test_mail_reception_from_relay(Command, Sudo):
    """
    Tests if mails can be sent from relay to servers configured to use the
    relay.
    """

    send = Command('swaks --suppress-data --to root@parameters-optional --server parameters-optional')
    assert send.rc == 0


def test_open_relay(Command):
    """
    Tests if mail forwarder behaves as open relay.
    """

    no_recipients_accepted = 24

    send = Command('swaks --suppress-data --to root@client1 --server parameters-optional')
    assert send.rc == no_recipients_accepted
    assert "Relay access denied" in send.stdout