def test_authorized_keys(host):

    """

    Tests if Ansible user authorized_keys has been set-up correctly.

    """

    with host.sudo():

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

        authorized_keys = host.file('/home/ansible/.ssh/authorized_keys')

        assert authorized_keys.is_file

        assert ssh_key in authorized_keys.content_string

def test_root_authorized_keys(host):

    """

    Tests if Ansible key been removed from root's authorized keys.

    """

    with host.sudo():

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

        assert ssh_key not in host.file('/root/.ssh/authorized_keys').content_string