# 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 logging
from typing import Any

from validation_workflow.api_request import ApiTest
from validation_workflow.validation_args import ValidationArgs

'''
This class is the collection of test cases to run.
'''


class ApiTestCases:

    def __init__(self) -> None:
        self.opensearch_image_version = ValidationArgs().stg_tag('opensearch').split(' ')[0]
        self.opensearch_dashboards_image_version = ValidationArgs().stg_tag('opensearch_dashboards').split(' ')[0]

    def test_cases(self) -> Any:
        pass_counter, fail_counter = 0, 0

        # the test case parameters are formated as ['<request_url>',<success_status_code>,'<validate_string(optional)>']
        test_cases = [
            ['https://localhost:9200/', 200, '"number" : "' + self.opensearch_image_version + '"'],
            ['https://localhost:9200/_cat/plugins?v', 200, ''],
            ['https://localhost:9200/_cat/health?v', 200, 'green'],
            ['http://localhost:5601/api/status', 200, '"number":"' + self.opensearch_dashboards_image_version + '"']
        ]

        for test_case in test_cases:
            request_url = test_case.__getitem__(0)
            success_status_code = test_case.__getitem__(1)
            validate_string = test_case.__getitem__(2)

            status_code, response_text = ApiTest(str(request_url)).api_get()
            logging.info(f"\nRequest_url ->{str(request_url)} \n")
            logging.info(f"\nStatus_code ->{status_code} \nresponse_text ->{response_text}")

            if status_code == success_status_code and (not validate_string or validate_string in response_text):
                pass_counter += 1
            else:
                fail_counter += 1

        return (fail_counter == 0, f"There are {pass_counter}/{pass_counter + fail_counter} test cases Pass")