82d417b9e6
Fresh start for the StarlingX automation framework. Change-Id: Ie265e0791024f45f71faad6315c2b91b022934d1
71 lines
2.7 KiB
Python
71 lines
2.7 KiB
Python
from datetime import datetime
|
|
|
|
import pytest
|
|
from config.configuration_manager import ConfigurationManager
|
|
from framework.database.objects.test_case_result import TestCaseResult
|
|
from framework.database.objects.testcase import TestCase
|
|
from framework.database.operations.run_content_operation import RunContentOperation
|
|
from framework.database.operations.test_case_result_operation import TestCaseResultOperation
|
|
from framework.runner.objects.test_executor_summary import TestExecutorSummary
|
|
|
|
|
|
class ResultCollector:
|
|
"""
|
|
Pytest plugin that allows us to get results and add them to the test summary object
|
|
"""
|
|
|
|
def __init__(self, test_executor_summary: TestExecutorSummary, test: TestCase, test_case_result_id: int = None):
|
|
self.test_executor_summary = test_executor_summary
|
|
self.test = test
|
|
self.start_time = datetime.now() # start time for the test
|
|
self.test_case_result_id = test_case_result_id
|
|
|
|
@pytest.hookimpl(tryfirst=True, hookwrapper=True)
|
|
def pytest_runtest_makereport(self, item, call):
|
|
"""
|
|
Called at the end of the pytest test, we then can append the test summary
|
|
Args:
|
|
item (): the test
|
|
call (): the stage of the test
|
|
|
|
Returns:
|
|
|
|
"""
|
|
outcome = yield
|
|
report = outcome.get_result()
|
|
if report.when == 'call':
|
|
self.test_executor_summary.increment_test_index()
|
|
self.test_executor_summary.append_tests_summary(f"{report.outcome} " f"{item.nodeid}")
|
|
if ConfigurationManager.get_database_config().use_database():
|
|
self.update_result_in_database(
|
|
report.outcome.upper(),
|
|
)
|
|
|
|
def update_result_in_database(self, outcome):
|
|
"""
|
|
Updates the result in the database
|
|
Args:
|
|
outcome (): the result of the test
|
|
|
|
Returns:
|
|
|
|
"""
|
|
|
|
# set values to PASS or FAIL
|
|
if outcome == 'PASSED':
|
|
outcome = 'PASS'
|
|
else:
|
|
outcome = 'FAIL'
|
|
|
|
# if we've been given a testcase result id, update the result and don't create a new one
|
|
if self.test_case_result_id:
|
|
test_case_result = TestCaseResult(-1, outcome, self.start_time, datetime.now())
|
|
test_case_result.set_test_case_result_id(self.test_case_result_id)
|
|
TestCaseResultOperation().update_test_case_result(test_case_result)
|
|
|
|
else:
|
|
# we don't have a result yet so create one
|
|
test_case_result = TestCaseResult(self.test.get_test_info_id(), outcome, self.start_time, datetime.now())
|
|
TestCaseResultOperation().create_test_case_result(test_case_result)
|
|
RunContentOperation().update_execution_status(self.test.get_run_content_id(), outcome)
|