261 lines
10 KiB
Python
261 lines
10 KiB
Python
#!/usr/bin/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
|
||
#
|
||
###################################################################
|
||
# install pip
|
||
# pip install src/connector/python/
|
||
|
||
# -*- coding: utf-8 -*-
|
||
import sys, os
|
||
import re
|
||
import platform
|
||
import getopt
|
||
import subprocess
|
||
# from this import d
|
||
import time
|
||
|
||
# input for server
|
||
|
||
opts, args = getopt.gnu_getopt(sys.argv[1:], 'v:m:u', ['version=', 'verMode='])
|
||
serverHost = ""
|
||
serverPort = 0
|
||
version = ""
|
||
uninstall = False
|
||
verMode = ""
|
||
for key, value in opts:
|
||
if key in ['--help']:
|
||
print('A collection of test cases written using Python')
|
||
print('-v test client version')
|
||
print('-u test uninstall process, will uninstall TDengine')
|
||
sys.exit(0)
|
||
|
||
if key in ['-v']:
|
||
version = value
|
||
if key in ['-u']:
|
||
uninstall = True
|
||
if key in ['-m']:
|
||
verMode = value
|
||
if not version:
|
||
print("No version specified, will not run version check.")
|
||
|
||
|
||
system = platform.system()
|
||
arch = platform.machine()
|
||
|
||
databaseName = re.sub(r'[^a-zA-Z0-9]', '', subprocess.getoutput("hostname")).lower()
|
||
# install taospy
|
||
taospy_version = ""
|
||
if system == 'Windows':
|
||
taospy_version = subprocess.getoutput("pip3 show taospy|findstr Version")
|
||
else:
|
||
taospy_version = subprocess.getoutput("pip3 show taospy|grep Version| awk -F ':' '{print $2}' ")
|
||
|
||
print("taospy version %s " % taospy_version)
|
||
if taospy_version == "":
|
||
subprocess.getoutput("pip3 install git+https://github.com/taosdata/taos-connector-python.git")
|
||
print("install taos python connector")
|
||
else:
|
||
subprocess.getoutput("pip3 install taospy")
|
||
|
||
# start taosd server
|
||
if system == 'Windows':
|
||
cmd = ["C:\\TDengine\\start-all.bat"]
|
||
elif system == 'Linux':
|
||
cmd = "systemctl start taosd".split(' ')
|
||
else:
|
||
cmd = "sudo launchctl start com.tdengine.taosd".split(' ')
|
||
process_out = subprocess.Popen(cmd,
|
||
stdin=subprocess.PIPE, stdout=subprocess.PIPE, text=True)
|
||
print(cmd)
|
||
time.sleep(5)
|
||
|
||
#get taosc version info
|
||
version_test_result = False
|
||
if version:
|
||
import taos
|
||
conn = taos.connect()
|
||
server_version = conn.server_info
|
||
print("server_version", server_version)
|
||
client_version = conn.client_info
|
||
print("client_version", client_version)
|
||
# Execute sql get version info
|
||
result: taos.TaosResult = conn.query("SELECT server_version()")
|
||
select_server = result.fetch_all()[0][0]
|
||
print("SELECT server_version():" + select_server)
|
||
result: taos.TaosResult = conn.query("SELECT client_version()")
|
||
select_client = result.fetch_all()[0][0]
|
||
print("SELECT client_version():" + select_client)
|
||
conn.close()
|
||
|
||
taos_V_output = ""
|
||
taosd_V_output = ""
|
||
taosadapter_V_output = ""
|
||
taoskeeper_V_output = ""
|
||
taosx_V_output = ""
|
||
taosB_V_output = ""
|
||
taosxVersion = False
|
||
if system == "Windows":
|
||
taos_V_output = subprocess.getoutput("taos -V | findstr version")
|
||
taosd_V_output = subprocess.getoutput("taosd -V | findstr version")
|
||
taosadapter_V_output = subprocess.getoutput("taosadapter -V | findstr version")
|
||
taoskeeper_V_output = subprocess.getoutput("taoskeeper -V | findstr version")
|
||
taosB_V_output = subprocess.getoutput("taosBenchmark -V | findstr version")
|
||
if verMode == "Enterprise":
|
||
taosx_V_output = subprocess.getoutput("taosx -V | findstr version")
|
||
else:
|
||
taos_V_output = subprocess.getoutput("taos -V | grep version | awk -F ' ' '{print $3}'")
|
||
taosd_V_output = subprocess.getoutput("taosd -V | grep version | awk -F ' ' '{print $3}'")
|
||
taosadapter_V_output = subprocess.getoutput("taosadapter -V | grep version | awk -F ' ' '{print $3}'")
|
||
taoskeeper_V_output = subprocess.getoutput("taoskeeper -V | grep version | awk -F ' ' '{print $3}'")
|
||
taosB_V_output = subprocess.getoutput("taosBenchmark -V | grep version | awk -F ' ' '{print $3}'")
|
||
if verMode == "Enterprise":
|
||
taosx_V_output = subprocess.getoutput("taosx -V | grep version | awk -F ' ' '{print $3}'")
|
||
|
||
print("taos -V output is: %s" % taos_V_output)
|
||
print("taosd -V output is: %s" % taosd_V_output)
|
||
print("taosadapter -V output is: %s" % taosadapter_V_output)
|
||
print("taoskeeper -V output is: %s" % taoskeeper_V_output)
|
||
print("taosBenchmark -V output is: %s" % taosB_V_output)
|
||
if verMode == "Enterprise":
|
||
print("taosx -V output is: %s" % taosx_V_output)
|
||
taosxVersion = version in taosx_V_output
|
||
else:
|
||
taosxVersion = True
|
||
if (version in client_version
|
||
and version in server_version
|
||
and version in select_server
|
||
and version in select_client
|
||
and version in taos_V_output
|
||
and version in taosd_V_output
|
||
and version in taosadapter_V_output
|
||
and version in taoskeeper_V_output
|
||
and version in taosB_V_output
|
||
and taosxVersion
|
||
):
|
||
version_test_result = True
|
||
leftFile = False
|
||
if uninstall:
|
||
print("Start to run rmtaos")
|
||
print("Platform: ", system)
|
||
# stop taosd server
|
||
if system == 'Windows':
|
||
cmd = "C:\\TDengine\\stop_all.bat"
|
||
elif system == 'Linux':
|
||
cmd = "systemctl stop taosd"
|
||
else:
|
||
cmd = "sudo launchctl stop com.tdengine.taosd"
|
||
process_out = subprocess.getoutput(cmd)
|
||
print(cmd)
|
||
time.sleep(10)
|
||
if system == "Linux":
|
||
# 创建一个subprocess.Popen对象，并使用stdin和stdout进行交互
|
||
process = subprocess.Popen(['rmtaos'],
|
||
stdin=subprocess.PIPE, stdout=subprocess.PIPE, text=True)
|
||
# 向子进程发送输入
|
||
process.stdin.write("y\n")
|
||
process.stdin.flush() # 确保输入被发送到子进程
|
||
process.stdin.write("I confirm that I would like to delete all data, log and configuration files\n")
|
||
process.stdin.flush() # 确保输入被发送到子进程
|
||
# 关闭子进程的stdin，防止它无限期等待更多输入
|
||
process.stdin.close()
|
||
# 等待子进程结束
|
||
process.wait()
|
||
# 检查目录清除情况
|
||
out = subprocess.getoutput("ls /etc/systemd/system/taos*")
|
||
if "No such file or directory" not in out:
|
||
print("Uninstall left some files: %s" % out)
|
||
leftFile = True
|
||
out = subprocess.getoutput("ls /usr/bin/taos*")
|
||
if "No such file or directory" not in out:
|
||
print("Uninstall left some files: %s" % out)
|
||
leftFile = True
|
||
out = subprocess.getoutput("ls /usr/local/bin/taos*")
|
||
if "No such file or directory" not in out:
|
||
print("Uninstall left some files: %s" % out)
|
||
leftFile = True
|
||
out = subprocess.getoutput("ls /usr/lib/libtaos*")
|
||
if "No such file or directory" not in out:
|
||
print("Uninstall left some files: %s" % out)
|
||
leftFile = True
|
||
out = subprocess.getoutput("ls /usr/lib64/libtaos*")
|
||
if "No such file or directory" not in out:
|
||
print("Uninstall left some files: %s" % out)
|
||
leftFile = True
|
||
out = subprocess.getoutput("ls /usr/include/taos*")
|
||
if "No such file or directory" not in out:
|
||
print("Uninstall left some files: %s" % out)
|
||
leftFile = True
|
||
out = subprocess.getoutput("ls /usr/local/taos")
|
||
#print(out)
|
||
if "No such file or directory" not in out:
|
||
print("Uninstall left some files in /usr/local/taos：%s" % out)
|
||
leftFile = True
|
||
if not leftFile:
|
||
print("*******Test Result: uninstall test passed ************")
|
||
|
||
elif system == "Darwin":
|
||
# 创建一个subprocess.Popen对象，并使用stdin和stdout进行交互
|
||
process = subprocess.Popen(['sudo', 'rmtaos'],
|
||
stdin=subprocess.PIPE, stdout=subprocess.PIPE, text=True)
|
||
# 向子进程发送输入
|
||
process.stdin.write("y\n")
|
||
process.stdin.flush() # 确保输入被发送到子进程
|
||
process.stdin.write("I confirm that I would like to delete all data, log and configuration files\n")
|
||
process.stdin.flush() # 确保输入被发送到子进程
|
||
# 关闭子进程的stdin，防止它无限期等待更多输入
|
||
process.stdin.close()
|
||
# 等待子进程结束
|
||
process.wait()
|
||
# 检查目录清除情况
|
||
out = subprocess.getoutput("ls /usr/local/bin/taos*")
|
||
if "No such file or directory" not in out:
|
||
print("Uninstall left some files: %s" % out)
|
||
leftFile = True
|
||
out = subprocess.getoutput("ls /usr/local/lib/libtaos*")
|
||
if "No such file or directory" not in out:
|
||
print("Uninstall left some files: %s" % out)
|
||
leftFile = True
|
||
out = subprocess.getoutput("ls /usr/local/include/taos*")
|
||
if "No such file or directory" not in out:
|
||
print("Uninstall left some files: %s" % out)
|
||
leftFile = True
|
||
#out = subprocess.getoutput("ls /usr/local/Cellar/tdengine/")
|
||
#print(out)
|
||
#if out:
|
||
# print("Uninstall left some files: /usr/local/Cellar/tdengine/%s" % out)
|
||
# leftFile = True
|
||
#if not leftFile:
|
||
# print("*******Test Result: uninstall test passed ************")
|
||
|
||
elif system == "Windows":
|
||
process = subprocess.Popen(['unins000','/silent'],
|
||
stdin=subprocess.PIPE, stdout=subprocess.PIPE, text=True)
|
||
process.wait()
|
||
time.sleep(10)
|
||
out = subprocess.getoutput("ls C:\TDengine")
|
||
print(out)
|
||
if len(out.split("\n")) > 3:
|
||
leftFile = True
|
||
print("Uninstall left some files: %s" % out)
|
||
|
||
if version_test_result:
|
||
print("**********Test Result: version test passed! **********")
|
||
else:
|
||
print("!!!!!!!!!!!Test Result: version test failed! !!!!!!!!!!")
|
||
if not leftFile:
|
||
print("**********Test Result: uninstall test passed! **********")
|
||
else:
|
||
print("!!!!!!!!!!!Test Result: uninstall test failed! !!!!!!!!!!")
|
||
if version_test_result and not leftFile:
|
||
sys.exit(0)
|
||
else:
|
||
sys.exit(1)
|
||
|