From e749ab8ddb8bcd1568fa92bdb4ada61bcda035d7 Mon Sep 17 00:00:00 2001
From: shujunqiao <shujun.qiao@chukong-inc.com>
Date: Wed, 23 Apr 2014 16:32:37 +0800
Subject: [PATCH] [ci skip], add ssh upload file  in emptytest.py.

---
 tools/jenkins-scripts/emptytest.py | 277 +++++++++++++++++++----------
 1 file changed, 183 insertions(+), 94 deletions(-)
 mode change 100755 => 100644 tools/jenkins-scripts/emptytest.py

diff --git a/tools/jenkins-scripts/emptytest.py b/tools/jenkins-scripts/emptytest.py
old mode 100755
new mode 100644
index 729711f9cd..8db144e6e6
--- a/tools/jenkins-scripts/emptytest.py
+++ b/tools/jenkins-scripts/emptytest.py
@@ -1,9 +1,9 @@
 #!/usr/bin/python
-# create new project by cocos-console
-# compile, deploy project and run
+# check crash of cppemptytest
+# install/open/uninstall apk
 # perpose: for emptytest.
-# now support: mac- mac/ios/android
-# will add: window-android,linux-android
+# now support: android
+# will add: autotest-win7
 
 import os
 import sys
@@ -11,6 +11,10 @@ import json
 import time
 import socket
 import platform
+import threading
+import codecs
+import re
+import paramiko
 
 payload = {}
 #get payload from os env
@@ -26,7 +30,6 @@ if payload.has_key('number'):
 print 'pr_num:' + str(pr_num)
 run_app_time = 5
 if os.environ.has_key('RUN_APP_TIME'):
-	global run_app_time
 	run_app_time = os.environ['RUN_APP_TIME']
 print 'run_app_time:', run_app_time
 
@@ -49,8 +52,6 @@ if os.environ.has_key('TEST_INDEX'):
 current_platform = platform.system()
 print 'current platform is:', current_platform
 
-empty_test_result = True
-empty_test_result_info = ''
 arrDevices = []
 def getDevices():
 	cmd = 'adb devices'
@@ -84,94 +85,95 @@ def mapIP():
 		ip_d = getADBDeviceIP(device['name'])
 		device['ip'] = ip_d
 
-device_info = {}
+allThreadIsRunning = {}
+def setThreadStatus():
+	for device in arrDevices:
+		allThreadIsRunning[device['name']] = 1
+
+devices_info = {}
 info_list = '{"product":["model","brand","name","cpu.abi","cpu.abi2","manufacturer","locale.language","locale.region"],"build":["id","version.sdk","version.release"]}'
 if os.environ.has_key('DEVICE_INFO_LIST'):
 	info_list = os.environ['DEVICE_INFO_LIST']
 info_list = eval(info_list)
 def getDeviceInfoByName(name):
 	cmd = ''
+	dev_name = name
 	if len(name) > 0:
 		cmd = 'adb -s '+name+' shell cat /system/build.prop'
 	else:
 		cmd = 'adb shell cat /system/build.prop'
+		dev_name = 'device one'
 	pip_cat = os.popen(cmd)
 	read_info = pip_cat.read()
 	read_info_list = read_info.split('\r\n')
-	def checkProperty(item_str):
+	device_info_one = {}
+	def checkProperty(item_str, device_name):
 		for argv in info_list:
 			for item in info_list[argv]:
 				prop = argv+'.'+item
 				if item_str.find(prop) > -1:
 					arr_item = item_str.split('=')
-					device_info[prop] = arr_item[1]
+					device_info_one[prop] = arr_item[1]
 					break
 	for item in read_info_list:
-		checkProperty(item)
+		checkProperty(item, dev_name)
+	devices_info[dev_name] = device_info_one
 
 #getDeviceInfoByName('')
 #print 'device_info:',device_info
-def logDeviceInfomation():
-    getDeviceInfoByName('')
-    for key in device_info:
-        print '\t'+key+':'+device_info[key]
+def getDeviceInfomation():
+	for device in arrDevices:
+		getDeviceInfoByName(device['name'])
 
 info_empty_test = {}
+info_empty_test_pro = ['install','open','socket','uninstall']
+def init_info_empty_test():
+	for item in info_empty_test_pro:
+		info_empty_test[item] = {}
+init_info_empty_test()
+
 apk_name = 'apks/'+test_name[gIdx]+'/'+test_name[gIdx]+'_'+str(pr_num)+'.apk'
-def install_apk():
-	print 'will install apk:', apk_name
-	global empty_test_result
-	if len(arrDevices) == 0:
-		empty_test_result = False
-		empty_test_result_info = 'no android device.'
-		print empty_test_result_info
-		return False
-	info_of_install = []
-	if not os.path.isfile(apk_name):
-		print apk_name, 'is not exist!'
-		empty_test_result = False
-		return False
-	for device in arrDevices:
-		name = device['name']
-		cmd = 'adb -s '+name+' install '+apk_name
-		print 'install on '+name
-		info_install = os.popen(cmd).read()
-		print 'infomation of install apk:', info_install
-		info_of_install.append(info_install.split('\r\n'))
-	info_empty_test['install'] = info_of_install
+def install_apk_on_device(device):
+	name = device['name']
+	cmd = 'adb -s '+name+' install '+apk_name
+	print 'install on '+name
+	info_install = os.popen(cmd).read()
+	print 'infomation of install apk:', info_install
+	info_install_arr = info_install.split('\r\n')
+	info_install_result = False
+	for item in info_install_arr:
+		if item.find('Success') > -1:
+			info_install_result = True
+	info_empty_test['install'][name] = info_install_result
 	return True
 
-def open_apk(type_of_apk):
+def open_apk_on_device(device):
 	print 'will open activity:'
-	for device in arrDevices:
-		cmd = 'adb -s '+device['name']+' shell am start -n '+package_name[gIdx]+'/'+activity_name[gIdx]
-		# print 'start activity:', cmd
-		info_start = os.popen(cmd).read()
-		info_start = info_start.split('\n')
-		# print 'info_start:', info_start
-		for info in info_start:
-			if info.find('Error:') > -1:
-				print 'infomation of open activity:',info
-				global empty_test_result
-				empty_test_result = False
-				empty_test_result_info = 'open :'+info
-				return info
-	if len(arrDevices):
-		print 'activity is opened.'
-	else:
-		print 'no device.'
+	name = device['name']
+	cmd = 'adb -s '+name+' shell am start -n '+package_name[gIdx]+'/'+activity_name[gIdx]
+	# print 'start activity:', cmd
+	info_start = os.popen(cmd).read()
+	info_start = info_start.split('\r\n')
+	# print 'info_start:', info_start
+	info_start_result = True
+	for info in info_start:
+		if info.find('Error:') > -1:
+			print 'infomation of open activity:',info
+			info_start_result = False
+	info_empty_test['open'][name] = info_start_result
 	return True
 
 PORT = 5678
-def socket_status(device_name):
+def socket_status_on_device(device):
+	name = device['name']
+	ip = device['ip']
 	soc = socket.socket( socket.AF_INET, socket.SOCK_STREAM )
 	status_socket = False
-	global empty_test_result
+	info_of_socket_result = ''
 	try:
-		print 'telnet ', device_name['ip'], PORT
-		soc.connect((device_name['ip'], PORT))
+		print 'telnet ', ip, PORT
+		soc.connect((ip, PORT))
 		cmd = 'resolution\r\n'
-		# print 'socket cmd :', cmd
 		print 'connected successfully.'
 		print 'send console command: resolution'
 		soc.send(cmd)
@@ -180,71 +182,158 @@ def socket_status(device_name):
 			if len(data):
 				print data
 				if data.find('size:') > -1:
-					print 'OK'
+					info_of_socket_result = 'OK'
 					print 'close', test_name[gIdx]
 					soc.send('director end')
-					print 'OK'
 					status_socket = True
 					break
 			if not data:
-				empty_test_result = False
-				empty_test_result_info = test_name[gIdx]+' is crashed!'
-				print empty_test_result_info
+				info_of_socket_result = test_name[gIdx]+' is crashed!'
 				break
 	except Exception, e:
-		empty_test_result = False
-		empty_test_result_info = test_name[gIdx]+' is crashed!'
-		print empty_test_result_info
+		info_of_socket_result = test_name[gIdx]+' is crashed!'
 	time.sleep(2)
 	soc.close()
+	time.sleep(2)
+	info_empty_test['socket'][name] = info_of_socket_result
 	return status_socket
 
-def uninstall_apk(idx):
+def uninstall_apk_on_device(device):
 	# adb shell pm uninstall -n org.cocos2dx.hellolua
-	print 'uninstall ', test_name[idx]
-	for device in arrDevices:
-		cmd = 'adb -s '+device['name']+' shell pm uninstall -n '+package_name[idx]
-		info_uninstall = os.popen(cmd).read()
-		if info_uninstall.find('Success') > -1:
-			print 'OK'
-		else:
-			empty_test_result_info = 'uninstall Failed!'
-			print empty_test_result_info
+	print 'uninstall ', test_name[gIdx]
+	name = device['name']
+	cmd = 'adb -s '+name+' shell pm uninstall -n '+package_name[gIdx]
+	info_uninstall = os.popen(cmd).read()
+	info_uninstall_result = ''
+	if info_uninstall.find('Success') > -1:
+		info_uninstall_result = 'OK'
+	else:
+		info_uninstall_result = 'uninstall Failed!'
+	info_empty_test['uninstall'][name] = info_uninstall_result
 	return True
 
+def excute_test_on_device(device):
+	uninstall_apk_on_device(device)
+	print device
+	info_install = install_apk_on_device(device)
+	print 'install:', info_install
+	info_open = open_apk_on_device(device)
+	print 'open:', info_open
+	time.sleep(3)
+	info_socket = socket_status_on_device(device)
+	print 'socket:', info_socket
+	info_uninstall = uninstall_apk_on_device(device)
+	print 'uninstall:', info_uninstall
+	allThreadIsRunning[device['name']] = 0
+
+def send_result_to_master():
+	if not os.environ.has_key('REMOTE_IP'):
+		return false
+	remote_ip = os.environ['REMOTE_IP']
+	remote_port = os.environ['REMOTE_PORT']
+	remote_user = os.environ['REMOTE_USER']
+	remote_pwd = os.environ['REMOTE_PWD']
+	remote_dir = os.environ['REMOTE_DIR']
+	print remote_dir
+	ssh = paramiko.SSHClient()
+	print 'ssh:',ssh
+	ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
+	ssh.connect(remote_ip, int(remote_port), remote_user, remote_pwd)
+	# create dir
+	stdin, stdout, stderr = ssh.exec_command("mkdir "+remote_dir)
+	ssh.close()
+	trans = paramiko.Transport((remote_ip,int(remote_port)))
+	trans.connect(username = remote_user, password = remote_pwd)
+	sftp = paramiko.SFTPClient.from_transport(trans)
+	remotepath = remote_dir+test_name[gIdx]+'_'+str(pr_num)+'.html'
+	localpath = 'html/cpp_empty_test/cpp_empty_test_'+str(pr_num)+'.html'
+	sftp.put(localpath, remotepath)
+	sftp.close()
+
+def check_thread_is_running():
+	while 1:
+		in_running = 0
+		for key in allThreadIsRunning:
+			in_running = in_running + allThreadIsRunning[key]
+		# print 'in_running:', in_running
+		time.sleep(3)
+		if not in_running:
+			break
+	print 'all cpp emptytest is finished.'
+	log_emptytest_result()
+	print 'will send result:'
+	send_result_to_master()
+	print 'end of check thread is running.'
+
+def generate_html_with_result(result):
+	target_path = 'html/cpp_empty_test/cpp_empty_test_'+str(pr_num)+'.html'
+	data = codecs.open('hello.html', encoding='UTF-8').read()
+	data = re.sub("content", result, data)
+	codecs.open(target_path, 'wb', encoding='UTF-8').write(data)
+
+empty_test_result = True
+str_result = ''
+def log_emptytest_result():
+	def appendToResult(one_str):
+		global str_result
+		str_result = str_result + one_str + '\r\n'
+	appendToResult('<pre>')
+	appendToResult('empty test start:')
+	for device in arrDevices:
+		name = device['name']
+		appendToResult('device infomation of : '+name)
+		for key in devices_info[name]:
+			appendToResult('\t'+key+':'+devices_info[name][key])
+		appendToResult('install: ' + str(info_empty_test['install'][name]))
+		appendToResult('open: ' + str(info_empty_test['open'][name]))
+		appendToResult('telnet ' + str(device['ip'])+' : ' + str(info_empty_test['socket'][name]))
+		appendToResult('close: ' + str(info_empty_test['socket'][name]))
+		appendToResult('uninstall: ' + str(info_empty_test['uninstall'][name]))
+		if not info_empty_test['install'][name] or not info_empty_test['open'][name] or not info_empty_test['socket'][name] or not info_empty_test['uninstall'][name]:
+			empty_test_result = False
+			appendToResult('empty test is failed!')
+		else: appendToResult('empty test is successfully!')
+		appendToResult('')
+	appendToResult('empty test end.</pre>')
+	generate_html_with_result(str_result)
+	print 'log empty test end'
+
+class myThread(threading.Thread):
+    def __init__(self,threadname):
+        threading.Thread.__init__(self,name=threadname)
+    def run(self):
+        time.sleep(2)
+        device = self.getName()
+        print type(device)
+        device = eval(device)
+        print type(device), device
+        excute_test_on_device(device)
+
+def run_emptytest():
+	for device in arrDevices:
+		th = myThread(device)
+		th.start()
+
 def main():
 	print 'in main:'
 	getDevices()
 	if len(arrDevices):
 		mapIP()
+		setThreadStatus()
 		print 'arrDevices:',arrDevices
-		uninstall_apk(gIdx)
 		time.sleep(1)
 	else:
 		print 'there is no device for emptytest, please check devices!'
 		return 1
-	print 'empty test start:'
-	print 'device infomation:'
 	if len(arrDevices):
-		logDeviceInfomation()
-	else:
-		print '\tno android device.'
-	install_info = install_apk()
-	open_info = open_apk(test_name[gIdx])
-	info_empty_test['open_info'] = open_info
-	if open_info:
-		time.sleep(5)
-		socket_info = socket_status(arrDevices[0])
-		info_empty_test['socket_info'] = socket_info
-	if install_info:
-		time.sleep(run_app_time)
-		info_uninstall = uninstall_apk(gIdx)
+		getDeviceInfomation()
+	run_emptytest()
+	check_thread_is_running()
 	print 'info_empty_test:', info_empty_test
 	print 'empty test end', empty_test_result
 	if empty_test_result:
 		return 0
 	else:
-		print 'empty_test_result_info:', empty_test_result_info
 		return 1
 
 # -------------- main --------------