Source code for cfme.fixtures.ssh_client

import pytest

import diaper
from cfme.fixtures.pytest_store import store
from cfme.utils.log import logger
from cfme.utils import ssh


@pytest.mark.hookwrapper
[docs]def pytest_sessionfinish(session, exitstatus): """Loop through the appliance stack and close ssh connections""" for ssh_client in store.ssh_clients_to_close: logger.debug('Closing ssh connection on %r', ssh_client) try: ssh_client.close() except: logger.debug( 'Closing ssh connection on %r failed, but ignoring', ssh_client) for session in ssh._client_session: with diaper: session.close() yield