# Copyright 2019-2021 IBM Corp. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""
Pytest fixtures for mocked HMCs.
"""
from __future__ import absolute_import
import os
import logging
import pytest
import zhmcclient
import zhmcclient_mock
from ._hmc_definition import HMCDefinition
from ._hmc_definitions import hmc_definitions
__all__ = ['hmc_definition', 'hmc_session']
# Log file
TESTLOGFILE = os.getenv('TESTLOGFILE', None)
if TESTLOGFILE:
LOG_HANDLER = logging.FileHandler(TESTLOGFILE, encoding='utf-8')
LOG_FORMAT_STRING = '%(asctime)s %(name)s %(levelname)s %(message)s'
LOG_HANDLER.setFormatter(logging.Formatter(LOG_FORMAT_STRING))
else:
LOG_HANDLER = None
LOG_FORMAT_STRING = None
def fixtureid_hmc_definition(fixture_value):
"""
Return a fixture ID to be used by pytest, for fixture `hmc_definition()`.
Parameters:
* fixture_value (HMCDefinition): The HMC definition of the HMC the test
runs against.
"""
if not isinstance(fixture_value, HMCDefinition):
return None # Use pytest auto-generated ID
hd = fixture_value
show_hmc = False # Enable to show HMC/userid or mock file in tests
if not show_hmc:
hmc_str = ""
elif hd.mock_file:
hmc_str = "(mock_file={f})".format(f=hd.mock_file)
else:
hmc_str = "(host={h}, userid={u})".format(
h=hd.host, u=hd.userid)
ret_str = "hmc_definition={n}{v}".format(n=hd.nickname, v=hmc_str)
return ret_str
[docs]@pytest.fixture(
params=hmc_definitions(load=None),
scope='module',
ids=fixtureid_hmc_definition
)
def hmc_definition(request):
"""
Fixture representing the set of HMC definitions to use for the end2end
tests.
A test function parameter using this fixture resolves to the
:class:`~zhmcclient.testutils.HMCDefinition`
object of each HMC to test against.
"""
return request.param
[docs]@pytest.fixture(
scope='module'
)
def hmc_session(request, hmc_definition):
# pylint: disable=redefined-outer-name,unused-argument
"""
Pytest fixture representing the set of HMC sessions to use for the
end2end tests.
Because the `hmc_definition` parameter of this fixture is again a fixture,
the :func:`zhmcclient.testutils.hmc_definition` function needs to be
imported as well when this fixture is used.
A test function parameter using this fixture resolves to the
:class:`zhmcclient.Session` or :class:`zhmcclient_mock.FakedSession` object
for the HMC session to test against.
The session is already logged on to the HMC.
Upon fixture teardown, the session is automatically logged off from the HMC.
"""
session = setup_hmc_session(hmc_definition)
yield session
teardown_hmc_session(session)
def setup_hmc_session(hd):
"""
Setup an HMC session and return a new zhmcclient.Session object for it.
If the HMC definition represents a real HMC, log on to an HMC and return
a new zhmcclient.Session object.
If the HMC definition represents a mocked HMC, create a new mock environment
from that and return a zhmcclient_mock.FakedSession object.
"""
# We use the cached skip reason from previous attempts
skip_msg = getattr(hd, 'skip_msg', None)
if skip_msg:
pytest.skip("Skip reason from earlier attempt: {0}".format(skip_msg))
if hd.mock_file:
# A mocked HMC
# Create a mocked session using the mock file from the inventory file
filepath = os.path.join(os.path.dirname(hd.filepath), hd.mock_file)
session = zhmcclient_mock.FakedSession.from_hmc_yaml_file(filepath)
else:
# A real HMC
# Enable debug logging if specified
if LOG_HANDLER:
logger = logging.getLogger('zhmcclient.hmc')
if LOG_HANDLER not in logger.handlers:
logger.addHandler(LOG_HANDLER)
logger.setLevel(logging.DEBUG)
logger = logging.getLogger('zhmcclient.api')
if LOG_HANDLER not in logger.handlers:
logger.addHandler(LOG_HANDLER)
logger.setLevel(logging.DEBUG)
rt_config = zhmcclient.RetryTimeoutConfig(
read_timeout=300,
)
# Creating a session does not interact with the HMC (logon is deferred)
session = zhmcclient.Session(
hd.host, hd.userid, hd.password, verify_cert=hd.verify_cert,
retry_timeout_config=rt_config)
# Check access to the HMC
try:
session.logon()
except zhmcclient.Error as exc:
msg = "Cannot log on to HMC {0} at {1} due to {2}: {3}". \
format(hd.nickname, hd.host, exc.__class__.__name__, exc)
hd.skip_msg = msg
pytest.skip(msg)
hd.skip_msg = None
session.hmc_definition = hd
return session
def teardown_hmc_session(session):
"""
Log off from an HMC session.
"""
session.logoff()