From 4d8fb3b91b2b161565a35eca3329b84df9d5ebe6 Mon Sep 17 00:00:00 2001
From: Shuduo Sang <sdsang@taosdata.com>
Date: Mon, 22 Jun 2020 18:41:50 +0800
Subject: [PATCH] add client/client.py to test client functions.

[TD-725]
---
 tests/pytest/client/client.py  | 38 ++++++++++++++++++++++++++++++++++
 tests/pytest/fulltest.sh       |  3 ++-
 tests/pytest/regressiontest.sh |  1 +
 tests/pytest/smoketest.sh      |  4 ++++
 tests/pytest/valgrind-test.sh  |  5 +++++
 5 files changed, 50 insertions(+), 1 deletion(-)
 create mode 100644 tests/pytest/client/client.py

diff --git a/tests/pytest/client/client.py b/tests/pytest/client/client.py
new file mode 100644
index 0000000000..b3a93bd363
--- /dev/null
+++ b/tests/pytest/client/client.py
@@ -0,0 +1,38 @@
+###################################################################
+#           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
+from util.log import *
+from util.cases import *
+from util.sql import *
+
+
+class TDTestCase:
+    def init(self, conn, logSql):
+        tdLog.debug("start to execute %s" % __file__)
+        tdSql.init(conn.cursor(), logSql)
+
+    def run(self):
+        tdSql.prepare()
+
+        ret = tdSql.query('select database()')
+
+        tdSql.checkData(0, 0, "db")
+
+    def stop(self):
+        tdSql.close()
+        tdLog.success("%s successfully executed" % __file__)
+
+
+tdCases.addWindows(__file__, TDTestCase())
+tdCases.addLinux(__file__, TDTestCase())
diff --git a/tests/pytest/fulltest.sh b/tests/pytest/fulltest.sh
index 0bacd63ec1..37f436ef78 100755
--- a/tests/pytest/fulltest.sh
+++ b/tests/pytest/fulltest.sh
@@ -1,6 +1,7 @@
 #!/bin/bash
 ulimit -c unlimited
 
+python3 ./test.py -f client/client.py
 python3 ./test.py -f insert/basic.py
 python3 ./test.py -f insert/int.py
 python3 ./test.py -f insert/float.py
@@ -148,4 +149,4 @@ python3 ./test.py -f stream/stream1.py
 python3 ./test.py -f stream/stream2.py
 
 #alter table
-python3 ./test.py -f alter/alter_table_crash.py
\ No newline at end of file
+python3 ./test.py -f alter/alter_table_crash.py
diff --git a/tests/pytest/regressiontest.sh b/tests/pytest/regressiontest.sh
index 955d6aa2ef..0c248f3387 100755
--- a/tests/pytest/regressiontest.sh
+++ b/tests/pytest/regressiontest.sh
@@ -1,6 +1,7 @@
 #!/bin/bash
 ulimit -c unlimited
 
+python3 ./test.py -f client/client.py
 python3 ./test.py -f insert/basic.py
 python3 ./test.py -f insert/int.py
 python3 ./test.py -f insert/float.py
diff --git a/tests/pytest/smoketest.sh b/tests/pytest/smoketest.sh
index 1c51da397e..c56a4726dd 100755
--- a/tests/pytest/smoketest.sh
+++ b/tests/pytest/smoketest.sh
@@ -1,6 +1,10 @@
 #!/bin/bash
 ulimit -c unlimited
 
+# client
+python3 ./test.py $1 -f client/client.py
+python3 ./test.py $1 -s && sleep 1
+
 # insert
 python3 ./test.py $1 -f insert/basic.py
 python3 ./test.py $1 -s && sleep 1
diff --git a/tests/pytest/valgrind-test.sh b/tests/pytest/valgrind-test.sh
index bf42cd59cd..9dc9de1b15 100755
--- a/tests/pytest/valgrind-test.sh
+++ b/tests/pytest/valgrind-test.sh
@@ -1,4 +1,9 @@
 #!/bin/bash
+
+# client
+PYTHONMALLOC=malloc python3 ./test.py -g -f client/client.py
+PYTHONMALLOC=malloc python3 ./test.py -g -s && sleep 1
+
 # insert
 PYTHONMALLOC=malloc python3 ./test.py -g -f insert/basic.py
 PYTHONMALLOC=malloc python3 ./test.py -g -s && sleep 1
-- 
GitLab