From 4191754f247885b85222fad2262b6466a2bc85f8 Mon Sep 17 00:00:00 2001
From: Shuduo Sang <sangshuduo@gmail.com>
Date: Fri, 1 May 2020 18:15:38 +0800
Subject: [PATCH] fix test.py to support steven's case

---
 tests/pytest/smoketest.sh | 12 ++++-----
 tests/pytest/test.py      | 57 +++++++++++++++++++--------------------
 2 files changed, 34 insertions(+), 35 deletions(-)

diff --git a/tests/pytest/smoketest.sh b/tests/pytest/smoketest.sh
index af597fb6c5..7dbefa9402 100755
--- a/tests/pytest/smoketest.sh
+++ b/tests/pytest/smoketest.sh
@@ -34,12 +34,12 @@ python3 ./test.py $1 -f table/db_table.py
 python3 ./test.py -s $1
 sleep 1
 
-python3 ./test.py $1 -f import_merge/importDataLastTO.py 
-python3 ./test.py -s $1
-sleep 1
-python3 ./test.py $1 -f import_merge/importDataLastT.py 
-python3 ./test.py -s $1
-sleep 1
+#python3 ./test.py $1 -f import_merge/importDataLastTO.py 
+#python3 ./test.py -s $1
+#sleep 1
+#python3 ./test.py $1 -f import_merge/importDataLastT.py 
+#python3 ./test.py -s $1
+#sleep 1
 python3 ./test.py $1 -f import_merge/importDataTO.py 
 python3 ./test.py -s $1
 sleep 1
diff --git a/tests/pytest/test.py b/tests/pytest/test.py
index 479406a00b..9bf1660634 100644
--- a/tests/pytest/test.py
+++ b/tests/pytest/test.py
@@ -83,36 +83,35 @@ if __name__ == "__main__":
 
         tdLog.exit('stop All dnodes')
 
-    if masterIp == "":
-        tdDnodes.init(deployPath)
-        tdDnodes.setTestCluster(testCluster)
-        tdDnodes.setValgrind(valgrind)
+    tdDnodes.init(deployPath)
+    tdDnodes.setTestCluster(testCluster)
+    tdDnodes.setValgrind(valgrind)
 
-        if testCluster:
-            tdLog.notice("Procedures for testing cluster")
-            if fileName == "all":
-                tdCases.runAllCluster()
-            else:
-                tdCases.runOneCluster(fileName)
-        else:
-            tdLog.notice("Procedures for testing self-deployment")
-            tdDnodes.stopAll()
-            tdDnodes.deploy(1)
-            tdDnodes.start(1)
-            conn = taos.connect(
-                host='127.0.0.1',
-                config=tdDnodes.getSimCfgPath())
-            if fileName == "all":
-                tdCases.runAllLinux(conn)
-            else:
-                tdCases.runOneLinux(conn, fileName)
-            conn.close()
+    tdDnodes.stopAll()
+    tdDnodes.deploy(1)
+    tdDnodes.start(1)
+
+    if masterIp == "":
+        host='127.0.0.1'
     else:
-        tdLog.notice("Procedures for tdengine deployed in %s" % (masterIp))
-        cfgPath = "../../build/test/cfg"   # was: tdDnodes.getSimCfgPath()
-        conn = taos.connect(host=masterIp, config=cfgPath)
+        host=masterIp
+
+    tdLog.notice("Procedures for tdengine deployed in %s" % (host))
+
+    if testCluster:
+        tdLog.notice("Procedures for testing cluster")
         if fileName == "all":
-            tdCases.runAllWindows(conn)
+            tdCases.runAllCluster()
         else:
-            tdCases.runOneWindows(conn, fileName)
-        conn.close()
+            tdCases.runOneCluster(fileName)
+    else:
+        tdLog.notice("Procedures for testing self-deployment")
+        conn = taos.connect(
+                host,
+                config=tdDnodes.getSimCfgPath())
+        if fileName == "all":
+            tdCases.runAllLinux(conn)
+        else:
+            tdCases.runOneLinux(conn, fileName)
+
+    conn.close()