From 86287ddbcb070a74c38cd62c46882ab41488d12e Mon Sep 17 00:00:00 2001
From: wangyazhou1313 <yzwang@taosdata.com>
Date: Fri, 6 Nov 2020 15:23:16 +0800
Subject: [PATCH] [TD-1592]<test>: add test case for kill qeury>

---
 tests/pytest/query/kill_query.py | 82 ++++++++++++++++++++++++++++++++
 1 file changed, 82 insertions(+)
 create mode 100644 tests/pytest/query/kill_query.py

diff --git a/tests/pytest/query/kill_query.py b/tests/pytest/query/kill_query.py
new file mode 100644
index 0000000000..8975eea268
--- /dev/null
+++ b/tests/pytest/query/kill_query.py
@@ -0,0 +1,82 @@
+###################################################################
+#           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 taos
+from util.log import tdLog
+from util.cases import tdCases
+from util.sql import tdSql
+from util.dnodes import tdDnodes
+import os
+import threading
+import time
+
+
+class TDTestCase:
+    """
+    kill query
+    """
+    def init(self, conn, logSql):
+        tdLog.debug("start to execute %s" % __file__)
+        tdSql.init(conn.cursor(), logSql)
+
+
+    def query(self):
+        conn = taos.connect(host='127.0.0.1', user='root', password='taosdata', config='/etc/config')
+        cursor = conn.cursor()
+        while True:
+            cursor.execute('show queries;')
+            print('show queries!')
+            temp = cursor.fetchall()
+            if temp:
+                print(temp[0][0])
+                cursor.execute('kill query %s ;' % temp[0][0])
+                print('kill query success')
+                break
+            time.sleep(0.5)
+        cursor.close()
+        conn.close()
+
+    def run(self):
+        tdSql.prepare()
+
+        print("==============step1")
+        os.system('yes | sudo taosdemo -n 100')
+        print('insert into test.meters 10000000 rows')
+
+
+        t1 = threading.Thread(target=self.query)
+        t1.setDaemon(True)
+        t1.start()
+
+        print("==============step2")
+        tdSql.execute('use test;')
+        try:
+            print('============begin select * from 10000000 rows')
+            tdSql.query('select * from test.meters;')
+            # print(tdSql.queryResult)
+        except Exception as e:
+            if not "ProgrammingError('Query terminated'" in str(e):
+                raise Exception('fail')
+
+        print('success')
+        print('kill query success')
+
+    def stop(self):
+        tdSql.close()
+        tdLog.success("%s successfully executed" % __file__)
+
+
+tdCases.addWindows(__file__, TDTestCase())
+tdCases.addLinux(__file__, TDTestCase())
+
-- 
GitLab