基于python+appium通过图片对比来做的UI自动化

#coding=utf-8
#python3
import unittest
from appium import webdriver
import time 
import os
from PIL import Image
from PIL import ImageChops
#如下导入解决‘image file is truncated’问题
from PIL import ImageFile
ImageFile.LOAD_TRUNCATED_IMAGES = True

PATH = lambda p: os.path.abspath(p) 
def screenshot(testcase):
	path = PATH(os.getcwd() + "/TestResult") 
	if not os.path.isdir(PATH(os.getcwd() + "/TestResult")): 
		os.makedirs(path) 
	os.popen("adb wait-for-device") 
	time.sleep(1)#由于多次出现截图延迟现象（每次截图都截的是上次操作的画面），故此处设置一个等待
	os.popen("adb shell screencap -p /data/local/tmp/tmp.png") 
	time.sleep(1)
	os.popen("adb pull /data/local/tmp/tmp.png " + PATH(path + "/"  + testcase + ‘.png‘)) 
	time.sleep(1)
	os.popen("adb shell rm /data/local/tmp/tmp.png") 
	time.sleep(1)
	im = Image.open(PATH(path + "/"  + testcase + ‘.png‘))
	cropedIm = im.crop((0, 70, 1079,2080 ))
	cropedIm.save(PATH(path + "/"  + testcase + ‘.png‘))
def compare_images(path_one, path_two, diff_save_location):
	"""
	比较图片，如果有不同则生成展示不同的图片

	@参数一: path_one: 第一张图片的路径
	@参数二: path_two: 第二张图片的路径
	@参数三: diff_save_location: 不同图的保存路径
	"""
	image_one = Image.open(path_one)
	image_two = Image.open(path_two)
	try: 
		diff = ImageChops.difference(image_one, image_two)
		if diff.getbbox() is None:
		# 图片间没有任何不同则直接退出
			return True
		else:
			diff.save(diff_save_location)
			return False
	except ValueError as e:
		text = ("表示图片大小和box对应的宽度不一致，参考API说明：Pastes another image into this image."
				"The box argument is either a 2-tuple giving the upper left corner, a 4-tuple defining the left, upper, "
				"right, and lower pixel coordinate, or None (same as (0, 0)). If a 4-tuple is given, the size of the pasted "
				"image must match the size of the region.使用2纬的box避免上述问题")
		print("【{0}】{1}".format(e,text))


class Test(unittest.TestCase):
	def setUp(self):
		#appium 固定设置，沿用之前的即可
		self.desired_caps = {}
		self.desired_caps[‘platformName‘] = ‘Android‘
		self.desired_caps[‘platformVersion‘] = ‘9‘
		self.desired_caps[‘deviceName‘] = ‘Android Emulator‘
		self.desired_caps[‘appPackage‘] = ‘com.android.settings‘
		self.desired_caps[‘appActivity‘] = ‘.Settings$NetworkDashboardActivity‘	
		self.desired_caps[‘noReset‘] = ‘true‘
		self.driver = webdriver.Remote(‘http://localhost:4723/wd/hub‘, self.desired_caps)
	def test01(self):
		‘‘‘Network & internetEN01‘‘‘
		driver = self.driver
		driver.wait_activity(".Settings$NetworkDashboardActivity", 30)
		screenshot(‘test_NetworkEN01‘)
		driver.find_element_by_xpath("//*[@text=‘Advanced‘]").click()
		time.sleep(2)
		#讲道理这个截图应该放到test02里面的，只是跑逻辑，就懒得改了
		screenshot(‘test_NetworkEN02‘)
		time.sleep(5)
		#下面这个可以用参数来写，会显得短一些，懒得改了
		ac = compare_images(‘./TestResult/test_NetworkEN01.png‘,‘./DesiredResult/test_NetworkEN01.png‘,‘./DiffPicture/test_NetworkEN01.png‘)
		self.assertEqual(ac, True)
	def test02(self):
		‘‘‘Network & internetEN02‘‘‘
		bc = compare_images(‘./TestResult/test_NetworkEN02.png‘,‘./DesiredResult/test_NetworkEN02.png‘,‘./DiffPicture/test_NetworkEN02.png‘)
		self.assertEqual(bc, True)
		#clear envirment
	def tearDown(self):

		#self.assertTrue(ac != preStatus)
		self.driver.quit()
if __name__ == ‘__main__‘:
	unittest.main()

#coding=utf-8

import time 
import unittest
from HTMLTestRunner import HTMLTestRunner
import os



if __name__==‘__main__‘:   
    print (‘=====AutoTest Start======‘)
    test_dir = ‘./Test_case/‘
    #知道测试报告的路径
    test_report_dir=‘./testReport/‘
    PATH = lambda p: os.path.abspath(p) 
    if not os.path.isdir(PATH(os.getcwd() + "/testReport")): 
        os.makedirs(test_report_dir) 
    discover=unittest.defaultTestLoader.discover(test_dir, pattern=‘test_*.py‘)
    now=time.strftime(‘%Y-%m-%d_%H_%M_%S_‘)
    filename = test_report_dir+‘/‘+ now + ‘result.html‘
    fp=open(filename ,‘wb‘)
    
    runner = HTMLTestRunner(stream=fp,title=u‘测试报告‘,description=u‘用例执行情况：‘)
    runner.run(discover)

#注意：调用函数一定要加括号，一个括号害死个人，哎，查了几天的问题，才发现导致html文件始终显示空白，就是因为close函数调用不正确，漏了括号。
    fp.close() 
    
    #2.取最新测试报告
   # new_report=new_file(test_report_dir)
#调试用的
#    print new_report
    
    #3.发送邮件，发送最新测试报告html
  #  send_email(new_report)
    print(‘=====AutoTest Over======‘)
time.sleep(10)

def invert(im1):
	IM = Image.open(im1)
	im3 = ImageChops.invert(IM)
	im3.save(‘./DesiredResult/test_NetworkEN04.png‘)
	im3.close()