#!/usr/bin/env python3
# -*- coding: utf-8 -*-

#
# Copyright (c) 2024 Huawei Device Co., Ltd.
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#

import os
import importlib

from util import file_exists, make_binary_file, print_failure, print_success, get_subclasses, get_idl
from test_base import Test

exclude_file = ["test_base.py", "unit_test.py", "util.py"]
for file_name in os.listdir(os.path.dirname(os.path.abspath(__file__))):
    if file_name.endswith('.py'):
        if file_name in exclude_file:
            continue
        module_name = file_name[:-3]
        importlib.import_module(module_name)


class Suite:
    test_cases = get_subclasses(Test)
    test_objs = list()

    @staticmethod
    def set_up_test_case():
        hdi_gen_file = get_idl()
        ret = file_exists(hdi_gen_file)
        if not ret:
            hdi_gen_path = "../../"
            if make_binary_file(hdi_gen_path)[0] == 0:
                ret = True
        if not ret:
            print_failure("[===========] failed to make idl-gen")
        return ret

    @staticmethod
    def run_suite():
        test_case_num = len(Suite.test_cases)
        success_case_num = 0
        print_success("[===========] start {} test".format(test_case_num))
        for test_case in Suite.test_cases:
            obj = test_case()
            Suite.test_objs.append(obj)
            if obj.test():
                success_case_num += 1
        print_success("[    PASSED ] {} test".format(success_case_num))
        failure_case_num = test_case_num - success_case_num
        if failure_case_num > 0:
            print_failure("[    FAILED ] {} test".format(failure_case_num))


if __name__ == "__main__":
    if not Suite.set_up_test_case():
        print_failure("test case set up failed!")
        exit(-1)
    Suite.run_suite()