Python Unittest 自动化单元测试框架Demo

1. python 测试框架(本文只涉及 PyUnit)
   
   https://wiki.python.org/moin/PythonTestingToolsTaxonomy

2. 环境准备
   
   首先确定已经安装有Python，之后通过安装PyUnit，Python版本比较新的已经集成有PyUnit(PyUnit 提供了一个图形测试界面UnittestGUI.py)
   
   参考： https://wiki.python.org/moin/PyUnit

3. 代码实例
   
   使用的IDE为 PyCharm,DEMO结构如图
   
   

   ---

   
   
   1.简单地一个实例

   
   # Test002_Fail.py
   
   # -*- coding:utf-8 -*-
   
   import  unittest
   
   class Test002_Fail(unittest.TestCase):
   
       #测试用例前执行
       def setUp(self):
           print 'Case Before'
           pass
   
       #测试用例后执行
       def tearDown(self):
           print 'Case After'
           pass
   
       #测试用例1
       def test_Case1(self):
           a = 3
           b = 2
           self.assertEqual(a+b,4,'Result Fail')
   
       #测试用例2
       def test_Case2(self):
           a = 2
           b = 3
           self.assertEqual(a*b,7,'Result Fail')
   
   if __name__ == '__main__':
       unittest.main()
   

   ---

   
   
   2.当case比较多的时候，可以使用Suite来管理多个case,suite 可以看做是一个case 的容器，可以装很多的case

   
   # SuiteDemo.py
   
   # -*- coding:utf-8 -*-
   
   #一个类，添加测试集
   
   import  unittest
   
   class SuiteDemo(unittest.TestCase):
   
       #测试用例前执行
       def setUp(self):
           print 'Case Before'
           pass
   
       #测试用例后执行
       def tearDown(self):
           print 'Case After'
           pass
   
       def test_Case1(self):
           a = 3
           b = 2
           self.assertEqual(a+b,5,'Result Fail')
           print 'Case1'
   
       def test_Case2(self):
           a = 2
           b = 3
           self.assertEqual(a*b,6,'Result Fail')
           print 'Case2'
   
   #定义一个测试集合，方便添加Case
   
   def suite():
       suiteTest = unittest.TestSuite()
       suiteTest.addTest(SuiteDemo("test_Case1"))
       suiteTest.addTest(SuiteDemo("test_Case2"))
       return suiteTest
   
   #默认运行时通过 Suite 运行
   
   if __name__ == '__main__':
       unittest.main(defaultTest='suite')
   

   ---

   
   
   3.Suite 和 Case可以分开写在不同的Python文件中，这样方便区分case模块和Suite管理模块，在测试测试的时候只需要在case模块里面添加case,之后添加到Suite模块即可了

   case模块

   
   # TestCaseDemo
   
   # -*- coding:utf-8 -*-
   
   import unittest
   
   class TestCaseDemo(unittest.TestCase):
       def setUp(self):
           pass
   
       def tearDown(self):
           pass
   
       def test_Case1(self):
           print 'test_Case1'
   
       def test_Case2(self):
           print 'test_Case2'
   
       def test_Case3(self):
           print 'test_Case3'
   
       def test_Case4(self):
           print 'test_Case4'
   

   接下来为三个Suite模块，都是先导入Case模块，之后添加Suite,再在Suite里面添加Case，最后执行的时候需要使用TestRunner,这个TestRunner 相当于Suite 的执行器

   简单地TestSuite:

   
   # TestCase_Demo1.py
   
   # -*- coding:utf-8 -*-
   
   #集合写入一个方法，main调用并启动
   
   import unittest
   from TestCase_Demo import TestCaseDemo
   
   #添加一个测试集合，并添加Case
   
   def suite():
       suiteTest = unittest.TestSuite()
       suiteTest.addTest(TestCaseDemo('test_Case1'))
       return suiteTest
   
   #指定并启动测试集合,运行集合方法
   
   if __name__ == '__main__':
   
       runner = unittest.TextTestRunner()
       runner.run(suite())
   

   也可以不定义Suite方法，直接在main方法中进行Case的添加，之后执行

   
   # TestCase_Demo2.py
   
   # -*- coding:utf-8 -*-
   
   #集合可以写在main
   
   import unittest
   from TestCase_Demo import TestCaseDemo
   
   #指定并启动测试集合
   
   if __name__ == '__main__':
   
       #添加测试集合Case,并启动
       suiteTest = unittest.TestSuite()
       suiteTest.addTest(TestCaseDemo('test_Case1'))
       suiteTest.addTest(TestCaseDemo('test_Case2'))
   
       #直接启动集合
       runner = unittest.TextTestRunner()
       runner.run(suiteTest)

   可以把不同的Case 添加道不同的Suite里面，同样，可以给Suite分组，用于区分不同的Case,并且，Suite 之间还可以包含

   
   # TestCase_Demo3.py
   
   # -*- coding:utf-8 -*-
   
   #包含集合，多集合
   
   import unittest
   from TestCase_Demo import TestCaseDemo
   
   #添加不同的合集
   
   def Suite1():
       suiteTest = unittest.TestSuite()
       suiteTest.addTest(TestCaseDemo("test_Case1"))
       suiteTest.addTest(TestCaseDemo("test_Case2"))
       print 'Suite1 运行'
       return suiteTest
   
   def Suite2():
       suiteTest = unittest.TestSuite()
       suiteTest.addTest(TestCaseDemo("test_Case3"))
       suiteTest.addTest(TestCaseDemo("test_Case4"))
       print 'Suite2 运行'
       return suiteTest
   
   #包含所有的Suite
   
   def AllSuite():
       allTest = unittest.TestSuite((Suite1(),Suite2()))
       return allTest
   
   #运行的时候，可以根据不同的要求，运行不同的Suite,或者全部运行，这样就方便管理每次运行的case
   
   if __name__ == '__main__':
       runner = unittest.TextTestRunner()
       runner.run(AllSuite())
   

   ---

4. 小结
   
   PyUnit 是Python的一个单元测试框架，使用它可以方便的管理自动化运行的case.
   
   在Python 中，每一个方法 只要方法名使用test_ 开头的都被作为是一个用例，都可以看做成一个case,这里介绍了使用Suite管理Case，并且分组运行Case,还有其他很多的功能，请参考：
   
   https://docs.python.org/2.7/library/unittest.html

版权声明：本文为博主原创文章，未经博主允许不得转载。