引入unittest测试框架

通过seleinium IDE 完成脚本的录制之后，可以将其导出为加了python unittest 单元测试框架的相应脚本。如下图所示：

将脚本导出，保存为baidu.py ，通过python IDLE 编辑器打开。如下：

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.support.ui import Select
from selenium.common.exceptions import NoSuchElementException
import unittest, time, re

 

class Baidu(unittest.TestCase):
    def setUp(self):
       self.driver = webdriver.Firefox()
       self.driver.implicitly_wait(30)
       self.base_url = "http://www.baidu.com/"

       self.verificationErrors = []
       self.accept_next_alert = True

 

   def test_baidu(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()
       driver.close()

   def is_element_present(self, how, what):
       try: self.driver.find_element(by=how, value=what)
       except NoSuchElementException, e: return False
       return True

   def is_alert_present(self):
       try: self.driver.switch_to_alert()
       except NoAlertPresentException, e: return False
       return True

   def close_alert_and_get_its_text(self):
     try:
         alert = self.driver.switch_to_alert()
         alert_text = alert.text
         if self.accept_next_alert:
             alert.accept()
         else:
             alert.dismiss()
         return alert_text
         finally: self.accept_next_alert = True

     def tearDown(self):
         self.driver.quit()
         self.assertEqual([], self.verificationErrors)

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

 

加入unittest 框架后，看上去比我们之前见的脚本复杂了很多，除了中间操作浏览器的几行是我们所熟悉的脚本，其它的方法都是做什么用的？我们来简单分析一下！

import unittest
首先要引入unittest 框架包。

 

class Baidu(unittest.TestCase):

Baidu 类继承unittest.TestCase 类，从TestCase 类继承是告诉unittest 模块的方式，这是一个测试案例。

 

def setUp(self):
self.driver = webdriver.Firefox()
self.base_url = "http://www.baidu.com/"

setUp 用于设置初始化的部分，在测试用例执行前，这个方法中的函数将先被调用。这里将浏览器的调用和URL 的访问放到初始化部分。

 

self.verificationErrors = []

脚本运行时，错误的信息将被打印到这个列表中。

 

self.accept_next_alert = True
是否继续接受下一个警告。

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

test_baidu 中放置的就是我们的测试脚本了，这部分我们并不陌生；因为我们执行的脚本就在这里。

 

def is_element_present(self, how, what):
       try: self.driver.find_element(by=how, value=what)
       except NoSuchElementException, e: return False
       return True

is_element_present 函数用来查找页面元素是否存在， try...except.... 为python 语言的异常捕捉。
is_element_present 函数在这里用处不大，通常删除，因为判断页面元素是否存在一般都加在testcase 中。

 

def is_alert_present(self):
     try: self.driver.switch_to_alert()
     except NoAlertPresentException, e: return False
     return True

对弹窗异常的处理

def close_alert_and_get_its_text(self):
try:
        alert = self.driver.switch_to_alert()
        alert_text = alert.text
        if self.accept_next_alert:
           alert.accept()
        else:
           alert.dismiss()
        return alert_text
finally: self.accept_next_alert = True

关闭警告以及对得到文本框的处理，if 判断语句前面已经多次使用，并不陌生；try....finally...为python的异常处理。

 

def tearDown(self):
     self.driver.quit()
     self.assertEqual([], self.verificationErrors)

tearDown 方法在每个测试方法执行后调用，这个地方做所有测试用例执行完成的清理工作，如退出浏览器等。

 

self.assertEqual([], self.verificationErrors)

这个是难点，对前面verificationErrors 方法获得的列表进行比较；如查verificationErrors 的列表不为空，输出列表中的报错信息。

 

if __name__ == "__main__":
unittest.main()
unitest.main()函数用来测试类中以test 开头的测试用例

 

这样一一分析下来，我们对unittest 框架有了初步的了解。运行脚本，因为引入了unittest 框架，所以控制台输出了用例的执行个数、时间以及是否OK 等信息。

 

>>> ========================= RESTART ================================
>>>
.
----------------------------------------------------------------------
Ran 1 test in 10.656s
OK

>>>

本节只是初步对unittest 的框架进行了分析，还有许细节将放在后面章节进行讨论，例如python 的异常处理机制等。