[译]Selenium Python文档：二、初步开始

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）。接着，可以像下面这样来运行它（你应该确保要使用的Python已经正确安装了selenium模块）：

$python python_org_search.py

2.2.示例讲解

selenium.webdriver模块提供了所有的WebDriver实现。当前Selenium支持的WebDriver实现由火狐、Chrome、IE和远程。Keys类提供了键盘按键的键值，如RETURN、F1、ALT等。

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

接着，Firefox WebDriver实例被创建:

driver = webdriver.Firefox()

driver.get方法根据给定的URL访问Web 页面。在将控制权交给你的测试脚本之前，WebDriver将会等待页面完全加载（也就是说，等待onload事件被触发）。值得注意的是如果你的页面在加载的时候使用了大量的AJAX，WebDriver有可能不知道它是否已经加载完成:

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

接下来的一行是一个断言，用来确保标题中有“Python”这个单词:

assert "Python" in driver.title

WebDriver为寻找页面元素提供了大量方法，他们都是find_element_by_*的形式。举个例子，input text元素可以根据他的name值使用find_element_by_name来进行定位。在后续的元素定位那一章将会对元素定位做更为详细的介绍。

elem = driver.find_element_by_name("q")

接下来我们发送了一个“按键”值，这和你使用键盘的输入非常相似。特殊案件可以通过从selenium.webdriver.common.keys导入的Keys类进行发送。安全起见，我们首先清除输入框中的预置文本（比如，“搜索”,"Serach"）。以保证它不会影响我们的搜索结果：

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

在页面提交之后，你应该会得到相应的搜索结果。为了保证一些结果被找到，作一个断言：

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

最后，关闭浏览器窗口。你也可以调用quit来代替close方法。quit将会关闭整个浏览器，而close将会仅关闭一个标签页。如果只打开了一个标签页的话，大多数浏览器默认会关闭整个浏览器：

driver.close()

2.3.使用Selenium编写测试

Selenium常被用于编写测试样例。selenium包本身并不提供测试工具/框架，你可以使用Python的unittest模块来编写测试。另外可选的测试工具/框架是py.test和nose。

本章我们选择使用unittest做测试框架。这里是经修改的unittest模块使用样例。本脚本用来测试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()

你可以从命令行来运行上面的测试样例，如：

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

OK

上面的结果表明测试已经成功完成。

2.4.通览示例

首先，加载所有需要的基础模块。unittest是基于Java JUnit的Python自建模块。该模块提供了组织测试样例的框架。selenium.driver模块提供了所有的WebDriver实现。当前支持的WebDriver实现有：火狐、Chrome、IE和远程。类提供了键盘按键的键值，如RETURN、F1、ALT等。

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

test case类继承自unittest.TestCase。从TestCase进行继承会告诉unittest模块这是一个测试样例：

class PythonOrgSearch(unittest.TestCase):

setUp用来进行初始化，在测试样例类中编写的所有测试函数执行之前被调用。在这里，我们创建了一个Firefox（火狐）WebDriver的实例：

def setUp(self):
    self.driver = webdriver.Firefox()

下面是一个测试样例方法。测试样例方法一般应首先进行字符测试。该函数中的第一行创建了一个本地引用，引用的是在setUp中创建的driver对象：

def test_search_in_python_org(self):
    driver = self.driver

driver.get方法根据给定的URL访问Web 页面。在将控制权交给你的测试或者脚本之前，WebDriver将会等待页面完全加载（也就是说，等待onload事件被触发）。值得注意的是如果你的页面在加载的时候使用了大量的AJAX，WebDriver有可能不知道它是否已经加载完成:

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

接下来的一行是一个断言，用来确保标题中有“Python”这个单词:

self.assertIn("Python" ,driver.title)

WebDriver为定位页面元素提供了大量方法，他们都是find_element_by_*的形式。举个例子，input text元素可以根据他的name值使用find_element_by_name定位。在后续的元素定位那一章将会对元素定位做更为详细的介绍。

elem = driver.find_element_by_name("q")

接下来我们发送了一个“按键”值，这和你使用键盘的输入非常相似。特殊案件可以通过从selenium.webdriver.common.keys导入的Keys类进行发送。安全起见，我们首先清除输入框中的预置文本（比如，“搜索”,"Serach"）。以保证它不会影响我们的搜索结果：

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

在页面提交之后，你应该会得到相应的搜索结果。为了保证一些结果被找到，作一个断言：

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

tearDown方法将会在所有的测试方法执行之后调用。此函数用来做一些清理工作。在当前方法中，清理工作是关闭浏览器。你也可以调用quit方法来代替close方法。quit将会关闭整个浏览器，而close将会仅关闭一个标签页。如果只打开了一个标签页的话，大多数浏览器默认会关闭整个浏览器：

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

最后一行用来运行测试套件：

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

2.5.通过远程WebDriver使用Selenium

要使用远程WebDriver，你应该先运行Selenium server。可以使用下面的命令启动它：

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

Selenium server启动之后你可以看到像下面的一条消息：

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

上面那行说明你可以使用这个URL来连接远程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是一个字典，要替换字典中默认键值的话，你可以显式指定对应的值：

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