homework-jianmu/tests/system-test/2-query/insert_select.py

220 lines
8.8 KiB
Python

###################################################################
# 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 random
import os
import time
import subprocess
from util.log import tdLog
from util.cases import tdCases
from util.sql import tdSql
class TDTestCase:
updatecfgDict = {'maxSQLLength':1048576,'debugFlag': 143 ,"querySmaOptimize":1}
def init(self, conn, logSql, replicaVar):
tdLog.debug("start to execute %s" % __file__)
tdSql.init(conn.cursor(), logSql)
self.testcasePath = os.path.split(__file__)[0]
self.testcaseFilename = os.path.split(__file__)[-1]
os.system("rm -rf %s/%s.sql" % (self.testcasePath,self.testcaseFilename))
self.db = "insert_select"
def dropandcreateDB_random(self,database,n):
ts = 1604298064000
tdSql.execute('''drop database if exists %s ;''' %database)
tdSql.execute('''create database %s keep 36500 ;'''%(database))
tdSql.execute('''use %s;'''%database)
tdSql.execute('''create table %s.tb (ts timestamp , i tinyint );'''%database)
tdSql.execute('''create table %s.tb1 (speed timestamp , c1 int , c2 int , c3 int );'''%database)
sql_before = "insert into %s.tb1 values" %database
sql_value = ''
for i in range(n):
sql_value = sql_value +"(%d, %d, %d, %d)" % (ts + i, i, i, i)
sql = sql_before + sql_value
tdSql.execute(sql)
tdSql.query("select count(*) from %s.tb1;"%database)
sql_result = tdSql.getData(0,0)
tdLog.info("result: %s" %(sql_result))
tdSql.query("reset query cache;")
tdSql.query("insert into %s.tb1 select * from %s.tb1;"%(database,database))
tdSql.query("select count(*) from %s.tb1;"%database)
sql_result = tdSql.getData(0,0)
tdLog.info("result: %s" %(sql_result))
def users_bug_TD_20592(self,database):
tdSql.execute('''drop database if exists %s ;''' %database)
tdSql.execute('''create database %s keep 36500 ;'''%(database))
tdSql.execute('''use %s;'''%database)
tdSql.execute('''create table %s.sav_stb (ts timestamp, c1 int, c2 bigint, c3 float, c4 double, c5 smallint, c6 tinyint, c7 bool, c8 binary(10), c9 nchar(10)) tags(t1 int, t2 int);'''%database)
tdSql.execute('''create table %s.tb1 using %s.sav_stb tags( 9 , 0);'''%(database,database))
tdSql.error('''insert into %s.tb1 (c8, c9) values(now, 1);'''%(database))
def use_select_sort(self,database):
ts = 1604298064000
tdSql.execute('''drop database if exists %s ;''' %database)
tdSql.execute('''create database %s keep 36500 ;'''%(database))
tdSql.execute('''use %s;'''%database)
tdSql.execute('''create stable %s.st (ts timestamp, val int, vt timestamp) tags (location NCHAR(100));'''%(database))
tdSql.execute('''create table %s.t1 using %s.st (location) tags ("0001");'''%(database,database))
tdSql.execute('''create table %s.t2 using %s.st (location) tags ("0002");'''%(database,database))
tdSql.execute('''create table %s.mt (ts timestamp, val int);'''%(database))
tdSql.execute(f'''insert into %s.t1 values({ts}, 1, {ts}) ({ts}, 2, {ts});'''%(database))
tdSql.query("select ts, val from %s.t1;"%database)
tdSql.checkData(0,1,2)
ts += 1
tdSql.execute(f'''insert into %s.t2 values({ts}, 1, {ts}) ({ts}, 5, {ts}) ({ts}, 2, {ts});'''%(database))
tdSql.query("select ts, val from %s.t2;"%database)
tdSql.checkData(0,1,2)
tdSql.execute('''delete from %s.t2;'''%(database))
tdSql.execute('''delete from %s.t1;'''%(database))
ts -= 10
tdSql.execute(f'''insert into %s.t1 values({ts}, 1, {ts}) %s.t2 values({ts}, 2, {ts});'''%(database,database))
ts += 11
tdSql.execute(f'''insert into %s.t1 values({ts}, 1, {ts}) %s.t2 values({ts}, 2, {ts});'''%(database,database))
ts += 1
tdSql.execute(f'''insert into %s.t1 values({ts}, 1, {ts}) %s.t2 values({ts}, 2, {ts});'''%(database,database))
tdSql.query("select count(*) from %s.st;"%database)
tdSql.checkData(0,0,6)
tdSql.query('''select vt, val from %s.st order by vt, val desc;'''%(database))
tdSql.checkData(0,1,2)
tdSql.checkData(1,1,1)
tdSql.checkData(2,1,2)
tdSql.checkData(3,1,1)
tdSql.checkData(4,1,2)
tdSql.checkData(5,1,1)
tdSql.execute('''insert into %s.mt select vt, val from %s.st order by vt, val desc;'''%(database,database))
tdSql.query("select count(*) from %s.mt;"%database)
tdSql.checkData(0,0,3)
tdSql.query('''select ts, val from %s.mt order by ts asc;'''%(database))
tdSql.checkData(0,1,1)
tdSql.checkData(1,1,1)
tdSql.checkData(2,1,1)
tdSql.execute('''delete from %s.mt;'''%(database))
tdSql.query('''select vt, val from %s.st order by vt, val asc;'''%(database))
tdSql.checkData(0,1,1)
tdSql.checkData(1,1,2)
tdSql.checkData(2,1,1)
tdSql.checkData(3,1,2)
tdSql.checkData(4,1,1)
tdSql.checkData(5,1,2)
tdSql.execute('''insert into %s.mt select vt, val from %s.st order by vt, val asc;'''%(database,database))
tdSql.query("select count(*) from %s.mt;"%database)
tdSql.checkData(0,0,3)
tdSql.query('''select ts, val from %s.mt order by ts asc;'''%(database))
tdSql.checkData(0,1,2)
tdSql.checkData(1,1,2)
tdSql.checkData(2,1,2)
tdSql.execute('''delete from %s.mt;'''%(database))
tdSql.query('''select vt, val from %s.st order by ts, val asc;'''%(database))
tdSql.checkData(0,1,1)
tdSql.checkData(1,1,2)
tdSql.checkData(2,1,1)
tdSql.checkData(3,1,2)
tdSql.checkData(4,1,1)
tdSql.checkData(5,1,2)
tdSql.execute('''insert into %s.mt select vt, val from %s.st order by ts, val asc;'''%(database,database))
tdSql.query("select count(*) from %s.mt;"%database)
tdSql.checkData(0,0,3)
tdSql.query('''select ts, val from %s.mt order by ts asc;'''%(database))
tdSql.checkData(0,1,2)
tdSql.checkData(1,1,2)
tdSql.checkData(2,1,2)
tdSql.execute('''delete from %s.mt;'''%(database))
ts += 1
tdSql.execute(f'''insert into %s.t1 values({ts}, -1, {ts}) %s.t2 values({ts}, -2, {ts});'''%(database,database))
tdSql.query('''select vt, val from %s.st order by val asc;'''%(database))
tdSql.checkData(0,1,-2)
tdSql.checkData(1,1,-1)
tdSql.checkData(2,1,1)
tdSql.checkData(3,1,1)
tdSql.checkData(4,1,1)
tdSql.checkData(5,1,2)
tdSql.checkData(6,1,2)
tdSql.checkData(7,1,2)
tdSql.execute('''insert into %s.mt select vt, val from %s.st order by val asc;'''%(database,database))
tdSql.query("select count(*) from %s.mt;"%database)
tdSql.checkData(0,0,4)
tdSql.query('''select ts, val from %s.mt order by ts asc;'''%(database))
tdSql.checkData(0,1,2)
tdSql.checkData(1,1,2)
tdSql.checkData(2,1,2)
tdSql.checkData(3,1,-1)
def run(self):
startTime = time.time()
os.system("rm -rf %s/%s.sql" % (self.testcasePath,self.testcaseFilename))
self.dropandcreateDB_random("%s" %self.db, random.randint(10000,30000))
#taos -f sql
print("taos -f sql start!")
taos_cmd1 = "taos -f %s/%s.sql" % (self.testcasePath,self.testcaseFilename)
_ = subprocess.check_output(taos_cmd1, shell=True)
print("taos -f sql over!")
self.users_bug_TD_20592("%s" %self.db)
self.use_select_sort("%s" %self.db)
#taos -f sql
print("taos -f sql start!")
taos_cmd1 = "taos -f %s/%s.sql" % (self.testcasePath,self.testcaseFilename)
_ = subprocess.check_output(taos_cmd1, shell=True)
print("taos -f sql over!")
endTime = time.time()
print("total time %ds" % (endTime - startTime))
def stop(self):
tdSql.close()
tdLog.success("%s successfully executed" % __file__)
tdCases.addWindows(__file__, TDTestCase())
tdCases.addLinux(__file__, TDTestCase())