Pyunittest的内容我们看的也差不多了，主体功能大致如此，现在我们来看看selenium如何使用Pyunittest执行自动化测试。
本期目标
- 打开百度，搜索“selenium”，关闭浏览器
- 打开百度，设置搜索的最大条数为20条（默认是10条），关闭浏览器
编写测试案例
先编写我们的测试类，在初始化的时候要做一些处理，代码如下
class Baidu(unittest.TestCase):
def setUp(self):
self.driver = webdriver.Chrome()
self.driver.implicitly_wait(30)
self.base_url = "http://www.baidu.com"
self.verificationErrors = []
self.accept_next_alert = True
Baidu类继承unittest.TestCase。初始化中实例化webdriver.Chrome()。
self.driver.implicitly_wait(30) #表示等待一段时间。
self.base_url = "http://www.baidu.com" #定义我们测试的地址。
self.verificationErrors = [] #这行定义的是一个错误接受的列表，暂时不用管。
self.accept_next_alert = True #这行表示接受弹出的alert窗口。
第一个目标
第一个目标很简单，用ID定位输入框和搜索按钮，具体代码如下
def test_baidu_search(self):
driver = self.driver
driver.get(self.base_url + "/")
driver.find_element_by_id("kw").send_keys("selenium webdriver")
driver.find_element_by_id("su").click()
time.sleep(2)
driver.close()
第二个目标
第二个目标有点麻烦，首先百度高级设置的地址是"http://www.baidu.com/gaoji/preferences.html"，我们定义的基础地址是“http://www.baidu.com”，要在后面把相关的字符串补齐。之后我们要定位到搜索条数的地方，修改为20.修改完毕之后会弹出一个alert窗口，我们要干掉这个窗口后再关闭浏览器。具体代码如下：
def test_baidu_set(self):
driver = self.driver
driver.get(self.base_url+"/gaoji/preferences.html")
m = driver.find_element_by_name("NR")
m.find_element_by_xpath("//option[@value='20']").click()
time.sleep(2)
driver.find_element_by_xpath("//input[@value='保存设置']").click()
time.sleep(2)
driver.switch_to_alert().accept()
同理，在这个测试案例中，我们要先引用初始化的driver。定位地址时用Python的字符串拼接方法。
m = driver.find_element_by_name("NR")
m.find_element_by_xpath("//option[@value='20']").click()
这两行表示定位到下拉选项为20的这个地方，第一行是定义到这个下拉组件中。
<select name="NR" id="nr">
<option value="10" selected="">每页显示10条</option>
<option value="20">每页显示20条</option>
<option value="50">每页显示50条</option>
</select>
上面是这个下拉组件的源码，先定位到这个select组件，再使用xpath的方式定位到20条记录这个下拉选项，调用点击按钮选中这条记录。谷歌浏览器的调试工具可以直接复制这条xpath地址（//*[@id="nr"]/option[2]）也可以直接使用这个来定位。
driver.find_element_by_xpath("//input[@value='保存设置']").click()
这条就是直接用xpath方法定位到保存按钮。
driver.switch_to_alert().accept()
这条命令就是接受弹出的alert对话框。
结束测试时清除的函数
每条测试案例开始时都会打开浏览器，然后执行测试案例。执行完毕后需要关闭浏览器，关闭的方法我们写在清除函数中，具体代码如下：
def tearDown(self):
self.driver.quit()
self.assertEqual([],self.verificationErrors)
开始跑测试案例
最后我们需要吧测试案例跑起来，验证我们写的对不对。在前面的章节已经讲过这里要这么跑了。具体代码如下：
if __name__ == '__main__':
testunit = unittest.TestSuite()
testunit.addTest(Baidu('test_baidu_search'))
testunit.addTest(Baidu('test_baidu_set'))
runner.run(testunit)
完整的代码
# -*- coding: utf-8 -*-
from selenium import webdriver
import unittest,time,re
class Baidu(unittest.TestCase):
def setUp(self):
self.driver = webdriver.Chrome()
self.driver.implicitly_wait(30)
self.base_url = "http://www.baidu.com"
self.verificationErrors = []
self.accept_next_alert = True
def test_baidu_search(self):
driver = self.driver
driver.get(self.base_url + "/")
driver.find_element_by_id("kw").send_keys("selenium webdriver")
driver.find_element_by_id("su").click()
time.sleep(2)
driver.close()
def test_baidu_set(self):
driver = self.driver
driver.get(self.base_url+"/gaoji/preferences.html")
driver.find_element_by_xpath('//*[@id="nr"]/option[3]').click()
time.sleep(2)
driver.find_element_by_xpath("//*[@id='save']").click()
time.sleep(2)
driver.switch_to_alert().accept()
def tearDown(self):
self.driver.quit()
self.assertEqual([],self.verificationErrors)
if __name__ == '__main__':
testunit = unittest.TestSuite()
testunit.addTest(Baidu('test_baidu_search'))
testunit.addTest(Baidu('test_baidu_set'))
runner.run(testunit)
