直接上干货撸代码，有一些是通用的工具类代码，一次性封装永久使用，期待大家的关注，一起加油！！！
配置文件
根据不同的业务需求进行配置，例如Goods服务、Order服务分开配置，每个服务下配置接口信息，每个接口文件里配置各种case。
/conf/GooodsApi/get_shop_info.yaml
shell
# 公共参数
case_common:
allureEpic: 商品接口
allureFeature: 店铺信息
allureStory: 店铺信息
get_user_info_01:
host: https://goodsapi.e.weibo.com
url: /mp/shop/getShopInfo
method: GET
detail: 获取商家店铺信息
headers:
# Content-Type: multipart/form-data;
# 这里cookie的值，写的是存入缓存的名称
# cookie: $cache{login_cookie}
# 请求的数据，是 params 还是 json、或者file、data
requestType: params
# 是否执行，空或者 true 都会执行
is_run:
data:
{"shop_id": xxxx}
# 是否有依赖业务，为空或者false则表示没有
dependence_case: False
# 依赖的数据
dependence_case_data:
assert:
# 断言接口状态码
errorCode:
jsonpath: $.code
type: ==
value: 0
AssertType:
sql:通用工具类
文件操作
/utils/file_operate.py
python
import os
def get_root_path():
"""获取(项目)根目录"""
return os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
def ensure_path_sep(path):
"""兼容windows和Linux不同环境的操作系统路径"""
if '/' in path:
path = os.sep.join(path.split('/'))
if '\\' in path:
path = os.sep.join(path.split('\\'))
return get_root_path() + path
def get_all_yaml_files(file_path, yaml_data_switch=False):
"""获取配置的yaml文件路径"""
file_name = []
for root, dir, files in os.walk(file_path):
for _file_path in files:
path = os.path.join(root, _file_path)
if yaml_data_switch:
if 'yaml' in path or '.yml' in path:
file_name.append(path)
file_name.append(path)
return file_name操作yaml
/utils/yaml_control.py
python
Class YamlControl:
def __init__(self, file_str):
self.file_str = file_str
def file_is_exists(self, filepath):
if os.path.exists(filepath):
return True
return False
def read_yaml_data(self):
"""获取yaml文件数据"""
if self.file_is_exists(self.file_str):
data = open(self.file_str, "r", encoding="UTF-8")
yaml_data = yaml.load(data, Loader = yaml.FullLoader)
else:
raise FileNotFoundError("文件路径不存在")
return yaml_data
def save_yaml_data(self, file_path, data):
if self.file_is_exists(file_path):
with open(file_path, 'w') as file:
yaml.safe_dump(data, file, default_flow_style=False)
else:
raise FileNotFoundError("文件路径不存在")自动生成用例代码
/utils/case_automatic_genration.py
python
import os
from utils.read_file_tools import get_all_yaml_files, ensure_path_sep
from utils.yaml_control import YamlControl
from utils.test_case import write_testcase_file
class TestCaseAutomaticGeneration:
def __init__(self):
self.yaml_case_data = None
self.file_path = None
@property
def allure_epic(self):
_allure_epic = self.yaml_case_data.get("case_common").get("allureEpic")
assert _allure_epic is not None, (
"用例中 allureEpic 为必填项，请检查用例内容, 用例路径：'%s'" % self.file_path
)
return _allure_epic
@property
def allure_feature(self):
_allure_feature = self.yaml_case_data.get("case_common").get("allureFeature")
assert _allure_feature is not None, (
"用例中 allureFeature 为必填项，请检查用例内容, 用例路径：'%s'" % self.file_path
)
return _allure_feature
@property
def allure_story(self):
_allure_story = self.yaml_case_data.get("case_common").get("allureStory")
assert _allure_story is not None, (
"用例中 allureStory 为必填项，请检查用例内容, 用例路径：'%s'" % self.file_path
)
return _allure_story
@property
def case_data_path(self):
"""返回 yaml 用例文件路径"""
return ensure_path_sep("/conf")
@property
def gen_file_name(self):
"""生成文件名 将.yaml替换成.py"""
# 例 /get_shop_info.yaml------>/get_shop_info.py
# 获取配置文件相对路径
l = len(self.case_data_path)
yaml_path = self.file_path[l:]
file_name = None
if '.yaml' in yaml_path:
file_name = yaml_path.replace('.yaml', '.py')
elif '.yml' in yaml_path:
file_name = yaml_path.replace('.yml', '.py')
return file_name
@property
def get_test_class_title(self):
"""自动生成类名"""
# 例 get_shop_info------>GetShopInfo
_file_name = os.path.split(self.gen_file_name)[1][:-3]
_name = _file_name.split("_")
_name_len = len(_name)
for i in range(_name_len):
_name[i] = _name[i].capitalize()
_class_name = "".join(_name)
return _class_name
@property
def get_func_title(self):
"""自动生成方法名"""
return os.path.split(self.gen_file_name)[1][:-3]
@property
def spilt_path(self):
# 使用"/"分割 获取文件名称
path = self.gen_file_name.split(os.sep)
path[-1] = path[-1].replace(path[-1], "test_" + path[-1])
return path
@property
def get_case_path(self):
"""生成测试用例目录"""
# 相对路径 test_case/x/test_get_shop_info.py
return ensure_path_sep("/test_case" + os.sep.join(self.spilt_path))
@property
def case_ids(self):
return [k for k in self.yaml_case_data.keys() if k != "case_common"]
@property
def get_file_name(self):
# 这里通过"/" 符号进行分割，提取出来文件名称
# 判断生成的 testcase 文件名称，需要以test_ 开头
case_name = self.spilt_path[-1].replace(
self.spilt_path[-1], "test_" + self.spilt_path[-1]
)
return case_name
def mk_dir(self) -> None:
""" 判断生成自动化代码的文件夹路径是否存在，如果不存在，则自动创建 """
# _LibDirPath = os.path.split(self.libPagePath(filePath))[0]
_case_dir_path = os.path.split(self.get_case_path)[0]
if not os.path.exists(_case_dir_path):
os.makedirs(_case_dir_path)
def get_case_automatic(self):
"""自动生成测试用例代码"""
# 获取配置文件下全部yaml文件
file_path = get_all_yaml_files(ensure_path_sep("/conf"), yaml_data_switch=True)
for file in file_path:
# 判断代理拦截的yaml文件，不生成test_case代码
if 'proxy_data.yaml' not in file:
# 获取配置yaml用例数据
self.yaml_case_data = YamlControl(file).read_yaml_data()
# yaml文件相对路径
self.file_path = file
# 判断用例需要用的文件夹路径是否存在，不存在则创建
self.mk_dir()
print(self.gen_file_name, "\n",
self.get_test_class_title,"\n",
self.get_func_title,"\n",
self.get_case_path,"\n",
self.case_ids,"\n",
self.get_file_name)
write_testcase_file(allure_epic = self.allure_epic,
allure_feature=self.allure_feature,
allure_story=self.allure_story,
class_title=self.get_test_class_title,
func_title=self.get_func_title,
case_path=self.get_case_path,
case_ids=self.case_ids,
file_name=self.get_file_name)下一篇介绍用例内容写入/utils/write_testcase_file.py，使用pytest实现自动化