Python接口测试之unittest(五)

Test-driven development(TDD)开发模式在今天已经不是什么新奇的事了,它的开发思维是在开发一个产品功能的时候,先编写好该功能的测试代码,在编写开发比如,比如要写二个数相除的函数,那么它的测试代码应该为:

#!/usr/bin/env python
#coding:utf-8

import  unittest
  
class TestDiv(unittest.TestCase):    
    def setUp(self):       
        pass
      def tearDown(self):        
        pass
      def test_001(self):
        self.assertEqual(div(1,1),1)    
      def test_002(self):
        self.assertRaises(ZeroDivisionError,div,1,0)

if __name__=='__main__':
    unittest.main(verbosity=2)

执行如上的代码,会提示如下NameError: global name 'div' is not defined的错误信息,事实上我们自己也是非常明白,因为我们实际没有实现这样的一个函数,而是先写了改函数功能的测试代码,那么现在来写函数部分,见完善后的源码:

#!/usr/bin/env python
#coding:utf-8

def div(a,b):    
    return a/b

import  unittest
  
class TestDiv(unittest.TestCase):    
    def setUp(self):        
        pass
      def tearDown(self):        
        pass
      def test_001(self):
        self.assertEqual(div(1,1),1)    

      def test_002(self):
        self.assertRaises(ZeroDivisionError,div,1,0)

if __name__=='__main__':
    unittest.main(verbosity=2)

再次执行我们的测试代码,就会通过,见执行的结果结果:

这就是一个测试驱动的过程,关于测试驱动的开发模式以及实战部分,建议看《Python Web开发测试驱动方法》这本书,在里面作者围绕Django的框架,有详细的案例介绍和代码论述。在这里我们只关注unittest框架,这也是本文章要总结的的部分。

不论对于开发还是测试,都离不开单元测试框架,对于开发而言使用单元测试框架,可以编写测试代码来验证验证自己编写的功能是否正确,对于测试而言,使用单元测试框架,可以编写自动化的测试用例,在Python中单元测试框架是Pyunit,即unittest,unittest我一直认为是一个很优秀的单元测试框架,至少我是这样认为在,它是python的标准库,官方详细的地址是:https://docs.python.org/2/library/unittest.html。单元测试支持测试自动化、 共享的安装程序和关闭代码测试、 聚合成集合,测试和报告框架从测试的独立性。单元测试模块提供可以很容易地支持这些素质的一组测试的类。关于unittest测试框架建议可以到官方查看详细的说明以及演示的实例。unittest各个模块的关系为:

在一个完整的单元测试用例中,是包含了测试固件(setUp()和tearDown()),在测试执行的阶段,我们更加愿意使用测试套件(TestSuite())来组织每个测试用例来执行(TestRunner)并得到测试结果(TestReport),什么 是测试固件,在unittest中,setUp()与tearDown()被成为测试固件,某些人称为钩子(仅仅只一个称呼而已),它的主要目标初始化测试用例,执行测试用例后,对测试用例执行的结果做后期的处理,我们再看上面的测试用例,总共是二个测试用例,不管执行那个测试用例,都会执行setUp()和tearDown(),也就是说,在一个测试类中,如果有N个测试用例,在执行该测试类中的测试用例的时候,会执行N次setUp()和tearDown(),我们修改源码并执行来看结果,见源码:

#!/usr/bin/env python
#coding:utf-8

def div(a,b):    
    return a/b

import  unittest
  
class TestDiv(unittest.TestCase):    
    def setUp(self):        
        print u'开始...'

      def tearDown(self):        
        print u'结束...'

      def test_001(self):
        self.assertEqual(div(1,1),1)    

      def test_002(self):
        self.assertRaises(ZeroDivisionError,div,1,0)

if __name__=='__main__':
    unittest.main(verbosity=2)

见执行的结果截图:

依据结果可以看到,执行了二个测试用例,也执行了2次setUp()和tearDown()方法,如果这样你感觉不明显,可以结合selenium的测试框架来看更加直观,见源码:

#!/usr/bin/env python
#coding:utf-8

import  unittest
from selenium import  webdriver
  
class TestDiv(unittest.TestCase):    
    def setUp(self):
            self.driver=webdriver.Firefox()
            self.driver.get('http://www.baidu.com')    

    def tearDown(self):
            self.driver.quit()    
 
    def test_001(self):
            self.assertEqual(self.driver.title,u'百度一下,你就知道')    

    def test_002(self):
          self.assertEqual(self.driver.current_url,
'https://www.baidu.com/')

if __name__=='__main__':
    unittest.main(verbosity=2)

执二后,会看到打开浏览器二次,当然关闭浏览器也是二次,这里不在进行截图了。那么可不可以让测试固件只执行一次了,也就是说在一个测试类中,有N个测试用例,执行这个测试类中的测试用例后,测试固件只执行一次。当然是可以的,unittest提供了这样的解决方案,在这里钩子方法使用的是类方法(关于实例方法,类方法,静态方法不熟悉的建议看下python的OOP部分),我们重构下代码来实现这样的一个过程,见源码:

#!/usr/bin/env python
#coding:utf-8

import  unittest
from selenium import  webdriver
  
class TestDiv(unittest.TestCase):
    @classmethod    
    def setUpClass(cls):
        cls.driver=webdriver.Firefox()
        cls.driver.get('http://www.baidu.com')

    @classmethod   
    def tearDownClass(cls):
        cls.driver.quit()    
 
    def test_001(self):
        self.assertEqual(self.driver.title,u'百度一下,你就知道')    

def test_002(self):
        self.assertEqual(self.driver.current_url,
'https://www.baidu.com/')

if __name__=='__main__':
    unittest.main(verbosity=2)

OK,带着疑问继续触发,在一个测试用例中,测试用例想有顺序的执行该如何实现了,可以使用addTest()方法来实现,也就是说,把需要执行的或者按顺序的添加到测试套件中,见修改后的源码:

#!/usr/bin/env python
#coding:utf-8

import  unittest
from selenium import  webdriver
  
class TestDiv(unittest.TestCase):
    @classmethod    
    def setUpClass(cls):
        cls.driver=webdriver.Firefox()
        cls.driver.get('http://www.baidu.com')

    @classmethod    
    def tearDownClass(cls):
        cls.driver.quit()    
 
    def test_001(self):
        self.assertEqual(self.driver.title,u'百度一下,你就知道')    

    def test_002(self):
        self.assertEqual(self.driver.current_url,
'https://www.baidu.com/')

if __name__=='__main__':
    suite=unittest.TestSuite()
    suite.addTest(TestDiv('test_001'))
    unittest.TextTestRunner(verbosity=2).run(suite)

或者我们可以把代码重构为:

#!/usr/bin/env python
#coding:utf-8

import  unittest
from selenium import  webdriver
  
class TestDiv(unittest.TestCase):
    @classmethod    
    def setUpClass(cls):
        cls.driver=webdriver.Firefox()
        cls.driver.get('http://www.baidu.com')

    @classmethod    
     def tearDownClass(cls):
        cls.driver.quit()    
 
     def test_001(self):
        self.assertEqual(self.driver.title,u'百度一下,你就知道')    

def test_002(self):
        self.assertEqual(self.driver.current_url,'https://www.baidu.com/')

    @staticmethod    
      def suites():
        tests=['test_001','test_002']        
            return unittest.TestSuite(map(TestDiv,tests))
 
if __name__=='__main__':
    unittest.TextTestRunner(verbosity=2).run(TestDiv.suites())

事实上,我个人不赞成使用addTest()方法,来把测试用例依次添加到测试套件中,理由非常简单,因为在一个测试类中,测试用例是非常多的,这样添加或者删除实在是浪费时间,我们可以把源码修改下,来实现执行一个测试,某些用例不执行的可以忽略,使用的方法makeSuite(),见修改后的源码:

#!/usr/bin/env python
#coding:utf-8

import  unittest
from selenium import  webdriver
  
class TestDiv(unittest.TestCase):
    @classmethod    
      def setUpClass(cls):
        cls.driver=webdriver.Firefox()
        cls.driver.get('http://www.baidu.com')

    @classmethod    
     def tearDownClass(cls):
        cls.driver.quit()    

     def test_001(self):
        self.assertEqual(self.driver.title,u'百度一下,你就知道')

    @unittest.skip(u'忽略该测试用例,谢谢!')    
      def test_002(self):
        self.assertEqual(self.driver.current_url,'https://www.baidu.com/')

if __name__=='__main__':
    suite=unittest.TestSuite(unittest.makeSuite(TestDiv))
    unittest.TextTestRunner(verbosity=2).run(suite)

见执行后的截图:

在这里总结的,只是个人喜好,每个人可以根据自己的实际情况来进行,这只是提供了一种选择而已。还有另外一种方法是TestLoader()加载测试类来执行测试类中的所有测试用例,见源码:

#!/usr/bin/env python 
#coding:utf-8

import  unittest
from selenium import  webdriver
  
class TestDiv(unittest.TestCase):
    @classmethod    
      def setUpClass(cls):
        cls.driver=webdriver.Firefox()
        cls.driver.get('http://www.baidu.com')

    @classmethod    
      def tearDownClass(cls):
        cls.driver.quit()    
 
      def test_001(self):
        self.assertEqual(self.driver.title,u'百度一下,你就知道')    

     def test_002(self):
        self.assertEqual(self.driver.current_url,'https://www.baidu.com/')

if __name__=='__main__':
    suite=unittest.TestLoader().loadTestsFromTestCase(TestDiv)
    unittest.TextTestRunner(verbosity=2).run(suite)

在一个测试用例中,会有期望结果这个说法,来验证这个测试用例是通过还是失败,在unittest的测试框架中,也提供了assert,我们先来看python中的断言assert,来修改下源码,看看python实际代码的断言,见源码:

#!/usr/bin/env python
#coding:utf-8

import  unittest
from selenium import  webdriver
  
class TestDiv(unittest.TestCase):
    @classmethod    
      def setUpClass(cls):
        cls.driver=webdriver.Firefox()
        cls.driver.get('http://www.baidu.com')

    @classmethod    
      def tearDownClass(cls):
        cls.driver.quit()  
   
      def test_001(self):        
            assert self.driver.title in u'百度一下,你就知道'

      def test_002(self):        
            assert self.driver.current_url in 'https://www.baidu.com/'

if __name__=='__main__':
    suite=unittest.TestLoader().loadTestsFromTestCase(TestDiv)
    unittest.TextTestRunner(verbosity=2).run(suite)

见执行的结果的截图:

上面的仅仅是python语言自带的assert,在unittest中提供了非常丰富的断言,具体见如下图:

下面就演示几个断言的使用方法,见案例的源码:

#!/usr/bin/env python
#coding:utf-8

import  unittest
from selenium import  webdriver
  
class TestDiv(unittest.TestCase):
    @classmethod    
      def setUpClass(cls):
        cls.driver=webdriver.Firefox()
        cls.driver.get('http://www.baidu.com')

    @classmethod    
      def tearDownClass(cls):
        cls.driver.quit()    
 
      def test_001(self):
        self.assertEqual(self.driver.title, u'百度一下,你就知道')    

      def test_002(self):
        self.assertTrue(self.driver.find_element_by_id('kw').is_enabled()) 
 
     def test_003(self):
        self.assertIsNot(self.driver.current_url,'www.baidu.com')

if __name__=='__main__':
    suite=unittest.TestLoader().loadTestsFromTestCase(TestDiv)
    unittest.TextTestRunner(verbosity=2).run(suite)

见执行后的结果:

unittest的断言是非常丰富的,这里就不在演示了,遇到了不知道,可以到官方查看。

在python中,提供了HTMLTestRunner.py来生成测试报告,把该文件下载后,直接放到C:\Python27\Lib的目录下,

就可以导入该模块使用了,见该实现的代码:

#!/usr/bin/env python
#coding:utf-8

import  unittest 
from selenium import  webdriver 
import  HTMLTestRunner
  
class TestDiv(unittest.TestCase):
    @classmethod    
      def setUpClass(cls):
        cls.driver=webdriver.Firefox()
        cls.driver.get('http://www.baidu.com')

    @classmethod    
      def tearDownClass(cls):
        cls.driver.quit()    
 
      def test_001(self):
        self.assertEqual(self.driver.title, u'百度一下,你就知道')    

      def test_002(self):
        self.assertTrue(self.driver.find_element_by_id('kw').is_enabled()) 

    def test_003(self):
        self.assertIsNot(self.driver.current_url,'www.baidu.com')

if __name__=='__main__':
    suite=unittest.TestLoader().loadTestsFromTestCase(TestDiv)
    runner=HTMLTestRunner.HTMLTestRunner(
        stream=file('testReport.html','wb'),
        title=u'TestReport',
        description=u'测试报告详细信息'
    )
    runner.run(suite)

执行后,会在当前目录下生成testReport.html的测试报告,见该报告的截图:

python的unittest库非常强大,这里只是介绍了一部分,详细的可以到官方继续查看或者关注本人的公众号咨询互相学习。

本文分享自微信公众号 - Python自动化测试(wuya-python)

原文出处及转载信息见文内详细说明,如有侵权,请联系 yunjia_community@tencent.com 删除。

原始发表时间:2017-01-11

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏Petrichor的专栏

shutdown & reboot & last

23630
来自专栏Petrichor的专栏

命名法 的 简洁归纳表

11340
来自专栏Petrichor的专栏

Dataset 列表:机器学习研究

In computer vision, face images have been used extensively to develop face recog...

86710
来自专栏Petrichor的专栏

requirements.txt 介绍 & 快捷生成

  requirements.txt 文件 里面记录了当前程序的所有依赖包及其精确版本号。

16010
来自专栏Petrichor的专栏

主机:各线介绍 & 问题速查表

  在独立组装过八九台主机,遇坑无数后,我写下了《如何自己组装电脑(从配件到整机)来省下一大笔钱》以及《组装台式机遇坑总结》这两篇技术博客。

12110
来自专栏Petrichor的专栏

Adblock:简单强大的广告过滤沙盒

  之前在使用浏览器时,经常会为太多弹窗广告所困。后来师兄教我说可以用一个叫做 AddBlock 的沙盒来过滤掉广告,遂自己上网查阅了相关资料。后来发现一个特别...

23220
来自专栏Petrichor的专栏

个人理解:import scipy语句 为什么不能导入 scipy.misc.imsave 模块

1.4K30
来自专栏Petrichor的专栏

显示器 设置选项 解析

12110
来自专栏Petrichor的专栏

忘记密码时,如何查看之前Chrome浏览器保存的密码

81420
来自专栏Petrichor的专栏

像素、分辨率 区别 & 介绍

举个例子,比如一张图片由100×100个像点组成,那么这张图片的像素数就是1万。但是,如果这张图片边长为1英寸,那么其分辨率就是100点/英寸,也就是100dp...

23020

扫码关注云+社区

领取腾讯云代金券

年度创作总结 领取年终奖励