194 lines
8.0 KiB
Python
194 lines
8.0 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 sys
|
||
import os
|
||
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)
|
||
|
||
self.ts = 1538548685000
|
||
self.numberOfTables = 10000
|
||
self.numberOfRecords = 100
|
||
|
||
def checkCommunity(self):
|
||
selfPath = os.path.dirname(os.path.realpath(__file__))
|
||
if ("community" in selfPath):
|
||
return False
|
||
else:
|
||
return True
|
||
|
||
def getPath(self, tool="taosdump"):
|
||
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
|
||
if (len(paths) == 0):
|
||
return ""
|
||
return paths[0]
|
||
|
||
def run(self):
|
||
if not os.path.exists("./taosdumptest/tmp1"):
|
||
os.makedirs("./taosdumptest/tmp1")
|
||
else:
|
||
os.system("rm -rf ./taosdumptest/tmp1")
|
||
os.makedirs("./taosdumptest/tmp1")
|
||
|
||
if not os.path.exists("./taosdumptest/tmp2"):
|
||
os.makedirs("./taosdumptest/tmp2")
|
||
else:
|
||
os.system("rm -rf ./taosdumptest/tmp2")
|
||
os.makedirs("./taosdumptest/tmp2")
|
||
|
||
tdSql.execute("drop database if exists db")
|
||
tdSql.execute("create database db duration 11 keep 3649")
|
||
tdSql.execute("create database db1 duration 12 keep 3640")
|
||
tdSql.execute("use db")
|
||
tdSql.execute(
|
||
"create table st(ts timestamp, c1 int, c2 nchar(10)) tags(t1 int, t2 binary(10))")
|
||
tdSql.execute("create table t1 using st tags(1, 'beijing')")
|
||
sql = "insert into t1 values"
|
||
currts = self.ts
|
||
for i in range(100):
|
||
sql += "(%d, %d, 'nchar%d')" % (currts + i, i % 100, i % 100)
|
||
tdSql.execute(sql)
|
||
tdSql.execute("create table t2 using st tags(2, 'shanghai')")
|
||
sql = "insert into t2 values"
|
||
currts = self.ts
|
||
for i in range(100):
|
||
sql += "(%d, %d, 'nchar%d')" % (currts + i, i % 100, i % 100)
|
||
tdSql.execute(sql)
|
||
|
||
binPath = self.getPath()
|
||
if (binPath == ""):
|
||
tdLog.exit("taosdump not found!")
|
||
else:
|
||
tdLog.info("taosdump found: %s" % binPath)
|
||
|
||
os.system("%s -y --databases db -o ./taosdumptest/tmp1" % binPath)
|
||
os.system(
|
||
"%s -y --databases db1 -o ./taosdumptest/tmp2" %
|
||
binPath)
|
||
|
||
tdSql.execute("drop database db")
|
||
tdSql.execute("drop database db1")
|
||
tdSql.query("select * from information_schema.ins_databases")
|
||
tdSql.checkRows(2)
|
||
|
||
os.system("%s -i ./taosdumptest/tmp1" % binPath)
|
||
os.system("%s -i ./taosdumptest/tmp2" % binPath)
|
||
|
||
tdSql.execute("use db")
|
||
tdSql.query("select * from information_schema.ins_databases")
|
||
tdSql.checkRows(4)
|
||
dbresult = tdSql.queryResult
|
||
# 6--duration,7--keep0,keep1,keep
|
||
|
||
isCommunity = self.checkCommunity()
|
||
print("iscommunity: %d" % isCommunity)
|
||
for i in range(len(dbresult)):
|
||
if dbresult[i][0] == 'db':
|
||
print(dbresult[i])
|
||
print(type(dbresult[i][6]))
|
||
print(type(dbresult[i][7]))
|
||
print((dbresult[i][6]))
|
||
assert dbresult[i][6] == "15840m"
|
||
print((dbresult[i][7]))
|
||
assert dbresult[i][7] == "5254560m,5254560m,5254560m"
|
||
if dbresult[i][0] == 'db1':
|
||
print((dbresult[i][6]))
|
||
assert dbresult[i][6] == "17280m"
|
||
print((dbresult[i][7]))
|
||
assert dbresult[i][7] == "5241600m,5241600m,5241600m"
|
||
|
||
tdSql.query("show stables")
|
||
tdSql.checkRows(1)
|
||
tdSql.checkData(0, 0, 'st')
|
||
|
||
tdSql.query("show tables")
|
||
tdSql.checkRows(2)
|
||
dbresult = tdSql.queryResult
|
||
print(dbresult)
|
||
for i in range(len(dbresult)):
|
||
assert ((dbresult[i][0] == "t1") or (dbresult[i][0] == "t2"))
|
||
|
||
tdSql.query("select * from t1")
|
||
tdSql.checkRows(100)
|
||
for i in range(100):
|
||
tdSql.checkData(i, 1, i)
|
||
tdSql.checkData(i, 2, "nchar%d" % i)
|
||
|
||
tdSql.query("select * from t2")
|
||
tdSql.checkRows(100)
|
||
for i in range(100):
|
||
tdSql.checkData(i, 1, i)
|
||
tdSql.checkData(i, 2, "nchar%d" % i)
|
||
|
||
# drop all databases，boundary value testing.
|
||
# length(databasename)<=32;length(tablesname)<=192
|
||
tdSql.execute("drop database db")
|
||
tdSql.execute("drop database db1")
|
||
os.system("rm -rf ./taosdumptest/tmp1")
|
||
os.system("rm -rf ./taosdumptest/tmp2")
|
||
os.makedirs("./taosdumptest/tmp1")
|
||
tdSql.execute("create database db12312313231231321312312312_323")
|
||
tdSql.error("create database db012345678911234567892234567893323456789423456789523456789bcdefe")
|
||
tdSql.execute("use db12312313231231321312312312_323")
|
||
tdSql.execute("create stable st12345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678_9(ts timestamp, c1 int, c2 nchar(10)) tags(t1 int, t2 binary(10))")
|
||
tdSql.error("create stable st_12345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678_9(ts timestamp, c1 int, c2 nchar(10)) tags(t1 int, t2 binary(10))")
|
||
tdSql.execute(
|
||
"create stable st(ts timestamp, c1 int, c2 nchar(10)) tags(t1 int, t2 binary(10))")
|
||
tdSql.error("create stable st1(ts timestamp, c1 int, col2_012345678901234567890123456789012345678901234567890123456789 nchar(10)) tags(t1 int, t2 binary(10))")
|
||
|
||
tdSql.execute("select * from db12312313231231321312312312_323.st12345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678_9")
|
||
tdSql.error("create table t0_12345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678912345678_9 using st tags(1, 'beijing')")
|
||
tdSql.query("show stables")
|
||
tdSql.checkRows(2)
|
||
os.system(
|
||
"%s -y --databases db12312313231231321312312312_323 -o ./taosdumptest/tmp1" %
|
||
binPath)
|
||
tdSql.execute("drop database db12312313231231321312312312_323")
|
||
os.system("%s -i ./taosdumptest/tmp1" % binPath)
|
||
tdSql.execute("use db12312313231231321312312312_323")
|
||
tdSql.query("show stables")
|
||
tdSql.checkRows(2)
|
||
os.system("rm -rf ./taosdumptest/tmp1")
|
||
os.system("rm -rf ./taosdumptest/tmp2")
|
||
os.system("rm -rf ./dump_result.txt")
|
||
os.system("rm -rf ./db.csv")
|
||
|
||
def stop(self):
|
||
tdSql.close()
|
||
tdLog.success("%s successfully executed" % __file__)
|
||
|
||
|
||
tdCases.addWindows(__file__, TDTestCase())
|
||
tdCases.addLinux(__file__, TDTestCase())
|