# 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.
# type: ignore

from validation_workflow.docker.validation_docker import ValidateDocker
from validation_workflow.rpm.validation_rpm import ValidateRpm
from validation_workflow.tar.validation_tar import ValidateTar
from validation_workflow.validation import Validation
from validation_workflow.validation_args import ValidationArgs
from validation_workflow.yum.validation_yum import ValidateYum


class ValidationTestRunner:
    RUNNERS = {
        "docker": ValidateDocker,
        "tar": ValidateTar,
        "rpm": ValidateRpm,
        "yum": ValidateYum
    }

    @classmethod
    def dispatch(cls, args: ValidationArgs, dist: str) -> Validation:
        return cls.RUNNERS[dist](args)