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