根据上一部分中,获取到了Excel中的第一行的接口字段列。下面可以遍历测试用例每一行,获取每一条测试用例中接口字段的值。
#!/bin/usr/env python3
# -*- coding: utf-8 -*-
# FileName: get_excelData.py
from ApiTestFrame.util.operate_excel import OperateExcel
from ApiTestFrame.constant import get_constant
from ApiTestFrame.util.operate_json import OperateJson
from ApiTestFrame.util.connect_db import ConnectDB
class GetExcelData:
def __init__(self):
self.operate_excel = OperateExcel()
# 获取excel行数,也就是case的个数
def get_case_lines(self):
return self.operate_excel.get_lines()
# 是否携带header
def is_header(self, row):
col = int(get_constant.get_header())
header = self.operate_excel.get_cell(row, col)
if header == 'yes':
return get_constant.get_header_value()
else:
return None
# 获取是否执行
def get_is_run(self, row):
flag = None
col = int(get_constant.get_run())
run_model = self.operate_excel.get_cell(row, col)
if run_model == 'yes':
flag = True
else:
flag = False
return flag
# 获取请求方式
def get_method(self, row):
col = int(get_constant.get_method())
method = self.operate_excel.get_cell(row, col)
return method
# 获取请求链接
def get_url(self, row):
col = int(get_constant.get_url())
url = self.operate_excel.get_cell(row, col)
return url
# 获取请求数据
def get_data(self, row):
col = int(get_constant.get_requestData())
data = self.operate_excel.get_cell(row, col)
if data == '':
return None
return data
# 通过获取关键字拿到data数据
def get_data_from_json(self, row):
operate_json = OperateJson()
key_data = self.get_data(row)
request_data = operate_json.get_keyData(key_data)
return request_data
# 获取预期结果
def get_except_data(self, row):
col = int(get_constant.get_expectedResult())
except_data = self.operate_excel.get_cell(row, col)
if except_data == '':
return None
return except_data
# 通过SQl获取预期结果
def get_except_data_form_mysql(self, row):
op_mysql = ConnectDB()
sql = self.get_except_data(row)
print(sql)
res = op_mysql.select_data(sql)
return res.decode('unicode-escape')
# 写入实际结果
def write_result(self, row, value):
col = int(get_constant.get_actualResult())
self.operate_excel.write_value(row, col, value)