salt/tests/integration/proxy/conftest.py
2020-02-26 00:57:58 +03:00

40 lines
1.2 KiB
Python

# -*- coding: utf-8 -*-
'''
tests.integration.proxy.conftest
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
Proxy related fixtures
'''
# pylint: disable=unused-argument,redefined-outer-name
# Import Python libs
from __future__ import absolute_import, unicode_literals
import os
import logging
# Import 3rd-party libs
import psutil
import pytest
log = logging.getLogger(__name__)
@pytest.fixture(scope='package', autouse=True)
def session_salt_proxy(request,
session_salt_proxy,
session_proxy_id,
session_master_config):
stats_key = ' Salt Proxy'
request.session.stats_processes[stats_key] = psutil.Process(session_salt_proxy.pid)
yield session_salt_proxy
# Terminate Proxy now, we want to cleanup its key before we move along
session_salt_proxy.terminate()
del request.session.stats_processes[stats_key]
proxy_key_file = os.path.join(session_master_config['pki_dir'], 'minions', session_proxy_id)
log.warning('KEY FILE: %s', proxy_key_file)
if os.path.exists(proxy_key_file):
os.unlink(proxy_key_file)
else:
log.warning('The proxy minion key was not found at %s', proxy_key_file)