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)