利用unittest+ddt进行接口测试(二)：使用yaml文件管理测试数据

知道ddt的基本使用方法之后，练习把之前用excel文件来维护的接口测试用例改用unittest+ddt来实现。

这里我选用yaml文件来管理接口参数，开始本来想用json，但是json无法添加注释，可读性不好。

1.向yaml文件中添加数据

下面截图是接口文档中的各个接口，每个接口都有一个固定的序号，所以在设计每个接口的测试数据时，以序号来区分不同接口

yaml文件内容如下，需要注意的是yaml的语法：

(1)键值对用冒号分割，但是冒号后需要加一个空格

(2)禁止使用tab缩进，只能使用空格键；缩进长度没有限制，只要元素对齐就表示这些元素属于一个层级

(3)字符串可以不用引号标注，也可以加引号，如果想把数字变为字符串，加引号即可

(4)使用#表示注释

详情可以参考博客：https://blog.csdn.net/vincent_hbl/article/details/75411243

2.简单demo：python读取yaml文件，取出接口参数

import yaml

fp = open('../dataconfig/信息互动模块接口.yaml', encoding='utf-8')   # 有中文字符的话，加编码格式
testdata = yaml.load(fp)
t = testdata['5.2.1.4']
print(t)

3.完整流程

(1)封装读取yaml文件方法

handle_yaml.py

# coding: utf-8
# author: hmk

import yaml
import os

class HandleYaml:
    def __init__(self, file_path=None):
        if file_path:
            self.file_path = file_path
        else:
            root_dir = os.path.dirname(os.path.abspath('.'))
            # os.path.abspath('.')表示获取当前文件所在目录；os.path.dirname表示获取文件所在父目录；所以整个就是项目的所在路径
            self.file_path = root_dir + '/dataconfig/信息互动模块接口.yaml'  # 获取文件所在的相对路径(相对整个项目)
        # self.data = self.get_data()

    def get_data(self):
        fp = open(self.file_path, encoding='utf-8')
        data = yaml.load(fp)
        return data

if __name__ == '__main__':
    test = HandleYaml()
    p = test.get_data()
    print(p['5.2.1.1'])

(2)封装requests请求方法

# coding: utf-8
# author: Archer

import requests
import json

class RunMethod:
    def post_main(self, url, data, header=None):if header is not None:
             res = requests.post(url=url, data=data, headers=header)
        else:
            res = requests.post(url=url, data=data)
        # print(res.status_code)
        # return res.json()
        return res    #为了方便后面断言，这里不再对服务器响应进行json格式编码

    def get_main(self, url, data=None, header=None):if header is not None:
            res = requests.get(url=url, params=data, headers=header)
        else:
            res = requests.get(url=url, params=data)
            print(res.status_code)
        # return res.json()
        return res

    def run_main(self, method, url, data=None, header=None):if method == 'POST':
            res = self.post_main(url, data, header)
        else:
            res = self.get_main(url, data, header)
        return res
        # return json.dumps(res, indent=2, sort_keys=False, ensure_ascii=False)  # 使用json模块格式化显示结果

(3)一个接口测试用例

# coding: utf-8
# author: Archer

import unittest
import ddt
from base.run_method import RunMethod
from utils.handle_yaml import HandleYaml

get_data = HandleYaml()  # 从yaml文件中取出该接口的参数
params = get_data.get_data()['5.2.1.4']

@ddt.ddt
class Test(unittest.TestCase):
    """加载咨询详情接口"""

    def setUp(self):
        self.url = 'http://localhost:8088/ApprExclusiveInterface/api/enterprise/info/consult/loadDetail.v'
        self.run = RunMethod()

    @ddt.data(*params)
    def test(self, value):
        r = self.run.run_main("GET", self.url, value)
        print(r)
        self.assertTrue(value['assert'] in r.text)

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

(4)利用HTMLTestRunner生成测试报告

run_report.py

# coding: utf-8
# author: hmk

from HTMLTestRunner import HTMLTestRunner
import unittest
import time, os

root_dir = os.path.dirname(os.path.abspath('.'))  # 获取当前文件所在目录的父目录的绝对路径,也就是项目所在路径E:\DDT_Interface
case_dir = root_dir + '/test_case/'  # 根据项目所在路径，找到用例所在的相对项目的路径
print(root_dir)
print(case_dir)
"""定义discover方法"""
discover = unittest.defaultTestLoader.discover(case_dir,
                                               pattern='test*.py',
                                               top_level_dir=None)
"""
1.case_dir即测试用例所在目录
2.pattern='test_*.py' ：表示用例文件名的匹配原则，“*”表示任意多个字符
3.top_level_dir=None：测试模块的顶层目录。如果没顶层目录（也就是说测试用例不是放在多级目录
中），默认为 None
"""

if __name__ == "__main__":
    """直接加载discover"""
    now = time.strftime("%Y-%m-%d %H_%M_%S")
    filename = root_dir + '/report/' + now + '_result.html'  # 定义报告存放路径
    print(filename)
    fp = open(filename, 'wb')
    runner = HTMLTestRunner(stream=fp,
                            title='个人网企业网接口测试报告',
                            description='测试结果如下: ')
    runner.run(discover)
    fp.close()

ok ，unittest+ddt进行接口测试就完成了，还有很多不足，yaml配置文件还可以继续设计优化，例如可以把请求url也加进去。

其实感觉如何设计测试用例，组织测试数据也是一件很有意思的事情，很多事情都必须先有一个良好的设计思路才会进行的更顺畅。总之勤于思考，多参考他人的思路。不是有句话吗，学而不思则罔，思而不学则殆。

---

2018-04-29 12:53:22