################################################################### # 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 os import time from util.log import * from util.cases import * from util.sql import * from util.dnodes import * class TDTestCase: def init(self, conn, logSql): tdLog.debug("start to execute %s" % __file__) tdSql.init(conn.cursor(), logSql) def getPath(self, tool="taosBenchmark"): selfPath = os.path.dirname(os.path.realpath(__file__)) if ("community" in selfPath): projPath = selfPath[:selfPath.find("community")] else: projPath = selfPath[:selfPath.find("tests")] paths = [] for root, dirs, files in os.walk(projPath): if ((tool) in files): rootRealPath = os.path.dirname(os.path.realpath(root)) if ("packaging" not in rootRealPath): paths.append(os.path.join(root, tool)) break return paths[0] def run(self): binPath = self.getPath("taosBenchmark") if (binPath == ""): tdLog.exit("taosBenchmark not found!") else: tdLog.info("taosBenchmark found in %s" % binPath) #-N:regular table -d:database name -t:table num -n:rows num per table -l:col num -y:force # regular old && new startTime = time.time() os.system("%s -N -d regular_old -t 1 -n 10 -l 1023 -y" % binPath) tdSql.execute("use regular_old") tdSql.query("show tables;") tdSql.checkRows(1) tdSql.query("select * from d0;") tdSql.checkCols(1024) tdSql.query("describe d0;") tdSql.checkRows(1024) os.system("%s -N -d regular_new -t 1 -n 10 -l 4095 -y" % binPath) tdSql.execute("use regular_new") tdSql.query("show tables;") tdSql.checkRows(1) tdSql.query("select * from d0;") tdSql.checkCols(4096) tdSql.query("describe d0;") tdSql.checkRows(4096) # super table -d:database name -t:table num -n:rows num per table # -l:col num -y:force os.system("%s -d super_old -t 1 -n 10 -l 1021 -y" % binPath) tdSql.execute("use super_old") tdSql.query("show tables;") tdSql.checkRows(1) tdSql.query("select * from meters;") tdSql.checkCols(1024) tdSql.query("select * from d0;") tdSql.checkCols(1022) tdSql.query("describe meters;") tdSql.checkRows(1024) tdSql.query("describe d0;") tdSql.checkRows(1024) os.system("%s -d super_new -t 1 -n 10 -l 4093 -y" % binPath) tdSql.execute("use super_new") tdSql.query("show tables;") tdSql.checkRows(1) tdSql.query("select * from meters;") tdSql.checkCols(4096) tdSql.query("select * from d0;") tdSql.checkCols(4094) tdSql.query("describe meters;") tdSql.checkRows(4096) tdSql.query("describe d0;") tdSql.checkRows(4096) tdSql.execute("create table stb_new1_1 using meters tags(1,2)") tdSql.query("select * from stb_new1_1") tdSql.checkCols(4094) tdSql.query("describe stb_new1_1;") tdSql.checkRows(4096) # insert: create one or mutiple tables per sql and insert multiple rows per sql # test case for https://jira.taosdata.com:18080/browse/TD-5213 os.system( "%s -f tools/taosdemoAllTest/TD-5213/insertSigcolumnsNum4096.json -y " % binPath) tdSql.execute("use json_test") tdSql.query("select count (tbname) from stb_old") tdSql.checkData(0, 0, 1) tdSql.query("select * from stb_old") tdSql.checkRows(10) tdSql.checkCols(1024) tdSql.query("select count (tbname) from stb_new") tdSql.checkData(0, 0, 1) tdSql.query("select * from stb_new") tdSql.checkRows(10) tdSql.checkCols(4096) tdSql.query("describe stb_new;") tdSql.checkRows(4096) tdSql.query("select * from stb_new_0") tdSql.checkRows(10) tdSql.checkCols(4091) tdSql.query("describe stb_new_0;") tdSql.checkRows(4096) tdSql.execute("create table stb_new1_1 using stb_new tags(1,2,3,4,5)") tdSql.query("select * from stb_new1_1") tdSql.checkCols(4091) tdSql.query("describe stb_new1_1;") tdSql.checkRows(4096) tdSql.query("select count (tbname) from stb_mix") tdSql.checkData(0, 0, 1) tdSql.query("select * from stb_mix") tdSql.checkRows(10) tdSql.checkCols(4096) tdSql.query("describe stb_mix;") tdSql.checkRows(4096) tdSql.query("select * from stb_mix_0") tdSql.checkRows(10) tdSql.checkCols(4092) tdSql.query("describe stb_mix_0;") tdSql.checkRows(4096) tdSql.query("select count (tbname) from stb_excel") tdSql.checkData(0, 0, 1) tdSql.query("select * from stb_excel") tdSql.checkRows(10) tdSql.checkCols(4096) tdSql.query("describe stb_excel;") tdSql.checkRows(4096) tdSql.query("select * from stb_excel_0") tdSql.checkRows(10) tdSql.checkCols(4092) tdSql.query("describe stb_excel_0;") tdSql.checkRows(4096) endTime = time.time() print("total time %ds" % (endTime - startTime)) os.system( "rm -rf tools/taosdemoAllTest/TD-5213/insertSigcolumnsNum4096.py.sql") def stop(self): tdSql.close() tdLog.success("%s successfully executed" % __file__) tdCases.addWindows(__file__, TDTestCase()) tdCases.addLinux(__file__, TDTestCase())