Python WebDriver 工作原理

webdriver 是按照 server – client 的经典设计模式设计的。 server 端就是 remote server，可以是任意的浏览器。当我们的脚本启动浏览器后，该浏览器就是remote server，它的职责就是等待 client 发送请求并做出响应； client 端简单说来就是我们的测试代码，我们测试代码中的一些行为，比如打开浏览器，转跳到特定的 url 等操作是以 http 请求的方式发送给被 测试浏览器，也就是 remote server；remote server 接受请求，并执行相应操作，并在 response 中返回执行状态、返回值等信息； webdriver 的工作流程：

WebDriver 启动目标浏览器， 并绑定到指定端口。 该启动的浏览器实例， 做为 web driver 的 remote server。

Client 端通过 CommandExcuter 发送 HTTPRequest 给 remote server 的侦听端口(通信协议： the webriver wire protocol)

Remote server 需要依赖原生的浏览器组件(如：IEDriverServer .exe、chromedriver .exe) ，来转化转化浏览器的 native 调用。

Python版Selenium提供了一套用于编写功能测试及验收测试的API。利用这套简单的API，不仅可以很直观的接触到Selenium WebDriver的所有功能，而且还可以很方便的访问各类WebDrivers，如Firefox、Ie、Chrome、Remote等。目前Selenium支持的Python版本：2.7、3.2、3.3和3.4.

Selenium可以从PyPI page for selenium package下载，但更好的方法是使用pip工具安装，Python3.4的标准库已经内置了pip工具。使用pip工具，可用如下命令安装Selenium：

pip install selenium
始编写第一个脚本
#coding = utf-8
from selenium import webdriver
from selenium.webdriver.common.keys import keys
driver = webdriver.Firefox()
driver.get('http://www.python.org')
assert 'python' in driver.title
elem = driver.find_element_by_name('q')
elem.send_keys('pycon')
elem.send_keys(keys.RETURN)
assert 'No results found.' not in driver.page_source
driver.close()

保存该脚本为python_org_search.py,并运行该脚本：

Python python_org_search.py

实例分析

Selenium.webdriver模块提供了webdriver的实现方法，目前支持这些方法的有Firefox、Chrome、IE和Remote。其中，Keys类提供了操作键盘的快捷键，如RETURE、F1、ALT等。

1)导入相关的包

from selenium import webdriver
from selenium.webdriver.common.keys importKeys

2)接下来，创建Firefoxwebdriver实例。

driver = webdriver.Firefox()

3)driver.get()尝试打开URL指定的网页，webdriver会等待网页元素加载完成之后才把控制权交回脚本。但是，如果要打开了页面在加载的过程中包含了很多AJAX，webdriver可能无法准确判断页面何时加载完成。

driver.get("http://www.python.org")

4)然后，使用断言判断页面标题包含“Python”：

assert "Python" in driver.title

5)webdriver提供了很多如find_element_by_*的方法来匹配要查找的元素。如，利用name属性查找方法

find_element_by_name 来定位输入框。元素定位方法可以参考后面章节-元素定位方法。

elem = driver.find_element_by_name("q")

6)send_keys方法可以用来模拟键盘操作，但首先要从selenium.webdriver.common.keys导入Keys类：

elem.send_keys("pycon")
elem.send_keys(Keys.RETURN)

7)然后，提交请求页面并获得返回结果，另外，为了判断结果是否成功返回，可加入断言：

assert "No results found." not in driver.page_source

8)最后，操作完成并关闭浏览器。当然，也可以调用quit()方法，两者的区别在于，quit()方法会退出浏览器，而close()方法只是关闭页面，但如果只有一个页面被打开，close()方法同样会退出浏览

器。

driver.close()

编写测试脚本

Selenium可用来编写测试用例，虽然Selenium并没有提供测试框架，但可以调用python的unittest模块，或py.test和nose。

本节给出了一个实现搜索python.org功能的测试用例，其中使用的是unittest框架。

import unittest
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
class PythonOrgSearch(unittest.TestCase):
def setUp(self):
self.driver = webdriver.Firefox()
def test_search_in_python_org(self):
driver = self.driver
driver.get("http://www.python.org")
self.assertIn("Python", driver.title)
elem = driver.find_element_by_name("q")
elem.send_keys("pycon")
elem.send_keys(Keys.RETURN)
assert "No results found." not in driver.page_source
def tearDown(self):
self.driver.close()
if __name__ == "__main__":
unittest.main()

使用如下命令即可运行上面的测试脚本：

python test_python_org_search.py
----------------------------------------------------------------------
Ran 1 test in 15.566s
OK