salt/tests/integration/conftest.py

# -*- coding: utf-8 -*-
"""
    tests.integration.conftest
    ~~~~~~~~~~~~~~~~~~~~~~~~~~

    Integration tests PyTest configuration/fixtures
"""
# pylint: disable=unused-argument,redefined-outer-name

# Import Python libs
from __future__ import absolute_import, unicode_literals

import logging
from collections import OrderedDict

# Import 3rd-party libs
import psutil
import pytest

log = logging.getLogger(__name__)


@pytest.fixture(scope="package", autouse=True)
def default_session_daemons(
    request,
    log_server,
    session_salt_master,
    session_salt_minion,
    session_secondary_salt_minion,
):

    request.session.stats_processes.update(
        OrderedDict(
            (
                ("Salt Master", psutil.Process(session_salt_master.pid)),
                ("Salt Minion", psutil.Process(session_salt_minion.pid)),
                ("Salt Sub Minion", psutil.Process(session_secondary_salt_minion.pid)),
            )
        ).items()
    )

    # Run tests
    yield

    # Stop daemons now(they would be stopped at the end of the test run session
    for daemon in (
        session_secondary_salt_minion,
        session_salt_minion,
        session_salt_master,
    ):
        try:
            daemon.terminate()
        except Exception as exc:  # pylint: disable=broad-except
            log.warning("Failed to terminate daemon: %s", daemon.__class__.__name__)