timezone.py

from tarfile import SYMTYPE
from time import sleep
from tkinter import E
import taos
import sys
import time
from tzlocal import get_localzone
from util.log import *
from util.sql import *
from util.cases import *

import os


class TDTestCase:

    def init(self, conn, logSql):
        tdLog.debug(f"start to excute {__file__}")
        tdSql.init(conn.cursor())

    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()
        
        tdLog.printNoPrefix("==========step1:create tables==========")
        tdSql.execute(
            '''create table if not exists ntb
            (ts timestamp, c1 int, c2 float,c3 double)
            '''
        )
        tdSql.execute(
            '''create table if not exists stb
            (ts timestamp, c1 int, c2 float,c3 double) tags(t0 int)
            '''
        )
        tdSql.execute(
            '''create table if not exists stb_1 using stb tags(100)
            '''
        )

        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)")

        tdLog.printNoPrefix("==========step3:query data==========")
        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.query("select timezone() from db.ntb")
            tdSql.checkRows(2)
            tdSql.checkData(0,0,time_zone)
            i+=1
        tdSql.query("select timezone() from stb")
        tdSql.checkRows(2)
        tdSql.checkData(0,0,time_zone)
        tdSql.query("select timezone() from db.stb")
        tdSql.checkRows(2)
        tdSql.checkData(0,0,time_zone)

        tdSql.query("select timezone() from stb_1")
        tdSql.checkRows(2)
        tdSql.checkData(0,0,time_zone)
        tdSql.query("select timezone() from db.stb_1 ")
        tdSql.checkRows(2)
        tdSql.checkData(0,0,time_zone)



    def stop(self):
        tdSql.close()
        tdLog.success(f"{__file__} successfully executed")

tdCases.addLinux(__file__, TDTestCase())
tdCases.addWindows(__file__, TDTestCase())