test_args.py

# 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.
#
# Modifications Copyright OpenSearch Contributors. See
# GitHub history for details.

import argparse
import logging
import uuid
from typing import List

from test_workflow.test_args_path_validator import TestArgsPathValidator
from test_workflow.test_kwargs import TestKwargs


class TestArgs:
    test_run_id: str
    components: List[str]
    keep: bool
    logging_level: int
    test_manifest_path: str
    paths: dict

    def __init__(self) -> None:
        parser = argparse.ArgumentParser(description="Test an OpenSearch Bundle")
        parser.add_argument("test_manifest_path", type=TestArgsPathValidator.validate, help="Specify a test manifest path.")
        parser.add_argument("-p", "--paths", nargs='*', action=TestKwargs, default={}, help="Specify paths for OpenSearch and OpenSearch Dashboards.")
        parser.add_argument("--test-run-id", type=int, help="The unique execution id for the test")
        parser.add_argument("--component", type=str, dest="components", nargs='*', help="Test a specific component or components instead of the entire distribution.")
        parser.add_argument("--keep", dest="keep", action="store_true", help="Do not delete the working temporary directory.")
        parser.add_argument(
            "-v", "--verbose", help="Show more verbose output.", action="store_const", default=logging.INFO, const=logging.DEBUG, dest="logging_level"
        )
        args = parser.parse_args()
        self.test_run_id = args.test_run_id or uuid.uuid4().hex
        self.components = args.components
        self.keep = args.keep
        self.logging_level = args.logging_level
        self.test_manifest_path = args.test_manifest_path
        self.paths = args.paths


TestArgs.__test__ = False  # type:ignore