TDengine/test/new_test_framework/utils/frame/cases.py

###################################################################
#           Copyright (c) 2016 by TAOS Technologies, Inc.
#                     All rights reserved.
#
#  This file is proprietary and confidential to TAOS Technologies.
#  No part of this file may be reproduced, stored, transmitted,
#  disclosed or used in any form or by any means other than as
#  expressly provided by the written permission from Jianhui Tao
#
###################################################################

# -*- coding: utf-8 -*-

import sys
import os
import time
import datetime
import inspect
import importlib
import traceback
from .log import *


class TDCase:
    def __init__(self, name, case):
        self.name = name
        self.case = case
        self._logSql = True


class TDCases:
    def __init__(self):
        """
        Initializes the TDCases class with empty case lists for different platforms.
        """
        self.linuxCases = []
        self.windowsCases = []
        self.clusterCases = []

    def __dynamicLoadModule(self, fileName):
        """
        Dynamically loads a module based on the file name.

        Args:
            fileName (str): The name of the file to load the module from.

        Returns:
            module: The loaded module.
        """
        moduleName = fileName.replace(".py", "").replace(os.sep, ".")
        return importlib.import_module(moduleName, package='..')

    def logSql(self, logSql):
        """
        Sets the logging behavior for SQL statements.

        Args:
            logSql (bool): If True, SQL statements will be logged.
        """
        self._logSql = logSql

    def addWindows(self, name, case):
        """
        Adds a Windows test case.

        Args:
            name (str): The name of the test case.
            case (object): The test case object.
        """
        self.windowsCases.append(TDCase(name, case))

    def addLinux(self, name, case):
        """
        Adds a Linux test case.

        Args:
            name (str): The name of the test case.
            case (object): The test case object.
        """
        self.linuxCases.append(TDCase(name, case))

    def addCluster(self, name, case):
        """
        Adds a cluster test case.

        Args:
            name (str): The name of the test case.
            case (object): The test case object.
        """
        self.clusterCases.append(TDCase(name, case))

    def runAllLinux(self, conn):
        """
        Runs all Linux test cases.

        Args:
            conn (object): The connection object to use for the test cases.
        """
        # TODO: load all Linux cases here
        runNum = 0
        for tmp in self.linuxCases:
            if tmp.name.find(fileName) != -1:
                case = testModule.TDTestCase()
                case.init(conn)
                case.run()
                case.stop()
                runNum += 1
                continue

        tdLog.info("total %d Linux test case(s) executed" % (runNum))

    def runOneLinux(self, conn, fileName, replicaVar=1):
        """
        Runs a specific Linux test case.

        Args:
            conn (object): The connection object to use for the test case.
            fileName (str): The name of the file containing the test case.
            replicaVar (int, optional): The replica variable. Defaults to 1.
        """
        testModule = self.__dynamicLoadModule(fileName)

        runNum = 0
        for tmp in self.linuxCases:
            if tmp.name.find(fileName) != -1:
                case = testModule.TDTestCase()
                case.init(conn, self._logSql, replicaVar)
                try:
                    case.run()
                except Exception as e:
                    tdLog.notice(repr(e))
                    traceback.print_exc()
                    tdLog.exit("%s failed" % (fileName))
                case.stop()
                runNum += 1
                continue

    def runAllWindows(self, conn):
        """
        Runs all Windows test cases.

        Args:
            conn (object): The connection object to use for the test cases.
        """
        # TODO: load all Windows cases here
        runNum = 0
        for tmp in self.windowsCases:
            if tmp.name.find(fileName) != -1:
                case = testModule.TDTestCase()
                case.init(conn)
                case.run()
                case.stop()
                runNum += 1
                continue

        tdLog.notice("total %d Windows test case(s) executed" % (runNum))

    def runOneWindows(self, conn, fileName, replicaVar=1):
        """
        Runs a specific Windows test case.

        Args:
            conn (object): The connection object to use for the test case.
            fileName (str): The name of the file containing the test case.
            replicaVar (int, optional): The replica variable. Defaults to 1.
        """
        testModule = self.__dynamicLoadModule(fileName)

        runNum = 0
        for tmp in self.windowsCases:
            if tmp.name.find(fileName) != -1:
                case = testModule.TDTestCase()
                case.init(conn, self._logSql,replicaVar)
                try:
                    case.run()
                except Exception as e:
                    tdLog.notice(repr(e))
                    tdLog.exit("%s failed" % (fileName))
                case.stop()
                runNum += 1
                continue
        tdLog.notice("total %d Windows case(s) executed" % (runNum))

    def runAllCluster(self):
        """
        Runs all cluster test cases.
        """
        # TODO: load all cluster case module here

        runNum = 0
        for tmp in self.clusterCases:
            if tmp.name.find(fileName) != -1:
                tdLog.notice("run cases like %s" % (fileName))
                case = testModule.TDTestCase()
                case.init()
                case.run()
                case.stop()
                runNum += 1
                continue

        tdLog.notice("total %d Cluster test case(s) executed" % (runNum))

    def runOneCluster(self, fileName):
        """
        Runs a specific cluster test case.

        Args:
            fileName (str): The name of the file containing the test case.
        """
        testModule = self.__dynamicLoadModule(fileName)

        runNum = 0
        for tmp in self.clusterCases:
            if tmp.name.find(fileName) != -1:
                tdLog.notice("run cases like %s" % (fileName))
                case = testModule.TDTestCase()
                case.init()
                case.run()
                case.stop()
                runNum += 1
                continue

        tdLog.notice("total %d Cluster test case(s) executed" % (runNum))


tdCases = TDCases()