From f95f9ac8c0f58b32d196ad77deedad8bb3d7759a Mon Sep 17 00:00:00 2001
From: changshuaiqiang <sqchang@taosdata.com>
Date: Wed, 13 May 2020 18:52:46 +0800
Subject: [PATCH] test: add create table as select case

---
 tests/pytest/insert/multi.py     | 10 +++++--
 tests/pytest/table/create.py     | 46 ++++++++++++++++++++++++++++++++
 tests/pytest/table/del_stable.py |  2 --
 3 files changed, 54 insertions(+), 4 deletions(-)
 create mode 100644 tests/pytest/table/create.py

diff --git a/tests/pytest/insert/multi.py b/tests/pytest/insert/multi.py
index 7725ccd210..c14d7dc2e0 100644
--- a/tests/pytest/insert/multi.py
+++ b/tests/pytest/insert/multi.py
@@ -37,9 +37,15 @@ class TDTestCase:
             "CREATE TABLE if not exists dev_001 using st tags('dev_01')")
 
         print("==============step2")
-        tdLog.info("multiple inserts")
+        tdLog.info("multiple inserts by insert")
         tdSql.execute(
-            "INSERT INTO dev_001 VALUES ('2020-05-13 10:00:00.000', 1),('2020-05-13 10:00:00.001', 1)")
+            "insert INTO dev_001 VALUES ('2020-05-13 10:00:00.000', 1),('2020-05-13 10:00:00.001', 1)")
+        tdSql.checkAffectedRows(2)
+
+        print("==============step3")
+        tdLog.info("multiple inserts by import")
+        tdSql.execute(
+            "import INTO dev_001 VALUES ('2020-05-13 10:00:00.000', 1),('2020-05-13 10:00:00.001', 1)")
         tdSql.checkAffectedRows(2)
 
     def stop(self):
diff --git a/tests/pytest/table/create.py b/tests/pytest/table/create.py
new file mode 100644
index 0000000000..ac65f1d9d9
--- /dev/null
+++ b/tests/pytest/table/create.py
@@ -0,0 +1,46 @@
+###################################################################
+#           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
+
+
+class TDTestCase:
+    def init(self, conn):
+        tdLog.debug("start to execute %s" % __file__)
+        tdSql.init(conn.cursor())
+
+    def run(self):
+        tdSql.prepare()
+
+        print("==============step1")
+        tdSql.execute("create table db.st (ts timestamp, i int) tags(j int)")
+        tdSql.execute("create table db.tb using st tags(1)")
+        tdSql.execute("insert into db.tb values(now, 1)")
+
+        print("==============step2")
+        try:
+            tdSql.execute("create table db.test as select * from db.st")
+        except Exception as e:
+            tdLog.exit(e)
+
+    def stop(self):
+        tdSql.close()
+        tdLog.success("%s successfully executed" % __file__)
+
+
+tdCases.addWindows(__file__, TDTestCase())
+tdCases.addLinux(__file__, TDTestCase())
diff --git a/tests/pytest/table/del_stable.py b/tests/pytest/table/del_stable.py
index de30e6bdfe..3932f32536 100644
--- a/tests/pytest/table/del_stable.py
+++ b/tests/pytest/table/del_stable.py
@@ -26,9 +26,7 @@ class TDTestCase:
     def run(self):
         tdSql.prepare()
 
-        tdSql.execute("drop database if exists db")
         print("==============step1")
-        tdSql.execute("create database db")
         tdSql.execute("create table db.st (ts timestamp, i int) tags(j int)")
         tdSql.execute("create table db.tb using st tags(1)")
         tdSql.execute("insert into db.tb values(now, 1)")
-- 
GitLab