阅读 38 SEO

【接口测试】数据分离-使用xlsx模块读取Excel文件

【接口测试】数据分离-使用xlsx模块读取Excel文件

数据直接写在代码文件里,不利于修改和构造数据。可使用Excel保存测试数据,实现代码和数据的分离。

新建Excel文件test_user_data.xlsx包含工作簿TestUserReg,并复制到项目根目录下,TestUserReg内容如下:

 

一、Excel读取方法

Python使用第三方库 xlrd 来读取 Excel:

复制代码

import xlrd

wb = xlrd.open_workbook("test_user_data.xlsx")  # 打开excelsh = wb.sheet_by_name("TestUserLogin")  # 按工作簿名定位工作表print(sh.nrows)  # 有效数据行数print(sh.ncols)  # 有效数据列数print(sh.cell(0, 0).value)  # 输出第一行第一列的值`case_name`print(sh.row_values(0))  # 输出第1行的所有值(列表格式)# 将数据和标题组装成字典,使数据更清晰# zip()用于将可迭代的对象作为参数,将对象中对应的元素打包成一个个元组,然后返回由这些元组组成的列表。print(dict(zip(sh.row_values(0), sh.row_values(1))))# 遍历excel,打印所有的数据for i in range(sh.nrows):    print(sh.row_values(i))

复制代码

 

二、封装读取excel操作

新建read_excel.py,目的是获取某条用例的数据,需要3个参数:excel数据文件名(data_file),工作簿名(sheet),用例名(case_name)。

如果只封装一个函数,每次调用(每条用例)都要打开一次excel并遍历一次,这样效率比较低。可以拆分成两个函数:excel_to_list(data_file, sheet),获取一个工作表的所有数据;get_test_data(data_list, case_name)从所有数据中去查找到该条用例的数据。

复制代码

import xlrddef excel_to_list(data_file, sheet):
    data_list = []  # 新建个空列表,来乘装所有的数据
    wb = xlrd.open_workbook(data_file)  # 打开excel
    sh = wb.sheet_by_name(sheet)  # 获取工作簿
    header = sh.row_values(0)  # 获取标题行数据
    for i in range(1, sh.nrows):  # 跳过标题行,从第二行开始取数据
        d = dict(zip(header, sh.row_values(i)))  # 将标题和每行数据组装成字典        data_list.append(d)    return data_list  # 列表嵌套字典格式,每个元素是一个字典def get_test_data(data_list, case_name):    for case_data in data_list:        if case_name == case_data['case_name']:  # 如果字典数据中case_name与参数一致
            return case_data  # 如果查询不到会返回Noneif __name__ == '__main__':   # 测试一下自己的代码
    data_list = excel_to_list("test_user_data.xlsx", "TestUserLogin")  # 读取excel,TestUserLogin工作簿的所有数据
    case_data = get_test_data(data_list, 'test_user_login_normal')  # 查找用例'test_user_login_normal'的数据
    print(case_data)

复制代码

 

三、在用例中调用封装好的方法

test_user_reg.py中代码如下:

复制代码

import unittestimport requestsfrom db import *from read_excel import *import json  # 用来转化excel中的json字符串为字典class TestUserReg(unittest.TestCase):
    @classmethod    def setUpClass(cls):  # 整个测试类只执行一次
        cls.data_list = excel_to_list("test_user_data.xlsx", "TestUserReg")  # cls.data_list 同 self.data_list 都是该类的公共属性

    def test_user_reg_normal(self):
        case_data = get_test_data(self.data_list, 'test_user_reg_normal')        if not case_data:  # 有可能为NONE
            print("用例数据不存在")
        url = case_data.get('url')
        data = json.loads(case_data.get('data'))  # 转为字典,需要取里面的name进行数据库检查
        expect_res = json.loads(case_data.get('expect_res'))  # 转为字典,断言时直接断言两个字典是否相等
        name = data.get("name")  # 范冰冰

        # 环境检查
        if check_user(name):
            del_user(name)        # 发送请求
        res = requests.post(url=url, json=data)  # 用 data=data 时可以直接传字符串
        # 响应断言(整体断言)        self.assertDictEqual(res.json(), expect_res)        # 数据库断言        self.assertTrue(check_user(name))        # 环境清理(由于注册接口向数据库写入了用户信息)        del_user(name)if __name__ == '__main__':    # 非必要,用于测试我们的代码
    unittest.main(verbosity=2)

复制代码

 

四、问题总结

1、使用 xlsx.open_workbook 读取excel数据失败,报错:xlrd.biffh.XLRDError: Excel xlsx file; not supported。

原因:xlrd 为2.0.1版本,更新版本后,xlrd 不支持 xlsx 格式数据的读取了。

解决方法:减低 xlsx 版本:先卸载2.0.1版本,再指定安装1.2.0版本。

 

2、excel 数据中是纯数字的,读取到的数据是浮点数。

解决方法:在excel文档中,设置单元格格式为文本格式。

 

3、使用 json.loads 将字符串转化为字典失败,报错:json.decoder.JSONDecodeError: Expecting property name enclosed in double quotes,即期望属性名要用双引号括起来。

原因:json 字符串中的key使用的是单引号,但json.loads()接受的参数并不是直接的一个字典,而需要固定格式的。引号必须为双引号。单引号不行。

解决方法:将单引号修改为双引号。(注意不要是中文的双引号,不然仍会报错)

 

4、扩展知识:json 模块常用的四个函数

import json
json.load() # 将一个存储在文件中的json对象(str)转化为相对应的python对象json.loads() # 将一个json对象(str)转化为相对应的python对象json.dump() # 将python的对象转化为对应的json对象(str),并存放在文件中json.dumps() # 将python的对象转化为对应的json对象(str)

 

 

原文地址:https://www.cnblogs.com/superhin/p/10339002.html


文章分类
后端
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 gxwowoo@163.com 举报,一经查实,本站将立刻删除。
相关推荐