import taos import sys import datetime import inspect from util.log import * from util.sql import * from util.cases import * import random class TDTestCase: updatecfgDict = {"countAlwaysReturnValue":0} def init(self, conn, logSql, replicaVar=1): tdLog.debug(f"start to excute {__file__}") tdSql.init(conn.cursor(), False) def prepare_data(self, dbname="db"): tdSql.execute( f"create database if not exists {dbname} keep 3650 duration 1000") tdSql.execute(f"use {dbname} ") tdSql.execute( f"create table {dbname}.tb (ts timestamp, c0 int)" ) tdSql.execute( f"create table {dbname}.stb (ts timestamp, c0 int) tags (t0 int)" ) tdSql.execute( f"create table {dbname}.ctb1 using {dbname}.stb tags (1)" ) tdSql.execute( f"create table {dbname}.ctb2 using {dbname}.stb tags (2)" ) tdSql.execute( f"insert into {dbname}.tb values (now(), NULL)") tdSql.execute( f"insert into {dbname}.ctb1 values (now(), NULL)") tdSql.execute( f"insert into {dbname}.ctb2 values (now() + 1s, NULL)") def test_results(self, dbname="db"): # count tdSql.query(f"select count(c0) from {dbname}.tb") tdSql.checkRows(0) tdSql.query(f"select count(NULL) from {dbname}.tb") tdSql.checkRows(0) tdSql.query(f"select c0,count(c0) from {dbname}.tb group by c0") tdSql.checkRows(1) tdSql.checkData(0, 0, None) tdSql.query(f"select count(c0) from {dbname}.stb") tdSql.checkRows(0) tdSql.query(f"select count(NULL) from {dbname}.stb") tdSql.checkRows(0) tdSql.query(f"select c0,count(c0) from {dbname}.stb group by c0") tdSql.checkRows(1) tdSql.checkData(0, 0, None) tdSql.query(f"select count(NULL)") tdSql.checkRows(1) tdSql.checkData(0, 0, 0) # hyperloglog tdSql.query(f"select hyperloglog(c0) from {dbname}.tb") tdSql.checkRows(0) tdSql.query(f"select hyperloglog(NULL) from {dbname}.tb") tdSql.checkRows(0) tdSql.query(f"select c0,hyperloglog(c0) from {dbname}.tb group by c0") tdSql.checkRows(1) tdSql.checkData(0, 0, None) tdSql.query(f"select hyperloglog(c0) from {dbname}.stb") tdSql.checkRows(0) tdSql.query(f"select hyperloglog(NULL) from {dbname}.stb") tdSql.checkRows(0) tdSql.query(f"select c0,hyperloglog(c0) from {dbname}.stb group by c0") tdSql.checkRows(1) tdSql.checkData(0, 0, None) tdSql.query(f"select hyperloglog(NULL)") tdSql.checkRows(1) tdSql.checkData(0, 0, 0) def run(self): tdSql.prepare() tdLog.printNoPrefix("==========step1:prepare data ==============") self.prepare_data() tdLog.printNoPrefix("==========step2:test results ==============") self.test_results() def stop(self): tdSql.close() tdLog.success(f"{__file__} successfully executed") tdCases.addLinux(__file__, TDTestCase()) tdCases.addWindows(__file__, TDTestCase())