WebDriver高阶API（5）

10、UI对象库
定位数据与程序分离

新建一个名叫GoGouTest的工程，工程下新建三个文件，分别为SoGou.py、ObjectMap.py以及UiObjectMap.ini

UiObjectMap.ini 页面元素定位表达式配置文件
[sogou]
searchBox=id>query
searchButton=id=stb

ObjectMap.py 表示ObjexctMap工具类文件，供测试程序调用

#encoding=utf-8
from selenium.webdriver.support.ui import WebDriverWait
import ConfigParser
import os

class ObjectMap(object):
    def __init__(self):
        #获取存放页面元素定位方式及定位表达式的配置文件所在绝对路径
        #os.path.abspath(__file__)表示获取当前文件所在路径目录
        self.uiObjectMapPath = os.path.dirname(os.path.abspath(__file__))+"\\UiObjectMap.ini"
        print self.uiObjectMapPath

    def getElementObject(self,driver,webSiteName,elementName):
        try:
            #创建一个读取配置文件的实例
            cf = ConfigParser.ConfigParser()
            #将配置文件内容加载到内存
            cf.read(self.uiObjectMapPath)
            #根据section和option获取配置文件中页面元素的定位方式及定位表达式组成的字符串，并用">"分割
            locators = cf.get(webSiteName,elementName).split(">")
            #得到定位方式
            locatorMethod = locators[0]
            #得到定位表达式
            locatorExpression = locators[1]
            print locatorMethod,locatorExpression
            #通过显式等待方式 获取页面元素
            element = WebDriverWait(driver,10).until(lambda x: x.find_element(locatorMethod,locatorExpression))
        except Exception,e:
            raise e
        else:
            #当页面元素被找到后，将该页面元素对象返回给调用者
            return element

SoGou.py中调用ObjectMap.py工具类实现测试逻辑

#encoding=utf-8
from selenium import webdriver
import unittest
import time,traceback
from ObjectMap import ObjectMap

class TestSoGouByObjectMap(unittest.TestCase):

    def setUp(self):
        self.obj = ObjectMap()
        #启动浏览器
        self.driver = webdriver.Ie(executable_path="D:\\IEDriverServer")

    def test_SoGouSearch(self):
        url = "http://www.sogou.com"
        #访问搜狗首页
        self.driver.get(url)
        try:
            #查找页面输入框
            searchBox = self.obj.getElementObject(self.driver,"sogou","searchBox")
            #在找到的搜索框中输入关键字
            searchBox.send_keys("World of Warcraft")
            #查找搜索按钮
            searchButton = self.obj.getElementObject(self.driver,"sogou","searchButton")
            #单击找到的搜索按钮
            searchButton.click()
            #等待2秒，以便页面加载完成
            time.sleep(2)
            #断言
            self.assertTrue("World of Warcraft" in self.driver.page_source,"assertError!!!")
        except Exception,e:
            print traceback.print_exc()

    def tearDown(self):
        self.driver.quit()

if __name__ == "__main__":
    unittest.main()

11、精确比较页面截图图片
安装pillow，命令行输入pip install pillow
命令行进行Python交互模块，执行 from PIL import Image没有报错则安装成功

#encoding=utf-8
import unittest,time
from selenium import webdriver
from PIL import Image

class ImageCompare(object):
    """
    本类实现 了对两张图片通过像素比对的算法，获取文件的像素个数大小，然后使用循环的方式将两张
    图片的所有项目进行一一对比，并计算比对结果的相似度的百分比
    """
    def make_regalur_image(self,img,size=(256,256)):
        #将图片尺寸强身重置为指定的size大小
        #然后再将其转换成RGB值
        return img.resize(size).convert("RGB")

    def split_image(self,img,part_size=(64,64)):
        #将图片按给定的大小切分
        w,h = img.size
        pw,ph = part_size
        assert w % pw == h % ph == 0
        return [img.crop((i,j,i+pw,j+ph)).copy() for i in xrange(0,w,pw) for j in xrange(0,h,pw)]

    def hist_similar(self,lh,rh):
        #统计切分后的每部分图片的相似度频率曲线
        assert len(lh) == len(rh)
        return sum(1-(0 if 1 == r else float(abs(1-r))/ max(1,r)) for l,r in zip(lh,rh)) / len(lh)

    def calc_similar(self,li,ri):
        #计算两张图片的相似度
        return sum(self.hist_similar(l.histogram(),r.histogram()) for l,r in zip(self.split_image(li),self.split_image(ri))) / 16.0

    def calc_similar_by_path(self,lf,rf):
        li,ri = self.make_regalur_image(Image.open(lf)),self.make_regalur_image(Image.open(rf))
        return self.calc_similar(li,ri)

class TestDemo(unittest.TestCase):

    def setUp(self):
        self.IC = ImageCompare()
        self.driver = webdriver.Ie(executable_path = "D:\\IEDriverServer")

    def test_ImageComparison(self):
        url = "http://www.sogou.com"
        self.driver.get(url)
        time.sleep(2)
        #截取第一次访问搜狗首页的图片，并保存在本地
        self.driver.save_screenshot("D:\\test\\test\\sogou1.png")
        self.driver.get(url)
        time.sleep(3)
        #截取第二次访问搜首页的图片，并保存在本地
        self.driver.save_screenshot("D:\\test\\test\\sogou2.png")
        #打印两张比对后的相似度，100表示完全匹配
        print self.IC.calc_similar_by_path("D:\\test\\test\\sogou1.png","D:\\test\\test\\sogou2.png") * 100

    def tearDown(self):
        self.driver.quit()

if __name__ == "__main__":
    unittest.main()