用python玩转selenium：2-入门实例及分析

selenium2入门

上篇-用python玩转selenium：1-准备环境已经介绍了selenium-python环境配置，或者说安装的python已经集成了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

2.4 分析2.3中测试脚本

1）首先需要导入相关模块，其中unittest是python的内置模块，它提供了组织测试用例的框架，而selenium.webdriver提供了所有WebDriver的实现，目前支持FireFox、Chrome、Ie和Remote。Keys类提供了关键字，如RETURN、F1、ALT等。代码如下：

importunittest
fromselenium import webdriver
fromselenium.webdriver.common.keys import Keys

2）测试用例继承于unittest，代码如下：

class PythonOrgSearch(unittest.TestCase):

3）SetUp方法是初始化的一部分，它会在每个测试功能开始之前被调用，创建firefox webdriver实例的代码如下：

defsetUp(self):

    self.driver = webdriver.Firefox()

4）创建测试用例，测试用例的方法名尽量以test字符串开头，代码的第一行创建webdriver实例对象的引用：

deftest_search_in_python_org(self):

    driver = self.driver

5）driver.get()方法打开URL定义的网址，webdriver会等待到页面完全加载完成后将控制权重新交给测试脚本，代码如下：

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

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

assert "Python" in driver.title

7）webdriver提供了很多如find_element_by_*的方法来匹配要查找的元素。如，利用name属性查找方法find_element_by_name来定位输入框。元素定位方法可以参考后面章节-元素定位方法。

elem = driver.find_element_by_name("q")

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

elem.send_keys("pycon")

elem.send_keys(Keys.RETURN)

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

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

10)所有测试用例执行完毕后会调用tearDown方法，这个方法主要执行清理工作。在本脚本中，tearDown方法执行关闭浏览器的操作，当然，也可以调用quit（）方法，两者的区别在于，quit（）方法会退出浏览器，而close（）方法只是关闭页面，但如果只有一个页面被打开，close（）方法同样会退出浏览器。

deftearDown(self):

    self.driver.close()

11)代码最后两行是执行测试套件的固定写法，代码如下：

if__name__ == "__main__":

    unittest.main()

2.5 使用remote WebDriver

使用remote 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

上面的信息指明了连接seleniumserver的地址http://127.0.0.1:4444/wd/hub，下面是实例代码：

fromselenium.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)

Desiredcapabilities是字典类型，因此除了使用默认值，也可以重新定义字典的值，代码如下：

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

 

-------------------未完待续-下一篇：用python玩转selenium：3-selenium常用操作-------------------