mirror of
https://github.com/taosdata/TDengine
synced 2026-05-24 10:09:01 +00:00
306 lines
15 KiB
Python
306 lines
15 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 -*-
|
|
from new_test_framework.utils import tdLog, tdSql, etool, tdCom
|
|
import os
|
|
|
|
class TestVTableQuerySameDB:
|
|
|
|
def setup_class(cls):
|
|
tdLog.info(f"prepare org tables.")
|
|
|
|
tdSql.execute("drop database if exists test_vtable_select;")
|
|
tdSql.execute("create database test_vtable_select;")
|
|
tdSql.execute("use test_vtable_select;")
|
|
|
|
tdLog.info(f"prepare org super table.")
|
|
tdSql.execute("select database();")
|
|
tdSql.execute(f"CREATE STABLE `vtb_org_stb` ("
|
|
"ts timestamp, "
|
|
"u_tinyint_col tinyint unsigned, "
|
|
"u_smallint_col smallint unsigned, "
|
|
"u_int_col int unsigned, "
|
|
"u_bigint_col bigint unsigned, "
|
|
"tinyint_col tinyint, "
|
|
"smallint_col smallint, "
|
|
"int_col int, "
|
|
"bigint_col bigint, "
|
|
"float_col float, "
|
|
"double_col double, "
|
|
"bool_col bool, "
|
|
"binary_16_col binary(16),"
|
|
"binary_32_col binary(32),"
|
|
"nchar_16_col nchar(16),"
|
|
"nchar_32_col nchar(32)"
|
|
") TAGS ("
|
|
"int_tag int,"
|
|
"bool_tag bool,"
|
|
"float_tag float,"
|
|
"double_tag double,"
|
|
"nchar_32_tag nchar(32),"
|
|
"binary_32_tag binary(32))")
|
|
|
|
tdLog.info(f"prepare org child table.")
|
|
for i in range(15):
|
|
tdSql.execute(f"CREATE TABLE `vtb_org_child_{i}` USING `vtb_org_stb` TAGS ({i}, false, {i}, {i}, 'child{i}', 'child{i}');")
|
|
|
|
tdLog.info(f"prepare org normal table.")
|
|
for i in range(15):
|
|
tdSql.execute(f"CREATE TABLE `vtb_org_normal_{i}` (ts timestamp, u_tinyint_col tinyint unsigned, u_smallint_col smallint unsigned, u_int_col int unsigned, u_bigint_col bigint unsigned, tinyint_col tinyint, smallint_col smallint, int_col int, bigint_col bigint, float_col float, double_col double, bool_col bool, binary_16_col binary(16), binary_32_col binary(32), nchar_16_col nchar(16), nchar_32_col nchar(32)) SMA(u_tinyint_col)")
|
|
|
|
tdLog.info(f"insert data into org tables.")
|
|
datafile = etool.getFilePath(__file__, "data", "data1.csv")
|
|
tdSql.execute("insert into vtb_org_normal_0 file '%s';" % datafile)
|
|
tdSql.execute("insert into vtb_org_child_0 file '%s';" % datafile)
|
|
|
|
datafile = etool.getFilePath(__file__, "data", "data2.csv")
|
|
tdSql.execute("insert into vtb_org_normal_1 file '%s';" % datafile)
|
|
tdSql.execute("insert into vtb_org_child_1 file '%s';" % datafile)
|
|
|
|
datafile = etool.getFilePath(__file__, "data", "data3.csv")
|
|
tdSql.execute("insert into vtb_org_normal_2 file '%s';" % datafile)
|
|
tdSql.execute("insert into vtb_org_child_2 file '%s';" % datafile)
|
|
|
|
tdLog.info(f"prepare virtual normal table.")
|
|
|
|
tdSql.execute(f"CREATE VTABLE `vtb_virtual_ntb_full` ("
|
|
"ts timestamp, "
|
|
"u_tinyint_col tinyint unsigned from vtb_org_normal_0.u_tinyint_col, "
|
|
"u_smallint_col smallint unsigned from vtb_org_normal_1.u_smallint_col, "
|
|
"u_int_col int unsigned from vtb_org_normal_2.u_int_col, "
|
|
"u_bigint_col bigint unsigned from vtb_org_normal_0.u_bigint_col, "
|
|
"tinyint_col tinyint from vtb_org_normal_1.tinyint_col, "
|
|
"smallint_col smallint from vtb_org_normal_2.smallint_col, "
|
|
"int_col int from vtb_org_normal_0.int_col, "
|
|
"bigint_col bigint from vtb_org_normal_1.bigint_col, "
|
|
"float_col float from vtb_org_normal_2.float_col, "
|
|
"double_col double from vtb_org_normal_0.double_col, "
|
|
"bool_col bool from vtb_org_normal_1.bool_col, "
|
|
"binary_16_col binary(16) from vtb_org_normal_2.binary_16_col,"
|
|
"binary_32_col binary(32) from vtb_org_normal_0.binary_32_col,"
|
|
"nchar_16_col nchar(16) from vtb_org_normal_1.nchar_16_col,"
|
|
"nchar_32_col nchar(32) from vtb_org_normal_2.nchar_32_col)")
|
|
|
|
tdSql.execute(f"CREATE VTABLE `vtb_virtual_ntb_half_full` ("
|
|
"ts timestamp, "
|
|
"u_tinyint_col tinyint unsigned from vtb_org_normal_0.u_tinyint_col, "
|
|
"u_smallint_col smallint unsigned from vtb_org_normal_1.u_smallint_col, "
|
|
"u_int_col int unsigned from vtb_org_normal_2.u_int_col, "
|
|
"u_bigint_col bigint unsigned, "
|
|
"tinyint_col tinyint, "
|
|
"smallint_col smallint, "
|
|
"int_col int from vtb_org_normal_0.int_col, "
|
|
"bigint_col bigint from vtb_org_normal_1.bigint_col, "
|
|
"float_col float from vtb_org_normal_2.float_col, "
|
|
"double_col double, "
|
|
"bool_col bool, "
|
|
"binary_16_col binary(16),"
|
|
"binary_32_col binary(32) from vtb_org_normal_0.binary_32_col,"
|
|
"nchar_16_col nchar(16) from vtb_org_normal_1.nchar_16_col,"
|
|
"nchar_32_col nchar(32) from vtb_org_normal_2.nchar_32_col)")
|
|
|
|
tdSql.execute(f"CREATE VTABLE `vtb_virtual_ntb_empty` ("
|
|
"ts timestamp, "
|
|
"u_tinyint_col tinyint unsigned, "
|
|
"u_smallint_col smallint unsigned, "
|
|
"u_int_col int unsigned, "
|
|
"u_bigint_col bigint unsigned, "
|
|
"tinyint_col tinyint, "
|
|
"smallint_col smallint, "
|
|
"int_col int, "
|
|
"bigint_col bigint, "
|
|
"float_col float, "
|
|
"double_col double, "
|
|
"bool_col bool, "
|
|
"binary_16_col binary(16),"
|
|
"binary_32_col binary(32),"
|
|
"nchar_16_col nchar(16),"
|
|
"nchar_32_col nchar(32))")
|
|
|
|
tdSql.execute(f"CREATE STABLE `vtb_virtual_stb` ("
|
|
"ts timestamp, "
|
|
"u_tinyint_col tinyint unsigned, "
|
|
"u_smallint_col smallint unsigned, "
|
|
"u_int_col int unsigned, "
|
|
"u_bigint_col bigint unsigned, "
|
|
"tinyint_col tinyint, "
|
|
"smallint_col smallint, "
|
|
"int_col int, "
|
|
"bigint_col bigint, "
|
|
"float_col float, "
|
|
"double_col double, "
|
|
"bool_col bool, "
|
|
"binary_16_col binary(16),"
|
|
"binary_32_col binary(32),"
|
|
"nchar_16_col nchar(16),"
|
|
"nchar_32_col nchar(32)"
|
|
") TAGS ("
|
|
"int_tag int,"
|
|
"bool_tag bool,"
|
|
"float_tag float,"
|
|
"double_tag double,"
|
|
"nchar_32_tag nchar(32),"
|
|
"binary_32_tag binary(32))"
|
|
"VIRTUAL 1")
|
|
|
|
tdLog.info(f"prepare virtual child table.")
|
|
|
|
tdSql.execute(f"CREATE VTABLE `vtb_virtual_ctb_full` ("
|
|
"u_tinyint_col from vtb_org_normal_0.u_tinyint_col, "
|
|
"u_smallint_col from vtb_org_normal_1.u_smallint_col, "
|
|
"u_int_col from vtb_org_normal_2.u_int_col, "
|
|
"u_bigint_col from vtb_org_normal_0.u_bigint_col, "
|
|
"tinyint_col from vtb_org_normal_1.tinyint_col, "
|
|
"smallint_col from vtb_org_normal_2.smallint_col, "
|
|
"int_col from vtb_org_normal_0.int_col, "
|
|
"bigint_col from vtb_org_normal_1.bigint_col, "
|
|
"float_col from vtb_org_normal_2.float_col, "
|
|
"double_col from vtb_org_normal_0.double_col, "
|
|
"bool_col from vtb_org_normal_1.bool_col, "
|
|
"binary_16_col from vtb_org_normal_2.binary_16_col,"
|
|
"binary_32_col from vtb_org_normal_0.binary_32_col,"
|
|
"nchar_16_col from vtb_org_normal_1.nchar_16_col,"
|
|
"nchar_32_col from vtb_org_normal_2.nchar_32_col)"
|
|
"USING `vtb_virtual_stb` TAGS (0, false, 0, 0, 'child0', 'child0')")
|
|
|
|
tdSql.execute(f"CREATE VTABLE `vtb_virtual_ctb_half_full` ("
|
|
"u_tinyint_col from vtb_org_normal_0.u_tinyint_col, "
|
|
"u_smallint_col from vtb_org_normal_1.u_smallint_col, "
|
|
"u_int_col from vtb_org_normal_2.u_int_col, "
|
|
"int_col from vtb_org_normal_0.int_col, "
|
|
"bigint_col from vtb_org_normal_1.bigint_col, "
|
|
"float_col from vtb_org_normal_2.float_col, "
|
|
"binary_32_col from vtb_org_normal_0.binary_32_col,"
|
|
"nchar_16_col from vtb_org_normal_1.nchar_16_col,"
|
|
"nchar_32_col from vtb_org_normal_2.nchar_32_col)"
|
|
"USING `vtb_virtual_stb` TAGS (1, false, 1, 1, 'child1', 'child1')")
|
|
|
|
tdSql.execute(f"CREATE VTABLE `vtb_virtual_ctb_empty` "
|
|
"USING `vtb_virtual_stb` TAGS (2, false, 2, 2, 'child2', 'child2')")
|
|
|
|
tdSql.execute(f"CREATE VTABLE `vtb_virtual_ctb_mix` ("
|
|
f"u_tinyint_col from vtb_org_child_0.u_tinyint_col, "
|
|
f"u_smallint_col from vtb_org_child_1.u_smallint_col, "
|
|
f"u_int_col from vtb_org_child_2.u_int_col, "
|
|
f"u_bigint_col from vtb_org_child_0.u_bigint_col, "
|
|
f"tinyint_col from vtb_org_child_1.tinyint_col, "
|
|
f"smallint_col from vtb_org_child_2.smallint_col, "
|
|
f"int_col from vtb_org_child_0.int_col, "
|
|
f"bigint_col from vtb_org_child_1.bigint_col, "
|
|
f"float_col from vtb_org_child_2.float_col, "
|
|
f"double_col from vtb_org_child_0.double_col, "
|
|
f"bool_col from vtb_org_child_1.bool_col, "
|
|
f"binary_16_col from vtb_org_child_2.binary_16_col,"
|
|
f"binary_32_col from vtb_org_child_0.binary_32_col,"
|
|
f"nchar_16_col from vtb_org_child_1.nchar_16_col,"
|
|
f"nchar_32_col from vtb_org_child_2.nchar_32_col)"
|
|
f"USING `vtb_virtual_stb` TAGS (3, false, 3, 3, 'child3', 'child3')")
|
|
|
|
def run_normal_query(self, testCase):
|
|
# read sql from .sql file and execute
|
|
tdLog.info(f"test case : {testCase}.")
|
|
self.sqlFile = os.path.join(os.path.dirname(__file__), "in", f"{testCase}.in")
|
|
self.ansFile = os.path.join(os.path.dirname(__file__), "ans", f"{testCase}.ans")
|
|
|
|
tdCom.compare_testcase_result(self.sqlFile, self.ansFile, testCase)
|
|
|
|
|
|
def test_select_virtual_normal_table(self):
|
|
"""Query: virtual normal table
|
|
|
|
1. test vstable select normal table projection
|
|
2. test vstable select normal table projection filter
|
|
3. test vstable select normal table projection timerange filter
|
|
4. test vstable select normal table interval
|
|
5. test vstable select normal table state
|
|
6. test vstable select normal table session
|
|
7. test vstable select normal table event
|
|
8. test vstable select normal table count
|
|
9. test vstable select normal table partition
|
|
10. test vstable select normal table group
|
|
11. test vstable select normal table orderby
|
|
|
|
Catalog:
|
|
- VirtualTable
|
|
|
|
Since: v3.3.6.0
|
|
|
|
Labels: virtual
|
|
|
|
Jira: None
|
|
|
|
History:
|
|
- 2025-3-15 Jing Sima Created
|
|
- 2025-5-6 Huo Hong Migrated to new test framework
|
|
- 2025-10-17 Jing Sima Add timerange filter test case
|
|
|
|
"""
|
|
self.run_normal_query("test_vtable_select_test_projection")
|
|
self.run_normal_query("test_vtable_select_test_projection_filter")
|
|
self.run_normal_query("test_vtable_select_test_projection_timerange_filter")
|
|
self.run_normal_query("test_vtable_select_test_function")
|
|
|
|
self.run_normal_query("test_vtable_select_test_interval")
|
|
self.run_normal_query("test_vtable_select_test_state")
|
|
self.run_normal_query("test_vtable_select_test_session")
|
|
self.run_normal_query("test_vtable_select_test_event")
|
|
self.run_normal_query("test_vtable_select_test_count")
|
|
|
|
self.run_normal_query("test_vtable_select_test_partition")
|
|
self.run_normal_query("test_vtable_select_test_group")
|
|
self.run_normal_query("test_vtable_select_test_orderby")
|
|
|
|
def test_select_virtual_child_table(self):
|
|
"""Query: virtual child table
|
|
|
|
1. test vstable select child table projection
|
|
2. test vstable select child table projection filter
|
|
3. test vstable select child table projection timerange filter
|
|
4. test vstable select child table interval
|
|
5. test vstable select child table state
|
|
6. test vstable select child table session
|
|
7. test vstable select child table event
|
|
8. test vstable select child table count
|
|
9. test vstable select child table partition
|
|
10. test vstable select child table group
|
|
11. test vstable select child table orderby
|
|
|
|
Catalog:
|
|
- VirtualTable
|
|
|
|
Since: v3.3.6.0
|
|
|
|
Labels: virtual
|
|
|
|
Jira: None
|
|
|
|
History:
|
|
- 2025-3-15 Jing Sima Created
|
|
- 2025-5-6 Huo Hong Migrated to new test framework
|
|
- 2025-10-17 Jing Sima Add timerange filter test case
|
|
|
|
"""
|
|
self.run_normal_query("test_vctable_select_test_projection")
|
|
self.run_normal_query("test_vctable_select_test_projection_filter")
|
|
self.run_normal_query("test_vctable_select_test_projection_timerange_filter")
|
|
self.run_normal_query("test_vctable_select_test_function")
|
|
|
|
self.run_normal_query("test_vctable_select_test_interval")
|
|
self.run_normal_query("test_vctable_select_test_state")
|
|
self.run_normal_query("test_vctable_select_test_session")
|
|
self.run_normal_query("test_vctable_select_test_event")
|
|
self.run_normal_query("test_vctable_select_test_count")
|
|
|
|
self.run_normal_query("test_vctable_select_test_partition")
|
|
self.run_normal_query("test_vctable_select_test_group")
|
|
self.run_normal_query("test_vctable_select_test_orderby")
|
|
|