2.1 簡單的例子

如果你安裝了Selenium的Python工具包葡兑，你可以通過Python來使用它缨历。

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.clear()
elem.send_keys("pycon")
elem.send_keys(Keys.RETURN)
assert "No results found." not in driver.page_source
driver.close()

把上面的代碼保存到文件（例如：python_org_search.py), 可以通過命令來運(yùn)行它：
python python_org_search.py

在運(yùn)行上面的代碼之前確保selenium模塊已經(jīng)安裝。

2.2 例子詳解

selenium.webdriver模塊提供所有的WebDriver實(shí)現(xiàn)千劈，現(xiàn)在支持的WebDriver實(shí)現(xiàn)有Firefox,Chrome,IE和遠(yuǎn)程執(zhí)行祭刚，keys類支持鍵盤鍵例如：回車，F(xiàn)1, ALT等等墙牌。

from selenium import webdriver
from selenium.webdriver.common.keys import Keys

下面涡驮，創(chuàng)建一個(gè)Firefox的WebDriver實(shí)例。

driver = webdriver.Firefox()

driver.get方法像瀏覽器那樣訪問url喜滨，WebDriver會(huì)等整個(gè)頁面下載完才把控制權(quán)交回給測試腳本捉捅，如果你的頁面用了很多ajax那么WebDriver可能不知道什么時(shí)候加載完成。

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

下面是一個(gè)斷言來確認(rèn)標(biāo)題里面有"Python"字符：

assert "Python" in driver.title

WebDriver提供了很多像find_element_by_*的方法來查找元素虽风，例如棒口，可以使用find_element_by_name方法通過其name屬性來定位輸入文本元素。

elem = driver.find_element_by_name("q")

接下來辜膝，我們發(fā)送按鍵无牵，這類似于使用鍵盤輸入按鍵〕Ф叮可以使用從selenium.webdriver.common.keys導(dǎo)入的keys類來發(fā)送特殊的鍵合敦。為了安全起見，我們將首先清除輸入字段(如“Search”)中任何預(yù)先填充的文本验游，這樣它就不會(huì)影響我們的搜索結(jié)果:

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

提交頁面后，如果有結(jié)果保檐，應(yīng)該得到結(jié)果耕蝉。為了確保找到某些結(jié)果，請作出以下斷言:

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

最后夜只，關(guān)閉瀏覽器窗口垒在。您也可以調(diào)用quit方法而不是close∪雍ィ“退出”將退出整個(gè)瀏覽器场躯，而“關(guān)閉”將關(guān)閉一個(gè)標(biāo)簽谈为，但如果只有一個(gè)標(biāo)簽被打開，默認(rèn)情況下大多數(shù)瀏覽器將完全退出踢关。

driver.close()

2.3 使用Selenium編寫測試腳本

Selenium主要用于編寫測試用例伞鲫。selenium本身并不提供測試工具/框架。您可以使用Python的unittest模塊編寫測試用例签舞。其他可選的工具/框架是py.test和nose秕脓。
在本章中，我們使用unittest作為選擇框架儒搭。下面是使用unittest模塊修改的示例吠架。這是一個(gè)測試python.org搜索功能的例子:

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()

你可以在shell中執(zhí)行上面的測試用例

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

OK

上面的結(jié)果表示測試用例執(zhí)行完成。

2.4 看一下例子

首先要導(dǎo)入基礎(chǔ)的模塊搂鲫，unittest是居于java的JUnit并且內(nèi)置于Python傍药，這個(gè)模塊提供了測試用例的框架，selenium.webdriver模塊提供所有的webdriver實(shí)現(xiàn)魂仍。目前支持webdriver實(shí)現(xiàn)的有
?Firefox, Chrome, Ie和遠(yuǎn)程調(diào)用拐辽，Keys類在鍵盤中提供諸如RETURN、F1蓄诽、ALT等鍵薛训。

import unittest
from selenium import webdriver
from selenium.webdriver.common.keys import Keys

測試用例類繼承于unittest.TestCase，繼承TestCase類著可以可以告訴unittest模塊這是一個(gè)測試用例仑氛。

class PythonOrgSearch(unittest.TestCase):

setUp是初始化方法乙埃，這個(gè)方法將會(huì)在測試用例里首先被調(diào)用，創(chuàng)建一個(gè)WebDriver的Firefox實(shí)例锯岖。

這里是測試用例的方法介袜，測試用例方法名總是以test開頭，方法的第一行創(chuàng)建一個(gè)在setUp方法創(chuàng)建的driver對象的本地引用出吹。

def test_search_in_python_org(self):
??? driver = self.driver

最后是運(yùn)行測試用例的程序入口

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

2.5 使用Selenium的遠(yuǎn)程WebDriver

使用遠(yuǎn)程WebDriver, 你必須運(yùn)行Selenium server, 下面是運(yùn)行命令遇伞。

java -jar selenium-server-standalone-2.x.x.jar

當(dāng)你啟動(dòng)Selenium server后，可以看到類似下面的信息捶牢。

15:43:07.541 INFO - RemoteWebDriver instances should connect to: http://127.0.0.1:4444/wd/hub

上面的信息是說你可以用這個(gè)url去鏈接遠(yuǎn)程WebDriver鸠珠，下面是例子。

from selenium import webdriver
from selenium.webdriver.common.desired_capabilities import DesiredCapabilities

driver = webdriver.Remote(
?? command_executor='http://127.0.0.1:4444/wd/hub',
?? desired_capabilities=DesiredCapabilities.CHROME)

driver = webdriver.Remote(
?? command_executor='http://127.0.0.1:4444/wd/hub',
?? desired_capabilities=DesiredCapabilities.OPERA)

driver = webdriver.Remote(
?? command_executor='http://127.0.0.1:4444/wd/hub',
?? desired_capabilities=DesiredCapabilities.HTMLUNITWITHJS)

desired_capabilities是一個(gè)字典秋麸，你可以指定字典的值渐排，而不是用缺省的字典。

driver = webdriver.Remote(
?? command_executor='http://127.0.0.1:4444/wd/hub',
?? desired_capabilities={'browserName': 'htmlunit',
???????????????????????? 'version': '2',
??????????????????????? 'javascriptEnabled': True})