diff --git a/tests/pytest/fulltest.sh b/tests/pytest/fulltest.sh index d712e04edf860ddef5b607bdced440ce3ec9def3..691675c89622926320023a0ff281d1dedcab7125 100755 --- a/tests/pytest/fulltest.sh +++ b/tests/pytest/fulltest.sh @@ -272,6 +272,7 @@ python3 ./test.py -f query/operator_cost.py python3 test.py -f query/nestedQuery/queryWithSpread.py python3 ./test.py -f query/bug6586.py # python3 ./test.py -f query/bug5903.py +python3 ./test.py -f query/queryLimit.py #stream python3 ./test.py -f stream/metric_1.py diff --git a/tests/pytest/query/queryLimit.py b/tests/pytest/query/queryLimit.py new file mode 100644 index 0000000000000000000000000000000000000000..b5984a43a406db874f0db0a308d040b4da02f09a --- /dev/null +++ b/tests/pytest/query/queryLimit.py @@ -0,0 +1,169 @@ +################################################################### +# 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 + +from numpy.lib.function_base import insert +import taos +from util.log import * +from util.cases import * +from util.sql import * +import numpy as np + +# constant define +WAITS = 5 # wait seconds + +class TDTestCase: + # + # --------------- main frame ------------------- + # + + def caseDescription(self): + ''' + limit and offset keyword function test cases; + case1: limit offset base function test + case2: limit offset advance test + ''' + return + + # init + def init(self, conn, logSql): + tdLog.debug("start to execute %s" % __file__) + tdSql.init(conn.cursor()) + tdSql.prepare() + self.create_tables(); + self.ts = 1500000000000 + + + # run case + def run(self): + # insert data + self.insert_data("t1", self.ts, 300*10000, 30000); + # test base case + self.test_case1() + tdLog.debug(" LIMIT test_case1 ............ [OK]") + # test advance case + self.test_case2() + tdLog.debug(" LIMIT test_case2 ............ [OK]") + + + # stop + def stop(self): + tdSql.close() + tdLog.success("%s successfully executed" % __file__) + + # + # --------------- case ------------------- + # + + # create table + def create_tables(self): + # super table + tdSql.execute("create table st(ts timestamp, i1 int) tags(area int)"); + # child table + tdSql.execute("create table t1 using st tags(1)"); + tdSql.execute("create table t2 using st tags(2)"); + tdSql.execute("create table t3 using st tags(3)"); + return + + # insert data1 + def insert_data(self, tbname, ts_start, count, batch_num): + pre_insert = "insert into %s values"%tbname + sql = pre_insert + tdLog.debug("doing insert table %s rows=%d ..."%(tbname, count)) + for i in range(count): + sql += " (%d,%d)"%(ts_start + i*1000, i) + if i >0 and i%batch_num == 0: + tdSql.execute(sql) + sql = pre_insert + # end sql + if sql != pre_insert: + tdSql.execute(sql) + + tdLog.debug("INSERT TABLE DATA ............ [OK]") + return + + # test case1 base + def test_case1(self): + # + # limit base function + # + # base no where + sql = "select * from t1 limit 10" + tdSql.waitedQuery(sql, 10, WAITS) + tdSql.checkData(0, 1, 0) + tdSql.checkData(9, 1, 9) + + # have where + sql = "select * from t1 where ts>='2017-07-14 10:40:01' and ts<'2017-07-14 10:40:06' limit 10" + tdSql.waitedQuery(sql, 5, WAITS) + tdSql.checkData(0, 1, 1) + tdSql.checkData(4, 1, 5) + + # + # offset base function + # + # no where + sql = "select * from t1 limit 10 offset 5" + tdSql.waitedQuery(sql, 10, WAITS) + tdSql.checkData(0, 1, 5) + tdSql.checkData(9, 1, 14) + + # have where only ts + sql = "select * from t1 where ts>='2017-07-14 10:40:10' and ts<'2017-07-14 10:40:20' limit 10 offset 5" + tdSql.waitedQuery(sql, 5, WAITS) + tdSql.checkData(0, 1, 15) + tdSql.checkData(4, 1, 19) + + # have where with other column condition + sql = "select * from t1 where i1>=1 and i1<11 limit 10 offset 5" + tdSql.waitedQuery(sql, 5, WAITS) + tdSql.checkData(0, 1, 6) + tdSql.checkData(4, 1, 10) + + # have where with ts and other column condition + sql = "select * from t1 where ts>='2017-07-14 10:40:10' and ts<'2017-07-14 10:40:50' and i1>=20 and i1<=25 limit 10 offset 5" + tdSql.waitedQuery(sql, 1, WAITS) + tdSql.checkData(0, 1, 25) + + return + + # test advance + def test_case2(self): + # + # OFFSET merge file data with memory data + # + + # offset + sql = "select * from t1 limit 10 offset 72000" + tdSql.waitedQuery(sql, 10, WAITS) + tdSql.checkData(0, 1, 72000) + + # each insert one row into NO.0 NO.2 NO.7 blocks + sql = "insert into t1 values (%d, 0) (%d, 2) (%d, 7)"%(self.ts+1, self.ts + 2*3300*1000+1, self.ts + 7*3300*1000+1) + tdSql.execute(sql) + # query result + sql = "select * from t1 limit 10 offset 72000" + tdSql.waitedQuery(sql, 10, WAITS) + tdSql.checkData(0, 1, 72000 - 3) + + # have where + sql = "select * from t1 where ts>='2017-07-14 10:40:10' and ts<'2017-07-22 18:40:10' limit 10 offset 72000" + tdSql.waitedQuery(sql, 10, WAITS) + tdSql.checkData(0, 1, 72000 - 3 + 10 + 1) + +# +# add case with filename +# +tdCases.addWindows(__file__, TDTestCase()) +tdCases.addLinux(__file__, TDTestCase()) \ No newline at end of file