# Copyright OpenSearch Contributors
# SPDX-License-Identifier: Apache-2.0
#
# The OpenSearch Contributors require contributions made to
# this file be licensed under the Apache-2.0 license or a
# compatible open source license.

import abc
import logging
import os
from pathlib import Path

from manifests.component_manifest import Components
from manifests.test_manifest import TestComponent, TestManifest
from system.temporary_directory import TemporaryDirectory
from test_workflow.integ_test.integ_test_suite import IntegTestSuite
from test_workflow.test_args import TestArgs
from test_workflow.test_recorder.test_recorder import TestRecorder
from test_workflow.test_result.test_suite_results import TestSuiteResults


class IntegTestRunner(abc.ABC):
    args: TestArgs
    test_manifest: TestManifest
    tests_dir: str
    test_recorder: TestRecorder
    components: Components

    def __init__(self, args: TestArgs, test_manifest: TestManifest, components: Components) -> None:
        self.args = args
        self.test_manifest = test_manifest
        self.components = components

        self.tests_dir = os.path.join(os.getcwd(), "test-results")
        os.makedirs(self.tests_dir, exist_ok=True)
        self.test_recorder = TestRecorder(self.args.test_run_id, "integ-test", self.tests_dir, args.base_path)

    def run(self) -> TestSuiteResults:
        with TemporaryDirectory(keep=self.args.keep, chdir=True) as work_dir:

            all_results = TestSuiteResults()
            for component in self.components.select(focus=self.args.components):
                if component.name in self.test_manifest.components:
                    test_config = self.test_manifest.components[component.name]
                    if test_config.integ_test:
                        test_suite = self.__create_test_suite__(component, test_config, work_dir.path)
                        test_results = test_suite.execute_tests()
                        [self.test_recorder.test_results_logs.save_test_result_data(result_data) for result_data in test_suite.result_data]
                        all_results.append(component.name, test_results)
                    else:
                        logging.info(f"Skipping integ-tests for {component.name}, as it is currently not supported")
                else:
                    logging.info(f"Skipping integ-tests for {component.name}, as it is currently not declared in the test manifest")

        return all_results

    @abc.abstractmethod
    def __create_test_suite__(self, component: TestComponent, test_config: TestComponent, work_dir: Path) -> IntegTestSuite:
        pass