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

selenium2入门

上篇-用python玩转selenium：1-准备环境已经介绍了selenium-python环境配置，或者说安装的python已经集成了selenium模块，现在我们就可以开始编写第一个脚本。

[python]  view plain copy print ?

1. #coding = utf-8  
2. from selenium import webdriver  
3. from selenium.webdriver.common.keys import keys  
4.   
5. driver = webdriver.Firefox()  
6. driver.get('http://www.python.org')  
7. assert 'python' in driver.title  
8. elem = driver.find_element_by_name('q')  
9. elem.send_keys('pycon')  
10. elem.send_keys(keys.RETURN)  
11. assert 'No results found.' not in driver.page_source  
12. driver.close()  

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

Python python_org_search.py

实例分析

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

1）导入相关的包

[python]  view plain copy print ?

1. from selenium import webdriver  
2. 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框架。

[python]  view plain copy print ?

1. import unittest  
2. from selenium import webdriver  
3. from selenium.webdriver.common.keys import Keys  
4.    
5. class PythonOrgSearch(unittest.TestCase):  
6.    
7.     def setUp(self):  
8.         self.driver = webdriver.Firefox()  
9.    
10.     def test_search_in_python_org(self):  
11.         driver = self.driver  
12.         driver.get("http://www.python.org")  
13.         self.assertIn("Python", driver.title)  
14.         elem = driver.find_element_by_name("q")  
15.         elem.send_keys("pycon")  
16.         elem.send_keys(Keys.RETURN)  
17.         assert "No results found." not in driver.page_source  
18.    
19.    
20.     def tearDown(self):  
21.         self.driver.close()  
22.    
23. if __name__ == "__main__":  
24.     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等。代码如下：

[python]  view plain copy print ?

1. importunittest  
2. fromselenium import webdriver  
3. 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，下面是实例代码：

[python]  view plain copy print ?

1. fromselenium.webdriver.common.desired_capabilities import DesiredCapabilities  
2.    
3. driver = webdriver.Remote(  
4.    command_executor='http://127.0.0.1:4444/wd/hub',  
5.    desired_capabilities=DesiredCapabilities.CHROME)  
6.    
7. driver = webdriver.Remote(  
8.    command_executor='http://127.0.0.1:4444/wd/hub',  
9.    desired_capabilities=DesiredCapabilities.OPERA)  
10.    
11. driver = webdriver.Remote(  
12.    command_executor='http://127.0.0.1:4444/wd/hub',  
13.    desired_capabilities=DesiredCapabilities.HTMLUNITWITHJS)  

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

[python]  view plain copy print ?

1. driver = webdriver.Remote(  
2.    command_executor='http://127.0.0.1:4444/wd/hub',  
3.    desired_capabilities={'browserName':'htmlunit',  
4.                          'version':'2',  
5.                         'javascriptEnabled':True})<span style="font-family: Arial, Helvetica, sans-serif; background-color: rgb(255, 255, 255);"> </span>  

 

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