majic-ansible-roles Files · roles/bootstrap/molecule/default/tests/test_parameters

Files @ 33f4baab1260

Branch filter:

Location: majic-ansible-roles/roles/bootstrap/molecule/default/tests/test_parameters_mandatory.py

33f4baab1260 912 B text/x-python Show Annotation Show as Raw Download as Raw

branko

MAR-149: Minor syntax fixes for Python 3 in tests for common role:

- Do not use the unicode function (this still might break, need to
test it).
- Fix syntax used for catching exceptions within variables.
- Fix one small linting error related to row length.

import os

import testinfra.utils.ansible_runner


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


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

    with host.sudo():

        ssh_key = open(os.path.expanduser('~/.ssh/id_rsa.pub'), 'read').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(os.path.expanduser('~/.ssh/id_rsa.pub'), 'read').read().strip()

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