172 lines
7.4 KiB
Python
172 lines
7.4 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 string
|
|
import sys
|
|
import taos
|
|
from util.log import *
|
|
from util.cases import *
|
|
from util.sql import *
|
|
import numpy as np
|
|
|
|
|
|
class TDTestCase:
|
|
def init(self, conn, logSql):
|
|
tdLog.debug("start to execute %s" % __file__)
|
|
tdSql.init(conn.cursor())
|
|
|
|
self.rowNum = 10
|
|
self.tbnum = 20
|
|
self.ts = 1537146000000
|
|
self.binary_str = 'taosdata'
|
|
self.nchar_str = '涛思数据'
|
|
|
|
def get_long_name(self, length, mode="mixed"):
|
|
"""
|
|
generate long name
|
|
mode could be numbers/letters/letters_mixed/mixed
|
|
"""
|
|
if mode == "numbers":
|
|
population = string.digits
|
|
elif mode == "letters":
|
|
population = string.ascii_letters.lower()
|
|
elif mode == "letters_mixed":
|
|
population = string.ascii_letters.upper() + string.ascii_letters.lower()
|
|
else:
|
|
population = string.ascii_letters.lower() + string.digits
|
|
return "".join(random.choices(population, k=length))
|
|
def first_check_base(self):
|
|
tdSql.prepare()
|
|
|
|
tdSql.execute('''create table stb(ts timestamp, col1 tinyint, col2 smallint, col3 int, col4 bigint, col5 tinyint unsigned, col6 smallint unsigned,
|
|
col7 int unsigned, col8 bigint unsigned, col9 float, col10 double, col11 bool, col12 binary(20), col13 nchar(20)) tags(loc nchar(20))''')
|
|
tdSql.execute("create table stb_1 using stb tags('beijing')")
|
|
tdSql.execute("insert into stb_1(ts) values(%d)" % (self.ts - 1))
|
|
|
|
for i in ['stb_1','db.stb_1','stb_1','db.stb_1']:
|
|
tdSql.query(f"select first(*) from {i}")
|
|
tdSql.checkRows(1)
|
|
tdSql.checkData(0, 1, None)
|
|
#!bug TD-16561
|
|
# for i in ['stb','db.stb']:
|
|
# tdSql.query(f"select first(*) from {i}")
|
|
# tdSql.checkRows(1)
|
|
# tdSql.checkData(0, 1, None)
|
|
for i in range(1, 14):
|
|
for j in ['stb_1','db.stb_1','stb_1','db.stb_1']:
|
|
tdSql.query(f"select first(col{i}) from {j}")
|
|
tdSql.checkRows(0)
|
|
for i in range(self.rowNum):
|
|
tdSql.execute(f"insert into stb_1 values(%d, %d, %d, %d, %d, %d, %d, %d, %d, %f, %f, %d, '{self.binary_str}%d', '{self.nchar_str}%d')"
|
|
% (self.ts + i, i + 1, i + 1, i + 1, i + 1, i + 1, i + 1, i + 1, i + 1, i + 0.1, i + 0.1, i % 2, i + 1, i + 1))
|
|
for i in range(1, 14):
|
|
for j in ['stb_1', 'db.stb_1', 'stb', 'db.stb']:
|
|
tdSql.query(f"select first(col{i}) from {j}")
|
|
tdSql.checkRows(1)
|
|
# tinyint,smallint,int,bigint,tinyint unsigned,smallint unsigned,int unsigned,bigint unsigned
|
|
if i >=1 and i<9:
|
|
tdSql.checkData(0, 0, 1)
|
|
# float,double
|
|
elif i>=9 and i<11:
|
|
tdSql.checkData(0, 0, 0.1)
|
|
# bool
|
|
elif i == 11:
|
|
tdSql.checkData(0, 0, False)
|
|
# binary
|
|
elif i == 12:
|
|
tdSql.checkData(0, 0, f'{self.binary_str}1')
|
|
# nchar
|
|
elif i == 13:
|
|
tdSql.checkData(0, 0, f'{self.nchar_str}1')
|
|
#!bug TD-16569
|
|
tdSql.query("select first(*),last(*) from stb where ts < 23 interval(1s)")
|
|
tdSql.checkRows(0)
|
|
tdSql.execute('drop database db')
|
|
def first_check_stb_distribute(self):
|
|
# prepare data for vgroup 4
|
|
dbname = self.get_long_name(length=10, mode="letters")
|
|
stbname = self.get_long_name(length=5, mode="letters")
|
|
child_table_num = 20
|
|
tdSql.execute(f"create database if not exists {dbname} vgroups 4")
|
|
tdSql.execute(f'use {dbname}')
|
|
# build 20 child tables,every table insert 10 rows
|
|
tdSql.execute(f'''create table {stbname}(ts timestamp, col1 tinyint, col2 smallint, col3 int, col4 bigint, col5 tinyint unsigned, col6 smallint unsigned,
|
|
col7 int unsigned, col8 bigint unsigned, col9 float, col10 double, col11 bool, col12 binary(20), col13 nchar(20)) tags(loc nchar(20))''')
|
|
for i in range(child_table_num):
|
|
tdSql.execute(f"create table {stbname}_{i} using {stbname} tags('beijing')")
|
|
tdSql.execute(f"insert into {stbname}_{i}(ts) values(%d)" % (self.ts - 1-i))
|
|
#!bug TD-16561
|
|
# for i in [f'{stbname}', f'{dbname}.{stbname}']:
|
|
# tdSql.query(f"select first(*) from {i}")
|
|
# tdSql.checkRows(1)
|
|
# tdSql.checkData(0, 1, None)
|
|
tdSql.query('show tables')
|
|
vgroup_list = []
|
|
for i in range(len(tdSql.queryResult)):
|
|
vgroup_list.append(tdSql.queryResult[i][6])
|
|
vgroup_list_set = set(vgroup_list)
|
|
# print(vgroup_list_set)
|
|
# print(vgroup_list)
|
|
for i in vgroup_list_set:
|
|
vgroups_num = vgroup_list.count(i)
|
|
if vgroups_num >=2:
|
|
tdLog.info(f'This scene with {vgroups_num} vgroups is ok!')
|
|
continue
|
|
else:
|
|
tdLog.exit('This scene does not meet the requirements with {vgroups_num} vgroup!\n')
|
|
|
|
for i in range(child_table_num):
|
|
for j in range(self.rowNum):
|
|
tdSql.execute(f"insert into {stbname}_{i} values(%d, %d, %d, %d, %d, %d, %d, %d, %d, %f, %f, %d, '{self.binary_str}%d', '{self.nchar_str}%d')"
|
|
% (self.ts + j + i, j + 1, j + 1, j + 1, j + 1, j + 1, j + 1, j + 1, j + 1, j + 0.1, j + 0.1, j % 2, j + 1, j + 1))
|
|
|
|
for i in range(1, 14):
|
|
for j in [f'{stbname}_{i}', f'{dbname}.{stbname}_{i}', f'{stbname}', f'{dbname}.{stbname}']:
|
|
tdSql.query(f"select first(col{i}) from {j}")
|
|
tdSql.checkRows(1)
|
|
# tinyint,smallint,int,bigint,tinyint unsigned,smallint unsigned,int unsigned,bigint unsigned
|
|
if i >=1 and i<9:
|
|
tdSql.checkData(0, 0, 1)
|
|
# float,double
|
|
elif i>=9 and i<11:
|
|
tdSql.checkData(0, 0, 0.1)
|
|
# bool
|
|
elif i == 11:
|
|
tdSql.checkData(0, 0, False)
|
|
# binary
|
|
elif i == 12:
|
|
tdSql.checkData(0, 0, f'{self.binary_str}1')
|
|
# nchar
|
|
elif i == 13:
|
|
tdSql.checkData(0, 0, f'{self.nchar_str}1')
|
|
#!bug TD-16569
|
|
tdSql.query(f"select first(*),last(*) from {stbname} where ts < 23 interval(1s)")
|
|
tdSql.checkRows(0)
|
|
tdSql.execute(f'drop database {dbname}')
|
|
|
|
|
|
|
|
pass
|
|
def run(self):
|
|
self.first_check_base()
|
|
self.first_check_stb_distribute()
|
|
|
|
|
|
def stop(self):
|
|
tdSql.close()
|
|
tdLog.success("%s successfully executed" % __file__)
|
|
|
|
tdCases.addWindows(__file__, TDTestCase())
|
|
tdCases.addLinux(__file__, TDTestCase())
|