在上一节课程中，学习了根据数据驱动测试，当时测试接口需要的数据是存储在excel文档中的，读取比较麻烦。下面是读取excel文档并获取到接口测试必须数据和验证结果数据的代码。首先我们需要创建一个空的list来存放读取到的数据，然后打开excel文档，从第二行开始读取用例内容（第一行是标题），获取到每行内容后，只保留接口测试需要的数据（原来的excel文档中第四列到第七列分别存放了url，请求方式，请求参数，预期结果），然后将这部分数据存入空的list中。这样cases就是一个二维数据，这个list中包含了excel中每一行的用例数据，每个用例数据里面包含了四个元素。类似于[[url1,method1,req_data1,check1],[url2,method2,req_data2,check2]]。代码中的apt_log是实例化的日志记录类。但是读取后，在使用这些元素进行接口测试时，还需要再次转换，因为request模块接口测试中的data是字典格式的。此处是list格式，届时还需要转换。
1 def get_case(self, file_path): 2 cases = [] # 存放所有的测试用例 3 if file_path.endswith('.xls') or file_path.endswith('.xlsx'): # 判断路径是否存在 4 try: 5 book = xlrd.open_workbook(file_path) 6 sheet = book.sheet_by_index(0) # 打开用例文件 7 for i in range(1, sheet.nrows): # 从第二行开始遍历excel文件内容（第一行是标题） 8 row_data = sheet.row_values(i) # 获取每行内容 9 cases.append(row_data[4:8]) # 在cases这个list中存每个用例的url,method,req_data,check 10 apt_log.info('共读取%s条用例' % (len(cases))) # 检查cases中的元素个数，日志记录读取了几条用例 11 self.file_path = file_path # 既然读取成功说明这个文件路径是正确的，那么这边定义下面写exccel就可以直接使用了 12 except Exception as e: 13 apt_log.error('【%s】用例打开失败，错误信息：%s' % (file_path, e)) 14 else: 15 apt_log.error('用例文件不合法，%s' % file_path) 16 return cases
可以看到读取excel步骤比较繁琐，需要各种格式转化。但是yaml格式的文件简化了读取用例的步骤，读取这个格式的文件会自动设置为字典模式。而且只要在yaml格式的文件中录入数据时遵循了文件固定的格式，也可以自动读取出双层字典，有多个用例时可以产生一个list包裹所有的字典格式的用例数据。如果不添加符号"-"，读取时文件中的所有数据都会转为字典格式，而添加后，文件数据读取为list格式，但符号内换行锁紧记录的数据是字典格式。但是check数据中，直接在符号"-"后添加数据，这部分数据会读取为list格式。
1 - # 添加减号可以把用例转为list，每一部分是一个字典 2 url: /api/user/login 3 detail: 正常登陆 4 method: post 5 data: # 存放多层嵌套字典的方式 6 username: ********* 7 passwd: ******* 8 # headers: 9 # User-agent: Firefox 10 # cookies: 11 # sign: adfdsfnsdfnsd 12 check: # 这样定义check可以用in来判断测试是否通过 13 - sign 14 - userId 15 16 - 17 url: /api/user/login 18 detail: 密码错误 19 method: post 20 data: 21 username: ********* 22 passwd: ********* 23 check: 24 - 密码错误 25 26 - 27 url: /api/user/login 28 detail: 必填参数未填，不填密码 29 method: post 30 data: 31 username: ********* 32 check: 33 - 必填参数未填
然后读取yaml格式的文件，就像一般文件读取一样操作就可以了。
1 import yaml 2 3 4 f = open('login.yaml',encoding='utf-8') 5 res = yaml.load(f) 6 print(res)
读取到的数据格式如下，可以看到一个list中包含了三个元素，每个元素都是字典格式的，这样读取数据就很方便了。
1 [{'url': '/api/user/login', 'check': ['sign', 'userId'], 'method': 'post', 'data': {'username': '*********', 'passwd': '*********'}, 'detail': '正常登陆'}, 2 {'url': '/api/user/login', 'check': ['密码错误'], 'method': 'post', 'data': {'username': '*********', 'passwd': '*********'}, 'detail': '密码错误'},
3 {'url': '/api/user/login', 'check': ['必填参数未填'], 'method': 'post', 'data': {'username': '*********'}, 'detail': '必填参数未填，不填密码'}]