Python单元测试框架unittest

学习接口自动化测试时接触了unittest单元测试框架，学习时参照了虫师编写的《selenium2自动化测试实战》，个人觉得里面讲的例子还比较容易理解的。

一、基础

1、main()和框架的简单使用，例：

import unittest   #导包
import calculator import Count
class TestCount(unittest.TestCase):  #继承unittest
　　def setUp(self):
　　　　print("test start")
　　def test_add(self):
　　　　i = Count(2,3)
　　　　self,assertEqual(j.add(),5)
　　def tearDown(self):
　　　　print("test end")
if __name__=='__main__':
　　unittets.main()

 上面的代码，先引入unittest模块，创建TestCout类继承unittest的TestCase类

assertEqual()是unittest框架提供的断言方法，判断两者是否相等

unittest提供了全局的main（）方法。可以将一个单元测试模块变成可以直接运行的测试脚本。

main()方法使用testLoader类来搜索所有包含在该模块总以“test”命名开头的测试方法，并自动执行他们

备注：setup和teardown和obotframework里的初始化清除类似。

2、执行测试集

if    __name__ == '__main__':
#构造测试集
　　suite=unittest(TestAdd("test_add"))
　　suite.addTest(TestAdd("test_add2"))
　　suite.addTest(TestSub("test_sub"))  #TestSub是要被执行测试用例的类名，test_sub方法名

#运行测试集合
8　　 runner = unittest.TestTestRunner()
9　　 runner.run(suite)

通过TestSuite类的addTest（）方法把不同测试类中的测试方法组装到测试套件中

3、常用断言方法

assertEqual(a,b)   #a==b
assertNotEqual(a,b)     #a!=b
asserTrue(x)        #bool(x)is True
assertFalse(x)        #bool(x) is False
assertIs(a,b)        #a is b
assertIsNot(a,b)    #a is not b
assertIsNone(x)    #x is None
assertIn(a,b)        #a in b 
assertNotIn(a,b)    #a not in b
assertIsInstance(a,b)    #isinstance(a,b)
assertNotIsInstance(a,b)    #not isinstance(a,b)

4、跳过测试

在运行测试时，有时需要跳过某些测试用例，或当用例复合某个条件时跳过测试，或直接将测试用例设置为失败。unittest提供了实现这些需求的装饰器

unittest.skip(reason)    #无条件地跳过装饰的测试，说明跳过的原因
unittest.skipIf(condition,reason)   #如果条件为真，跳过装饰的测试
unittest.skipUnless(condition,reason)   #除非条件为真，跳过装饰的测试
unittest.expectedFailure()    #不管执行直接是否失败，都标记为失败

例：

@unittest.skip("直接跳过测试")
def test_skip(self):
 3　　 print("test")

@unittest.skip(3>2,“当条件为True时跳过测试”)
def test_skip_if(self):
　　print("test111")

@unittest.skipUnless(3>2,"当条件为True时，执行测试")
def    test_skip_unless(self):
　　print("test22")

@unittest.expected_Failure
def    test_expected_failure(self):
　　assertEqual(2,3)

if __name__=='__main__':
18　　 unittest.main()

5、discover（）

a、TestLoader()

该类负责根据各自标准加载测试用例，并将他们返回给测试套件。unittest提供了可以共享的defaultTestLoader类，可以使用其子类和方法创建实例，比如discover（）

b、discover(start_dir,pattern='test*.py',top_level_dir=None)

作用：找到指定目录下所有测试模块，并可递归查到子目录下的测试模块，只有匹配到文件名才能被加载。如果启动的不是顶层目录，那么顶层目录必须单独指定 

start_dir:要测试的模块名或测试用例目录

pattern='test*.py':匹配文件名以“test"开头的”.py"类型的文件，“*”表示任意多个字符

top_level_dir=None:测试模块的顶层目录，如果没有顶层目录，默认为None

示例：

import unittest

#定义测试用例的目录为当前目录
test_dir='./'
discover=unittest.defaultTestLoader.discover(test_dir,pattern='test*.py')

if __name__=='__main__':
　　runner=unittest.TextTestRunner()
　　runner.run(discover)

discover()方法会自动根据测试目录（test_dir）匹配查找测试用例（test*.py)，并将查找到的测试用例组装到测试条件中。

 

二、unittest数据参数化

unittest本身不支持参数化，但是可以借助其他方法

1、unittest中使用第3方插件parameterized进行参数化

首先在DOS中键入以下命令安装parameterized
python -m pip install parameterized

导包：
from parameterized import parameterized

例：

import unittest
from pylibrary.PyLib import *

class   Interface_Report2(unittest.TestCase):

#开始使用parameterized
    @parameterized.expand([   #引用装饰器，准备参数，传入下方的函数
        ("2018-09-26", '2018-09-26', "测试", "2001",49,'TotalCallNum'),
        ("2018-09-26", '2018-09-26', "测试", "2001",60,'TotalCallAnsweredNum'),
        ("2018-09-26", '2018-09-26', "测试", "2001",60,'RingNum'),
        ("2018-09-13", '2018-09-13', "测试", "2001",4,'TotalCallNum_Transfer'),
    ])


    def test_Agreport1(self,start_time,end_time,template_name,ag_num,testData,reportData):   #当执行到此会自动接收上面传入的参数
        self.num = testData
        self.report_num1 = PyLib().getAgReport(start_time,end_time,template_name,ag_num)
        self.report_num=self.report_num1[0][reportData]
        self.assertEqual(self.num,self.report_num)

if  __name__ == '__main__':

    unittest.main(verbosity=2)

#verbosity表示测试结果的信息复杂度，分别使用0,1,2，其具体的含义如下：
#0（静默模式）：仅仅获取总的测试用例数以及总的结果
#1（默认模式）：此时在每个成功的用例前面有个’.’,每个失败的用例前面有个’F’
#2(详细模式)：测试结果会显示每个测试用例的所有相关信息

2、unittest的ddt库

pip install ddt

首先创建一个数据文件：test_ddt_file.json

{
    "test_Agreport1_01":{
        "start_time":"2018-09-26",
        "end_time":"2018-09-26",
        "template_name": "测试",
        "ag_num":"2001",
        "testData":49,
        "reportData":"TotalCallNum"
    },
    "test_Agreport1_02":{
        "start_time":"2018-09-26",
        "end_time":"2018-09-26",
        "template_name": "测试",
        "ag_num":"2001",
        "testData":60,
        "reportData":"TotalCallAnsweredNum"
    },
    "test_Agreport1_03":{
        "start_time":"2018-09-26",
        "end_time":"2018-09-26",
        "template_name": "测试",
        "ag_num":"2001",
        "testData":60,
        "reportData":"RingNum"
    },
    "test_Agreport1_04":{
        "start_time":"2018-09-13",
        "end_time":"2018-09-13",
        "template_name": "测试",
        "ag_num":"2001",
        "testData":4,
        "reportData":"TotalCallNum_Transfer"
    }
}

创建测试用例：

import unittest
from pylibrary.PyLib import *
from ddt import ddt, file_data


@ddt
class   Interface_Report2(unittest.TestCase):

    @file_data("D:\\Python_code\\test_ddt_file.json")
    def test_Agreport1(self,start_time,end_time,template_name,ag_num,testData,reportData):
        self.num = testData
        self.report_num1 = PyLib().getAgReport(start_time,end_time,template_name,ag_num)
        self.report_num=self.report_num1[0][reportData]
        self.assertEqual(self.num,self.report_num)



if  __name__ == '__main__':

    unittest.main(verbosity=2)

 三、命令行运行

命令行运行我使用的比较少，一般用pycharm直接就运行了。以下作为以后参考使用吧

 

通过命令行运行：模块、类、或者单个测试方法，具体的操作是如下：

运行测试模块：python -m unittest test_module1 test_module2

运行测试类：python -m unittest test_module.TestClass

运行测试方法：python -m unittest test_module.TestClass.test_method

可以在一个列表中添加需要运行的模块名、类名、方法名。

可以通过使用-v参数获取更详细的测试信息：python -m unittest -v test_module

可以通过-h参数查看命令行所有参数：python -m unittest -h

命令行参数：

-b --buffer

在测试运行期间标准输出流和标准错误流被缓存。输出通过测试信息。

输出在测试失败或错误时正常回显，并添加到失败消息中。

-c --catch

Ctrl+C在测试运行期间等待当前测试结束，然后报告到目前为止的所有结果。

第二个Ctrl+C会引发KeyboardInterrupt异常。

-f --failfast

在第一个错误或故障时停止测试运行。

命令行也可用于测试发现(Test Discovery)，用于运行项目中的所有测试或仅用于子集。

Unittest支持简单的测试发现(Test Discovery)。为了与测试发现兼容，

所有测试文件必须是可从项目的根目录导入的模块或包（这意味着它们的文件名必须是有效的标识符）。

Test Discovery是通过TestLoader.discover()实现，也可以通过命令行实现。

基本的命令行用法：

cd project_directory #进入项目目录根目录

python -m unittest discover #执行命令

discover子命令的参数如下：

-v , --verbose

详细输出

-s , --start-directory directory

执行发现的起始目录(directory)，默认是当前目录(.)

-p, --pattern pattern

匹配测试文件的模式(pattern)，默认是test*.py

-t, --top-level-directory directory

项目(directory)的目的的根目录(默认是起始目录)

-s、-p、-t命令可以在一个命令行中联合使用。下面两个命令是等价的：

python -m unittest discover -s project_directory -p "*_test.py"

python -m unittest discover project_directory "*_test.py"

作为一个路径，可以传递一个包名称，例如myproject.subpackage.test作为开始目录。

然后将导入提供的软件包名称，并将其在文件系统上的位置用作开始目录。

注意：测试发现通过导入来加载测试。一旦测试发现已经从指定的开始目录中找到所有测试文件，

它将路径转换为要导入的包名称。例如foo / bar / baz.py将导入为foo.bar.baz。

如果有全局安装的软件包，并尝试在软件包的不同副本上进行测试发现，

那么导入可能发生在错误的位置。如果发生这种情况，测试发现将警告并退出。

如果提供开始目录作为包名称而不是目录的路径，则发现假定它从其导入的任何位置是想要的位置，因此不会得到警告。

测试模块和包可以通过load_tests协议自定义测试加载和发现。

'''

import os

#进入到TestHello.py路径下

os.chdir(".//TestSuit_01")

print os.getcwd()

print '''可以通过命令行测试信息'''

#运行测试模块TestHello.py

os.system("python -m unittest TestHello")

#运行测试类TestHello

os.system("python -m unittest TestHello.TestHello")

#运行测试方法

os.system("python -m unittest TestHello.TestHello.test_Hello")

print "#"*50

print

print '''可以通过使用-v参数获取更详细的测试信息'''

#运行测试模块TestHello.py

os.system("python -m unittest -v TestHello")

#运行测试类TestHello

os.system("python -m unittest -v TestHello.TestHello")

#运行测试方法

os.system("python -m unittest -v TestHello.TestHello.test_Hello")

print "#"*50

print

print '''可以通过-h参数查看命令行所有参数：python -m unittest -h'''

os.system("python -m unittest -h")

print "#"*50

print

print "命令行参数：-b -c -f"

#运行测试模块TestHello.py

os.system("python -m unittest -b TestHello")

#运行测试类TestHello

os.system("python -m unittest -c TestHello.TestHello")

#运行测试方法

os.system("python -m unittest -f TestHello.TestHello.test_HelloType")

print "#"*50

print

#使用测试发现命令执行所有测试

print "使用测试发现执行项目中的所有测试"

#进入到UnttestPro

os.chdir("../")

print "The current path:",os.getcwd()

#全部使用默认的参数