majic-ansible-roles Files · roles/backup_server/molecule/default/tests/conftest.py

Files @ b3e67fd6c0c1

Branch filter:

Location: majic-ansible-roles/roles/backup_server/molecule/default/tests/conftest.py

b3e67fd6c0c1 1.0 KiB text/x-python Show Annotation Show as Raw Download as Raw

branko

MAR-187: Reorder the deprecation tasks for xmpp_server role to make the runs reliable:

- Deprecated apt repository really needs to be removed first in case
it has an unresolvable URL (so the backports repository cache update
would trigger correctly).
- Use the file removal for Prosody repository as well to make sure the
file is dropped.

import os

import pytest


@pytest.fixture(scope='session')
def prepare_ssh_client_private_key_permissions():
    """
    Helper fixture used to fix the file permissions of client private
    keys that are used directly from the local machine to test
    connectivity towards the server.

    Permissions are set-up in such a way that the ssh tool does not
    report any errors (e.g. setting the mode to 0600).

    The fixture will restore old permissions once the tests have been
    completed.

    The fixture is used with the session scope.

    The following private keys will be processed:

    - tests/data/ssh/client1
    - tests/data/ssh/client2
    """

    private_keys = [
        "tests/data/ssh/client1",
        "tests/data/ssh/client2"
    ]

    private_key_old_permissions = {}

    for private_key in private_keys:
        private_key_old_permissions[private_key] = os.stat(private_key).st_mode
        os.chmod(private_key, 0o600)

    yield

    for private_key in private_keys:
        os.chmod(private_key, private_key_old_permissions[private_key])