timezone.py

tests/system-test/2-query/timezone.py

+from pytz import timezone
 import taos
 from util.log import *
 from util.sql import *
@@ -15,8 +16,9 @@ class TDTestCase:
     def run(self):  # sourcery skip: extract-duplicate-method
         tdSql.prepare()
         # get system timezone
-        time_zone = os.popen('timedatectl | grep zone').read().strip().split(':')[1].lstrip()
-        
+        time_zone = os.popen('timedatectl | grep zone').read(
+        ).strip().split(':')[1].lstrip()
+
         tdLog.printNoPrefix("==========step1:create tables==========")
         tdSql.execute(
             '''create table if not exists ntb
@@ -34,40 +36,47 @@ class TDTestCase:
         )
 
         tdLog.printNoPrefix("==========step2:insert data==========")
-        tdSql.execute("insert into ntb values(now,10,99.99,11.111111)(today(),100,11.111,22.222222)")
-        tdSql.execute("insert into stb_1 values(now,111,99.99,11.111111)(today(),1,11.111,22.222222)")
+        tdSql.execute(
+            "insert into ntb values(now,10,99.99,11.111111)(today(),100,11.111,22.222222)")
+        tdSql.execute(
+            "insert into stb_1 values(now,111,99.99,11.111111)(today(),1,11.111,22.222222)")
 
         tdLog.printNoPrefix("==========step3:query data==========")
-        for i in range(0,10):
+        for i in range(0, 10):
             tdSql.query("select timezone() from ntb")
             tdSql.checkRows(2)
-            tdSql.checkData(0,0,time_zone)
-            i+=1
-            
-        for i in range(0,10):
+            tdSql.checkData(0, 0, time_zone)
+            i += 1
+
+        for i in range(0, 10):
             tdSql.query("select timezone() from db.ntb")
             tdSql.checkRows(2)
-            tdSql.checkData(0,0,time_zone)
-            i+=1
+            tdSql.checkData(0, 0, time_zone)
+            i += 1
         tdSql.query("select timezone() from stb")
         tdSql.checkRows(2)
-        tdSql.checkData(0,0,time_zone)
+        tdSql.checkData(0, 0, time_zone)
         tdSql.query("select timezone() from db.stb")
         tdSql.checkRows(2)
-        tdSql.checkData(0,0,time_zone)
+        tdSql.checkData(0, 0, time_zone)
 
         tdSql.query("select timezone() from stb_1")
         tdSql.checkRows(2)
-        tdSql.checkData(0,0,time_zone)
+        tdSql.checkData(0, 0, time_zone)
         tdSql.query("select timezone() from db.stb_1 ")
         tdSql.checkRows(2)
-        tdSql.checkData(0,0,time_zone)
+        tdSql.checkData(0, 0, time_zone)
 
+        tdSql.error("select timezone(1) from ntb")
+        tdSql.error("select timezone(now()) from stb")
 
+        tdSql.query(f"select * from ntb where timezone()='{time_zone}'")
+        tdSql.checkRows(2)
 
     def stop(self):
         tdSql.close()
         tdLog.success(f"{__file__} successfully executed")
 
+
 tdCases.addLinux(__file__, TDTestCase())
 tdCases.addWindows(__file__, TDTestCase())