关于python中unittest框架的简单介绍

今天的主人公unittest，是python的单元测试框架，也称为pyUnit。它是通过test fixture, test case, test suite, test runner来实现自动化测试。

我觉得只有理解了这几个概念，才能真正的理解单元测试的基本原理，下面就主要围绕这几个概念来展开这篇文章。

首先来看一下这几个概念，以及他们之间的关系，他们是如何在一起工作的：

一个TestCase的实例就是一个测试用例。就是一个完整的测试流程，包括测试前准备环境的搭建(setUp)，执行测试代码(run)，以及测试后环境的还原(tearDown)。单元测试(unit test)的本质也就在这里，一个测试用例是一个完整的测试单元，通过运行这个测试单元，可以对某一个问题进行验证。
而多个测试用例集合在一起，就是TestSuite，而且TestSuite也可以嵌套TestSuite。
TestLoader是用来加载TestCase到TestSuite中的，其中有几个loadTestsFrom__()方法，就是从各个地方寻找TestCase，创建它们的实例，然后add到TestSuite中，再返回一个TestSuite实例。
TextTestRunner是来执行测试用例的，其中的run(test)会执行TestSuite/TestCase中的run(result)方法。
测试的结果会保存到TextTestResult实例中，包括运行了多少测试用例，成功了多少，失败了多少等信息。

这样整个流程就清楚了，首先是要写好TestCase，然后由TestLoader加载TestCase到TestSuite，然后由TextTestRunner来运行TestSuite，运行的结果保存在TextTestResult中，整个过程集成在unittest.main模块中。

下面来看一段实践代码：

测试用例login.py：

class Login(unittest.TestCase):
def __init__(self, testname, driver):
super(Login, self).__init__(testname)
self.driver = driver

def setUp(self):
self.driver.get(“http://” + imon_conf.IMON_IP + “/”)

def test_login(self):
try:
driver = self.driver
self.assertEqual(u”普华基础软件股份有限公司　　版权所有”, driver.find_element_by_css_selector(“div.footer”).text, “未找到普华公司标识”)
except AssertionError,msg:
print msg
driver.find_element_by_id(“username”).clear()
driver.find_element_by_id(“username”).send_keys(imon_conf.IMON_NAME)
driver.find_element_by_id(“password”).clear()
driver.find_element_by_id(“password”).send_keys(imon_conf.IMON_PASSWD)
driver.find_element_by_id(“enter”).click()
try:
self.assertTrue(driver.find_element_by_id(“alarm”), “未从页面找到告警图标”)
print “系统登录成功”
except AssertionError,msg:
print msg
print “登录系统失败”

执行测试用例代码runtest.py:

def all_tests(driver):
suite = unittest.TestSuite()
suite.addTest(Login(“test_login”, driver))
return suite

if __name__ == “__main__”:
suite = all_tests(driver)
runner = unittest.TextTestRunner()
runner.run(suite)

通过以上栗子，总结unittest的基本使用方法几点：
1.import unittest
2.定义一个继承自unittest.TestCase的测试用例类。
3.定义setUp和tearDown，在每个测试用例前后做一些辅助工作。
4.定义测试用例，名字以test开头。
5.一个测试用例应该只测试一个方面，测试目的和测试内容应很明确。主要是调用assertEqual、assertRaises等断言方法判断程序执行结果和预期值是否相符。
6.将用例添加到TestSuite，再用TextTestRunner启动测试。

unnitest是Python语言下的单元测试框架，可能大家会有疑问，这与自动化测试工具selenium有什么关系，在selenium IDE中我们录制的脚本可以通过Export Test Case As导出Python 2/unittest/WebDriver形式，如下图所示，所以学习unittest可以更好的修改和完善selenium IDE录制的脚本。

之前有同事分享过unittest单元框架的介绍，请结合其他人写的介绍，查看下面的实例，例子是演示DLMMCP系统登录的unittest单元测试，其中具体的细节也在注释中表明了，请查看。

#!/usr/bin/python
# -*- coding: utf-8 -*-
#############################################################
#操作：通过unittest单元测试框架测试DLMMCP系统登录
#version:V1.0
#author:ZLL
##############################################################
 
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
from selenium.common.exceptions import NoAlertPresentException
import unittest, time, re
 
class UnittestLogin(unittest.TestCase):
 
'''
    初始化。
''' 
    def setUp(self):
        self.driver = webdriver.Firefox()
        self.driver.implicitly_wait(30)
        self.base_url = "http://192.168.39.141:8080/"
        self.verificationErrors = []    #脚本运行时的错误信息将被记录到这个数组中
        self.accept_next_alert = True   #是否继续接受下一个警告，初始状态为True
 
'''
    登录脚本。
'''   
    def test_unittest_login(self):
        driver = self.driver
        driver.get(self.base_url + "dlmmcp/index.action")
        driver.find_element_by_id("username").clear()
        driver.find_element_by_id("username").send_keys("yjs")
        driver.find_element_by_id("password").clear()
        driver.find_element_by_id("password").send_keys("yjs12345678")
        driver.find_element_by_id("enter").click()
        time.sleep(3)
 
'''
    用于查找页面元素是否存在，通过find_element()来接收元素的定位方法和定位值，
    如果定位到元素，则返回为True，否则抛出异常并返回false。
'''    
    def is_element_present(self, how, what):
        try: self.driver.find_element(by=how, value=what)
        except NoSuchElementException as e: return False
        return True
 
'''
    用于判断当前页面是否存在警告框，利用WebDriver提供的switch_to_alert()方法来
    捕捉页面上的警告框。如果捕捉到警告框则返回True，否则抛出异常并返回false。
'''      
    def is_alert_present(self):
        try: self.driver.switch_to_alert()
        except NoAlertPresentException as e: return False
        return True
 
 '''
    关闭警告并获得警告信息。通过switch_to_alert()获得警告，通过text获得警告框信息。
    通过if语句判断accept_next_alert的状态，在setUP()中已经初始化状态为True，如果
    为True，如果为True，则通过accept()接受，否则dismiss()忽略此警告。
'''   
    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()

关于python中unittest框架的简单介绍

猜你喜欢