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

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=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)#运行用例集,用例执行的顺序是按照字母排序的

suite=unittest.TestSuite()#定义用例集，是个list
suite.addTest(unittest.makeSuite(Testcase))#将类TestCase先变成一个用例集合，然后再加入到suite中
report=bf(suite)##先实例化这个类
report.report(description='用例描述',filename='用例文件名字')
#description必填参数,log_path是非必写的，默认在当前目录下。也可以定义别的目录
#filename可写可不写，回自动生成个名字。名字后缀默认就是html。可写上也可不写

- #用减号-区分不同的用例。最终是个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.否则断言失败并给出错误信息msgsuite=unittest.TestSuite()
suite.addTest(unittest.makeSuite(Login))
run=bf(suite)
run.report('login_test','登录测试用例')#第一个是描述，第二个是文件名
print(run.success_count)#通过的次数
print(run.failure_count)#失败次数)