82d417b9e6
Fresh start for the StarlingX automation framework. Change-Id: Ie265e0791024f45f71faad6315c2b91b022934d1
114 lines
3.6 KiB
Python
114 lines
3.6 KiB
Python
import os
|
|
|
|
from config.configuration_file_locations_manager import ConfigurationFileLocationsManager
|
|
from config.configuration_manager import ConfigurationManager
|
|
from framework.logging import log_banners
|
|
from framework.logging.automation_logger import configure_testcase_log_handler, get_logger, remove_testcase_handler
|
|
|
|
|
|
def pytest_addoption(parser):
|
|
parser.addoption("--lab_config_file", action="store")
|
|
parser.addoption("--k8s_config_file", action="store")
|
|
parser.addoption("--logger_config_file", action="store")
|
|
parser.addoption("--docker_config_file", action="store")
|
|
parser.addoption("--web_config_file", action="store")
|
|
parser.addoption("--database_config_file", action="store")
|
|
parser.addoption("--rest_api_config_file", action="store")
|
|
|
|
def pytest_sessionstart(session):
|
|
"""
|
|
This is run once at test start up.
|
|
Returns:
|
|
|
|
"""
|
|
|
|
configuration_locations_manager = ConfigurationFileLocationsManager()
|
|
|
|
configuration_locations_manager.set_configs_from_pytest_args(session)
|
|
|
|
ConfigurationManager.load_configs(configuration_locations_manager)
|
|
log_configuration()
|
|
|
|
|
|
def log_configuration():
|
|
"""
|
|
This function will log all the configurations that are loaded
|
|
"""
|
|
|
|
get_logger().log_debug("----- LOGGER CONFIG -----")
|
|
for config_string in ConfigurationManager.get_logger_config().to_log_strings():
|
|
get_logger().log_debug(config_string)
|
|
|
|
get_logger().log_debug("----- LAB CONFIG -----")
|
|
for config_string in ConfigurationManager.get_lab_config().to_log_strings():
|
|
get_logger().log_debug(config_string)
|
|
|
|
|
|
def pytest_runtest_setup(item):
|
|
"""
|
|
This will run before any test case starts its execution
|
|
Args:
|
|
item: The test case item that we are about to execute.
|
|
|
|
Returns: None
|
|
|
|
"""
|
|
|
|
# add testcase log handler at test start
|
|
configure_testcase_log_handler(ConfigurationManager.get_logger_config(), item.name)
|
|
log_banners.log_test_start_banner(item)
|
|
|
|
|
|
def pytest_runtest_makereport(item, call):
|
|
"""
|
|
This Pytest hook gets called after the execution of a test case or test suite.
|
|
Args:
|
|
item: The test case that was executed.
|
|
call: Information about the execution.
|
|
|
|
Returns: None
|
|
|
|
"""
|
|
|
|
# Handle Exceptions that happened during the test case.
|
|
if call.excinfo:
|
|
log_exception(call.excinfo)
|
|
if call.when == 'teardown':
|
|
# remove testcase logger file handler at end of each test
|
|
remove_testcase_handler(item.name)
|
|
|
|
|
|
def log_exception(exception_info):
|
|
"""
|
|
This function will log the StackTrace of the exception_info passed in.
|
|
Args:
|
|
exception_info (ExceptionInfo): Obtained from call.excinfo in pytest_runtest_makereport
|
|
|
|
Returns: None
|
|
|
|
"""
|
|
|
|
get_logger().log_exception("")
|
|
get_logger().log_exception("---------- EXCEPTION ----------")
|
|
|
|
full_traceback = exception_info.traceback
|
|
|
|
# Remove the Traceback Entries from on_keyword_wrapped_function
|
|
for entry in full_traceback:
|
|
|
|
# Skip over the keyword-wrapping-hook
|
|
if entry.name == 'on_keyword_wrapped_function':
|
|
continue
|
|
|
|
# Skip over run_engine functions
|
|
filename = entry.path
|
|
relative_filename = os.path.relpath(filename)
|
|
if "_pytest" in relative_filename or "pluggy" in relative_filename:
|
|
continue
|
|
|
|
get_logger().log_exception(f"{relative_filename}:{entry.lineno} in {entry.name}")
|
|
get_logger().log_exception(entry.statement)
|
|
|
|
get_logger().log_exception(f"EXCEPTION! {exception_info.typename}: {exception_info.value}")
|
|
get_logger().log_exception("")
|