import testinfra.utils.ansible_runner


testinfra_hosts = testinfra.utils.ansible_runner.AnsibleRunner(
    '.molecule/ansible_inventory').get_hosts('parameters-optional')


def test_authorized_keys(File, Sudo):
    """
    Tests if Ansible user authorized_keys has been set-up correctly.
    """

    with Sudo():

        ssh_key = open('tests/data/ansible_key.pub', 'read').read().strip()
        authorized_keys = File('/home/ansible/.ssh/authorized_keys')

        assert authorized_keys.is_file
        assert ssh_key in authorized_keys.content


def test_root_authorised_keys(File, Sudo):
    """
    Tests if Ansible key been removed from root's authorized keys.
    """

    with Sudo():

        ssh_key = open('tests/data/ansible_key.pub', 'read').read().strip()

        assert ssh_key not in File('/root/.ssh/authorized_keys').content