关于chromedriver和selenium(webdriver)的调研和修改(二)

浏览数：26 / 时间：2015年06月09日

之前在（一）中，主要写的是如何满足当时的需求，来修改chromedriverh和之后的测试。按照之前的思路，我们并不需要使用chromedriver打开浏览器，只是创建session的时候，直接挂到已经启动的浏览器上，当然，这个启动的进程需要打开--remote-debugging-port的参数。由于代码不能公开，所以在（一）里并没有把如何使用的测试代码贴出来。之后自己重新写了一下，代码中不再有和项目相关的东西，可以用来讨论DIY的chromedriver的使用。

首先需要重新描述一下上下文：我们现在需要对chrome进行GUI测试，这些已经基于msaa进行了封装，chrome的界面是支持的（关于msaa和chrome的自动化部分，请查询之前写的一些东西）；在测试chrome的gui界面的同时，我们也需要测试一下页面的情况，这两个是相互独立的。由于要先启动chrome对其gui进行测试，所以后续需要用chromedriver挂到已经打开的chrome上，而不是启动新的session。因为我们重新对chromedriver进行了改造，init_session的时候禁止它启动chrome，并且定死了一个通信端口，具体的做法见（一）。

对已经改造的chromedriver，使用的方法在（一）中已经提到，如下：

1、通过进程启动浏览器——启动命令需要指定port

2、启动chromedriver.exe进行，将port传入

3、通过selenium控制chromedriver.exe进行浏览器的各种操作

可以直接使用selenium操作chromedriver（推荐），也可以直接编写代码操作（不推荐）

测试的代码，遵照这个思路，把操作gui的Chrome测试实例，和selenium&webdriver融合到了一起，进行操作。由于这里只介绍如何融合在一起，怎么对chrome的gui操作进行封装不具体讨论。详细的测试代码如下：

#!/usr/bin/env python
#coding: utf-8

from selenium.webdriver import *
import unittest
import subprocess
import time

from pychrome import PyChrome
import pylog

CHROME_PATH = r"chrome.exe"
DRIVER_PATH = r"chromedriver.exe" 

PORT_ID = ‘myport‘

STARTUP_CMD = r‘--dom-automation=True --remote-debugging-port=‘ + PORT_ID

AUTO_INS = False
BR = None
DRIVER = None

TEST_URLS = (
             "http://www.baidu.com", 
             "http://www.163.com", 
             "http://www.youku.com", 
             "http://www.so.com"
             )

def _beforeSetUp():
    subprocess.Popen(‘taskkill /F /IM chrome.exe‘).wait()
    time.sleep(1)
    subprocess.Popen(CHROME_PATH + ‘ ‘ + STARTUP_CMD).pid
    time.sleep(2)
    subprocess.Popen(DRIVER_PATH).pid
    pass

def _afterTearDown():
        subprocess.Popen(‘taskkill /F /IM chrome.exe‘).wait()
        subprocess.Popen(‘taskkill /F /IM chromedriver.exe‘).wait()

def timeCount(func):
    def wrapper(*arg,**kwarg):
        pylog.log.info(‘-----------%s--------------‘%func)
        start = time.clock()
        func(*arg,**kwarg)
        end =time.clock()
        pylog.log.info(‘used: %s seconds‘, end - start)
    return wrapper

class TestChromedriver(unittest.TestCase):
    @timeCount
    def test_00_SetUp(self):
        global AUTO_INS
        global BR
        global DRIVER
        if not AUTO_INS:
            time.sleep(1) 
            #setup chrome_options for specific path
            Options = ChromeOptions()
            Options.binary_location = CHROME_PATH
            Desiredcapabilities = DesiredCapabilities.CHROME
            #init chrome with option, driver and capability
            DRIVER = Chrome(chrome_options=Options, executable_path = DRIVER_PATH, port = 9515)
            self.assertTrue(DRIVER is not None, "ther‘s no driver instance")
            
            time.sleep(1)
            #create BR instance to hold gui
            BR = PyChrome()
            BR.open()
            time.sleep(1)
            self.assertTrue(DRIVER is not None, "driver‘s created")
        pass
    
    @timeCount
    def tearDown(self):
        pass
    
    @timeCount
    def test_01_SessionId(self):
        self.assertTrue(DRIVER.session_id is not None, "there‘s no session‘s created")
        pass
    
    @timeCount
    def test_02_WindowHandles(self):
        handles = DRIVER.window_handles
        self.assertTrue(len(handles) >= 0, "no window‘s handle")
    
    @timeCount
    def test_03_TabsByManual(self):
        self.assertTrue(DRIVER.current_window_handle is not None)
        DRIVER.get("http://www.baidu.com")
        map(lambda x: BR.open_tab(x), TEST_URLS)
        time.sleep(1)
        self.assertTrue(len(DRIVER.window_handles) > len(TEST_URLS), "invalid handles‘ number")
        pass
    
    @timeCount
    def test_04_SwitchWindows(self):
        handles = DRIVER.window_handles

        def _switch(x):
            try:
                DRIVER.switch_to_window(x)
                DRIVER.get(‘http://www.baidu.com‘)
                return True
            except Exception as e:
                return e
            
        _r = map(lambda x: _switch(x), handles)
        self.assertTrue(any(_r))
        pass
    
    @timeCount
    def test_05_NavigateTab(self):
        DRIVER.switch_to_window(DRIVER.window_handles[0])
        _url = DRIVER.current_url
        map(lambda x: DRIVER.get(x), TEST_URLS)
        try:
            map(lambda x: DRIVER.back(), TEST_URLS)
        except:
            pass
        DRIVER.forward()
        self.assertEqual(_url, DRIVER.current_url)
        
        _url = DRIVER.current_url
        try:
            map(lambda x: DRIVER.forward(), TEST_URLS)
        except:
            pass
        
        self.assertNotEqual(_url, DRIVER.current_url)
        pass
    
    @timeCount
    def test_06_JavaScript(self):
        _url = DRIVER.current_url
        
        script = "location.hash=‘test‘;return location.hash"
        pylog.log.info(DRIVER.execute_script(script))
        
        self.assertNotEqual(_url, DRIVER.current_url)
    
    @timeCount
    def test_07_Alert(self):
        _error = []
        _alert = None
        
        try:
            DRIVER.execute_script(‘alert(555)‘)
        except Exception as e:
            _error.append(str(e))
            
        try:
            _alert = DRIVER.switch_to_alert()
        except Exception as e:
            _error.append(str(e))
            
        if _alert:
            try:
                _alert.accept()
            except Exception as e:
                _error.append(str(e))
    
        self.assertTrue(_error is not None)
        pass
    
    @timeCount
    def test_08_Close(self):
        _handles = DRIVER.window_handles
        DRIVER.close()
        self.assertTrue(len(_handles) > len(DRIVER.window_handles))
    
    @timeCount
    def test_09_Quit(self):
        DRIVER.quit()

if __name__ == ‘__main__‘:
        _beforeSetUp()
        try:
            unittest.main()
        finally:
            _afterTearDown()