import json
import pytest
import requests
#加载excel文件
from openpyxl import load_workbook
#读取文件
wb = load_workbook("date/sml.xlsx")
#打印文件中的表(看工作簿中有几个表)
print(wb.worksheets)
#读取文件中的Sheet表
ws = wb['Sheet']
#表sheet中一共有多少行 结果:4行
rows = ws.max_row
#表sheet一共有多少列 结果:6列
cols = ws.max_column
#定义一个空列表 (下面需要装读取的最终内容)
test_data_tt = []
#因为第一行是表头,用不到所以从2行开始循环。
for i in range(2,rows+1):
#定义空列表,循环每一个参数写入其中
testcast_data = []
for y in range(2,cols+1):
testcast_data.append(ws.cell(row = i,column = y).value)
test_data_tt.append(testcast_data)
#使用参数化引用变量
@pytest.mark.parametrize("url,method,topic_data,code,msg",test_data_tt)
#定义函数,以及对应参数
def test_topiv_n(url,method,topic_data,code,msg):
if method == 'post':
#因为在上一步读取过程中将topic_data当成了一个字符串,但是我们在使用时需要的是字典形式,所以使用json.loads将其转化为字典。
res = requests.post(url,data=json.loads(topic_data))
print(res.json())
#eval函数将字符串转化为字典
msg = eval(msg)
assert res.status_code == code
assert res.json() == msg
最终运行结果如下:
需要注意:
以上内容需要在Terminal中执行(否则读取文件位置会报错,
解决办法修改wb = load_workbook("date/sml.xlsx") 为 ("../date/sml.xlsx"))
- 读取文件时,我们用max_row或者max_column是为了知道他有多少行和列,但是实际用的过程中根据自己情况来决定选取行列数。(可以在下面的循环中range修改数据)
- 正常在使用requests库中,将字符串转化为字典使用json.loads就可以了,但是大家会发现,上面的案例运行过程中msg如果也用同样的方法是不行的,所以需要用到eval函数来进行字符串转化为字典。
- 如果msg中有\或者特殊字符,也可以用
y = msg.replace('\\','') 去除\和""