TDengine/test/cases/13-StreamProcessing/99-Others/test_dev_basic5.py
2025-05-28 15:14:37 +08:00

94 lines
3.2 KiB
Python

import time
import math
from new_test_framework.utils import tdLog, tdSql, tdStream
class TestStreamDevBasic:
def setup_class(cls):
tdLog.debug(f"start to execute {__file__}")
def test_stream_dev_basic(self):
"""basic test 3
Verification testing during the development process.
Catalog:
- Streams:Others
Since: v3.3.3.7
Labels: common,ci
Jira: None
History:
- 2025-5-26 Simon Guan Created
"""
self.basic1()
# self.basic2()
def basic1(self):
tdLog.info(f"basic test 1")
tdStream.dropAllStreamsAndDbs()
tdStream.createSnode()
tdLog.info(f"=============== create database")
tdSql.prepare(dbname="test", vgroups=1)
tdLog.info(f"=============== create super table")
tdSql.execute(
f"create stable stb (ts timestamp, id int, c1 int) tags(t1 int);"
)
tdSql.query(f"show stables")
tdSql.checkRows(1)
tdLog.info(f"=============== write query data")
sqls = [
"insert into t1 using stb tags(1) values ('2025-01-01 00:00:00' , 0, 0);",
"insert into t2 using stb tags(2) values ('2025-01-01 00:00:00.102', 1, 0);",
"insert into t1 using stb tags(1) values ('2025-01-01 00:00:01' , 1, 1);",
"insert into t2 using stb tags(2) values ('2025-01-01 00:00:01.400', 2, 1);",
"insert into t1 using stb tags(1) values ('2025-01-01 00:00:02' , 2, 2);",
"insert into t2 using stb tags(2) values ('2025-01-01 00:00:02.600', 3, 2);",
]
tdSql.executes(sqls)
tdSql.query("select _wstart, avg(id) from stb interval(1s)")
tdSql.printResult()
tdLog.info(f"=============== create trigger table")
tdSql.execute("create table stream_trigger (ts timestamp, id int, c1 int);")
tdSql.query(f"show tables")
tdSql.checkKeyExist("stream_trigger")
tdLog.info(f"=============== create stream")
tdSql.execute(
"create stream s5 state_window (id) from stream_trigger into out5 as select _twstart ts, count(*) c1, avg(id) c2, first(id) c3, last(id) c4 from %%trows;"
)
tdStream.checkStreamStatus()
tdLog.info(f"=============== write trigger data")
tdSql.execute(
"insert into stream_trigger values ('2025-01-01 00:00:00', 0, 0), ('2025-01-01 00:00:01', 1, 1), ('2025-01-01 00:00:02', 2, 2);"
)
tdLog.info(f"=============== check stream result")
result_sql = "select ts, c1, c2, c3, c4 from test.out5"
tdSql.checkResultsByFunc(
sql=result_sql,
func=lambda: tdSql.getRows() == 2
and tdSql.compareData(0, 0, "2025-01-01 00:00:00.000")
and tdSql.compareData(0, 1, 1)
and tdSql.compareData(0, 2, 0)
and tdSql.compareData(0, 3, 0)
and tdSql.compareData(0, 4, 0)
and tdSql.compareData(1, 0, "2025-01-01 00:00:01.000")
and tdSql.compareData(1, 1, 1)
and tdSql.compareData(1, 2, 1)
and tdSql.compareData(1, 3, 1)
and tdSql.compareData(1, 4, 1),
)