单元测试unittest（基于数据驱动的框架：unittest+HTMLTestRunner/BeautifulReport+yaml+ddt）

一、定义

unittest单元测试框架不仅可以适用于单元测试，还可以适用WEB自动化测试用例的开发与执行，该测试框架可组织执行测试用例，并且提供了丰富的断言方法，判断测试用例是否通过，最终生成测试结果。

unittest通过建立类并继承父类unittest.TestCase将用例组织起来，统一执行

示例：

import unittest

class Testcase(unittest.TestCase):#类继承unittest.TestCase

    def setUp(self):#每条用例运行前，运行。一般可用来做些准备工作

        print('我是setup，什么时候运行')

    def tearDown(self):#每条用例结束后，运行，一般可用来做些收尾工作

        print('teardown什么时候运行')

    def testa(self):

        print('测试用例a')

        self.assertEqual(1,2,'计算结果不正确')

        #assertEqual是父类unittest.TestCase的函数，用来验证实际结果是否与预期结果相等

       #第一个参数是预期结果，第二个参数是实际结果，第三个参数是当前面两个不相等时，回显示的提示信息。第三个参数是非必填的

    def testb(self):

        '''这里是测试用例的描述，会最终显示在测试报告中的，必须用三个单引号括起来才可以，用#d不行'''

        print('测试用例b')

        self.assertEqual(1,1)

    def c(self):

        print('c函数不是以test开头的，所以不会当做用例被执行')

        # self.assertIn('a','abc')

    @classmethod

    def setUpClass(cls):#所有的用例运行前，运行setupClass

        print('什么时候运行setupClass')

    def tearDownClass(cls):#所有用例运行后，执行tearDownClass

        print('什么时候运行tearDownClass')

unittest.main()#执行当前python文件中的所有用例，会在控制台显示运行结果，不会产生测试报告

#运行的时候不要右键运行run unittest for...，这个是pycharm自己给运行的unittest,不会产生测试报告

# 应点击菜单栏中Run--run,弹出提示框后，选择自己要运行的文件进行run

#函数名用例必须以test开头，才会被执行

二、用例集suite+HTMLTestRunner/BeautifulReport

用例集即测试套件，可以吧所有的用例放入用例集当中执行，并结合HTMLTestRunner或者BeautifulReport产生出直观的漂亮的测试报告

#-------------------------------------------------------------------HTMLTestRunner写报告（丑）
#将下载好的HTMLTestRunner.py 放到 pycharm的External Libraries目录下的lib目录下，然后导入：import HTMLTestRunner,紧接着上面的代码：（把unittest.main注释掉）

suite=unittest.TestSuite()#定义用例集，是个list

suite.addTest(unittest.makeSuite(Testcase))#将类TestCase先变成一个用例集合，然后再加入到suite中

f=open('测试报告.html','wb')#建立一个测试报告文件

runner=HTMLTestRunner.HTMLTestRunner(f,title='测试报告标题',description='测试报告描述')#title和decription都是非必填。将用例结果写入测试报告.html

runner.run(suite)#运行用例集,用例执行的顺序是按照字母排序的

然后通过run-run选中要执行的当前文件，运行后，就会产生一个html格式的报告，可用浏览器打开，如下：

#--------------------------------------------------------BeautifulReport 报告（美）

#下载BeautifulReport ,解压后运行setup.py。可以将解压后的文件拷贝到pycharm中，然后运行setup.py也可。然后要导入：

from BeautifulReport import BeautifulReport as bf #bf就是个别名，方便使用

suite=unittest.TestSuite()#定义用例集，是个list

suite.addTest(unittest.makeSuite(Testcase))#将类TestCase先变成一个用例集合，然后再加入到suite中

report=bf(suite)##先实例化这个类

report.report(description='用例描述',filename='用例文件名字')

#description必填参数,log_path是非必写的，默认在当前目录下。也可以定义别的目录

#filename可写可不写，回自动生成个名字。名字后缀默认就是html。可写上也可不写

执行后，生成html格式报告：打开：

三、用unittest测接口示例

1、接口数据：yaml格式文件，存有3条用例数据

2、用ddt自动读取用例数据并传递给测试类中的测试用例函数

3、执行用例

4、写报告

用例数据

- #用减号-区分不同的用例。最终是个list,然后每个list是个字典:  [{}{}]

  url: /api/user/login

  detail: 正常登录

  method: post

  data:

    username: aaa

    passwd: aA123456

  check:

    - userId

    - error_code

-

  url: /api/user/login

  detail: 密码错误

  method: post

  data:

    username: aaa

    passwd: aA1234

  check:

    - 密码错误

-

  url: /api/user/login

  detail: 必填参数未填，用户名不传

  method: post

  data:

    passwd: aA1234

  check:

    - 必填参数未填

接口测试代码：获取数据，执行测试用例并发送测试结果报告

import unittest

import  ddt#自动读取文件内容,该模块需要先安装pip install ddt

import requests

from BeautifulReport import BeautifulReport as bf

from urllib import parse

@ddt.ddt #声明这个类要用ddt

class Login(unittest.TestCase):

    base_url='http://100.20.20.40'

    @ddt.file_data('login1.yaml')#ddt自动读取文件，并获取内容传给下面的函数，循环调用.运行如果出现字符编码问题，就直接打开file_data源文件，搜索open，将打开文件的代码中，加上encoding=utf-8

    def test_request(self,**kwargs):#**kwargs可以传入字典参数，不限个数.此处用来接收从login1.yaml中读出来的用例，是字典

        detail=kwargs.get('detail','没写用例描述')#如果没有定义detail，默认给个值’没写用例描述‘

        self._testMethodDoc=detail#设置用例描述

        url=kwargs.get('url')

        url=parse.urljoin(self.base_url,url)#自动拼接url,

        method=kwargs.get('method','get')#method默认给个get方法

        data=kwargs.get('data',{})#如果没给data,默认给个空字典

        header=kwargs.get('header',{})#如果没有header参数，默认给个空字典

        cookie=kwargs.get('cookie',{})

        check=kwargs.get('check')#获取login1.yaml中的预期结果check,是list

        method=method.lower()#

        try:

            if method=='get':

                res=requests.get(url,params=data,cookies=cookie,headers=header).text

                #因为实际结果失败的话，不会返回json串，所以不能转成字典

            else:

                res=requests.post(url,data,cookies=cookie,headers=header).text

        except Exception as e:

            print('接口请求出错')

            res=e

        for c in check:#循环遍历check中的所有预期结果，判断是否包含在实际结果中

            self.assertIn(c,res,msg='预期结果不符，预期结果%s,实际结果%s'%(c,res))#断言，如果预期结果在实际结果中，pass.否则断言失败并给出错误信息msg

suite=unittest.TestSuite()

suite.addTest(unittest.makeSuite(Login))

run=bf(suite)

run.report('login_test','登录测试用例')#第一个是描述，第二个是文件名

print(run.success_count)#通过的次数

print(run.failure_count)#失败次数)

单元测试unittest（基于数据驱动的框架：unittest+HTMLTestRunner/BeautifulReport+yaml+ddt）的相关教程结束。