case: remove util folder

tests/army/frame/autogen.py

@@ -1,9 +1,9 @@
 # -*- coding: utf-8 -*-
 
 import sys
-from util.log import *
+from log import *
-from util.cases import *
+from cases import *
-from util.sql import *
+from sql import *
 import threading
 import random
 import string

tests/army/frame/cases.py

@@ -18,7 +18,7 @@ import datetime
 import inspect
 import importlib
 import traceback
-from util.log import *
+from log import *
 
 
 class TDCase:

tests/army/frame/cluster.py

@@ -5,11 +5,11 @@ import time
 import os 
 import  socket
 
-from util.log import *
+from log import *
-from util.sql import *
+from sql import *
-from util.cases import *
+from cases import *
-from util.dnodes import *
+from dnodes import *
-from util.common import *
+from common import *
 
 class ClusterDnodes(TDDnodes):
     """rewrite TDDnodes and make MyDdnodes as TDDnodes child class"""

tests/army/frame/common.py

@@ -18,14 +18,14 @@ import time
 import socket
 import json
 import toml
-from util.boundary import DataBoundary
+from boundary import DataBoundary
 import taos
-from util.log import *
+from log import *
-from util.sql import *
+from sql import *
-from util.cases import *
+from cases import *
-from util.dnodes import *
+from dnodes import *
-from util.common import *
+from common import *
-from util.constant import *
+from constant import *
 from dataclasses import dataclass,field
 from typing import List
 from datetime import datetime

tests/army/frame/dnodes-default.py

@@ -15,7 +15,7 @@ import sys
 import os
 import os.path
 import subprocess
-from util.log import *
+from log import *
 
 
 class TDSimClient:

tests/army/frame/dnodes-no-random-fail.py

@@ -15,7 +15,7 @@ import sys
 import os
 import os.path
 import subprocess
-from util.log import *
+from log import *
 
 
 class TDSimClient:

tests/army/frame/dnodes-random-fail.py

@@ -15,7 +15,7 @@ import sys
 import os
 import os.path
 import subprocess
-from util.log import *
+from log import *
 
 
 class TDSimClient:

tests/army/frame/dnodes.py

@@ -22,7 +22,7 @@ import base64
 import json
 import copy
 from fabric2 import Connection
-from util.log import *
+from log import *
 from shutil import which
 
 

tests/army/frame/pathFinding.py

@@ -13,7 +13,7 @@
 
 
 import os
-from util.log import *
+from log import *
 
 
 

tests/army/frame/sql.py

@@ -20,8 +20,8 @@ import traceback
 import psutil
 import shutil
 import pandas as pd
-from util.log import *
+from log import *
-from util.constant import *
+from constant import *
 
 # from datetime import timezone
 import time
@@ -607,14 +607,14 @@ class TDSql:
         pstate = 0
         for i in range(30):
             pstate = 0
-            pl = psutil.pids()
+            pl = pspids()
             for pid in pl:
                 try:
-                    if psutil.Process(pid).name() == 'taosd':
+                    if psProcess(pid).name() == 'taosd':
                         print('have already started')
                         pstate = 1
                         break
-                except psutil.NoSuchProcess:
+                except psNoSuchProcess:
                     pass
             if pstate == state :break
             if state or pstate:
@@ -646,7 +646,7 @@ class TDSql:
             tdLog.exit("dir: %s doesn't exist" %dir)
     def createDir(self, dir):
         if os.path.exists(dir):
-            shutil.rmtree(dir)
+            shrmtree(dir)
             tdLog.info("dir: %s is removed" %dir)
         os.makedirs( dir, 755 )
         tdLog.info("dir: %s is created" %dir)

tests/army/frame/sqlset.py

@@ -11,7 +11,7 @@
 
 # -*- coding: utf-8 -*-
 
-from util.sql import tdSql
+from sql import tdSql
 
 class TDSetSql:
     def init(self, conn, logSql):

tests/army/frame/sub.py

@@ -15,7 +15,7 @@ import sys
 import os
 import time
 import datetime
-from util.log import *
+from log import *
 
 class TDSub:
 	def __init__(self):

tests/army/frame/taosadapter.py

@@ -1,7 +1,7 @@
 import requests
 from fabric2 import Connection
-from util.log import *
+from log import *
-from util.common import *
+from common import *
 
 
 class TAdapter:

tests/army/frame/taosdemoCfg.py

@@ -19,7 +19,7 @@ import inspect
 import psutil
 import shutil
 import json
-from util.log import *
+from log import *
 from multiprocessing import cpu_count
 
 

tests/army/pytest.sh

@@ -58,6 +58,12 @@ echo "SIM_DIR  : $SIM_DIR"
 echo "CODE_DIR : $CODE_DIR"
 echo "ASAN_DIR  : $ASAN_DIR"
 
+# prevent delete / folder or /usr/bin
+if [ ${#SIM_DIR} -lt 10 ]; then
+   echo "len(SIM_DIR) < 10 , danger so exit. SIM_DIR=$SIM_DIR"
+   exit 1
+fi
+
 rm -rf $SIM_DIR/*
 
 mkdir -p $PRG_DIR

tests/system-test/1-insert/alter_replica.py

@@ -12,7 +12,7 @@ from util.cases import *
 from util.dnodes import *
 
 
-class TDTestCase:
+class TDTestCase(TBaseCase):
     def init(self, conn, logSql, replicaVar=1):
         self.replicaVar = int(replicaVar)
         tdLog.debug("start to execute %s" % __file__)

tests/system-test/pytest.sh

@@ -58,6 +58,12 @@ echo "SIM_DIR  : $SIM_DIR"
 echo "CODE_DIR : $CODE_DIR"
 echo "ASAN_DIR  : $ASAN_DIR"
 
+# prevent delete / folder or /usr/bin
+if [ ${#SIM_DIR} -lt 10 ]; then
+   echo "len(SIM_DIR) < 10 , danger so exit. SIM_DIR=$SIM_DIR"
+   exit 1
+fi
+
 rm -rf $SIM_DIR/*
 
 mkdir -p $PRG_DIR