floraachy
106 lines
5.2 KiB
Python
106 lines
5.2 KiB
Python
# -*- coding: utf-8 -*-
|
|
# @Version: Python 3.9
|
|
# @Time : 2023/1/12 15:22
|
|
# @Author : chenyinhua
|
|
# @File : case_handle.py
|
|
# @Software: PyCharm
|
|
# @Desc: 生成测试用例文件并返回用例数据
|
|
|
|
import os
|
|
from config.project_path import CONF_DIR, DATA_DIR, AUTO_CASE_DIR
|
|
from common_utils.excel_handle import ReadExcel
|
|
from common_utils.yaml_handle import HandleYaml
|
|
from config.global_vars import CaseFileType
|
|
from config.settings import CASE_FILE_TYPE
|
|
from string import Template
|
|
from loguru import logger
|
|
from common_utils.files_handle import get_files
|
|
|
|
|
|
def get_case_data():
|
|
"""
|
|
根据配置文件，从指定类型文件中读取用例数据，并调用生成用例文件方法，生成用例文件
|
|
"""
|
|
cases = []
|
|
# 判断配置文件里面CASE_DATA_TYPE,判断用例数据是从excel还是yaml文件中读取
|
|
# 从excel中读取用例数据
|
|
if CASE_FILE_TYPE == CaseFileType.EXCEL.value:
|
|
# 在用例数据"DATA_DIR"目录中寻找后缀是xlsx, xls的文件
|
|
files = get_files(target=DATA_DIR, start="test_", end=".xlsx") + get_files(target=DATA_DIR, start="test_",
|
|
end=".xls")
|
|
for file in files:
|
|
# 这里file文件绝对路径，filename才是文件名称
|
|
filename = os.path.basename(file)
|
|
# 读取excel文件中的用例数据，存储到data中
|
|
data = ReadExcel(file).read()
|
|
# 将用例数据的名称作为测试用例文件名称
|
|
func_name = os.path.splitext(filename)[0]
|
|
# 测试用例test_demo.py的类名是TestDemo
|
|
class_name = func_name.split("_")[0].title() + func_name.split("_")[1].title()
|
|
# 调用gen_case方法生成测试用例test_demo.py
|
|
gen_case_file(func_name, data, class_name)
|
|
# 将excel中读取的用例数据放到cases列表中
|
|
cases.extend(data)
|
|
logger.debug(f"从excel中读取到的用例数据是：{cases}")
|
|
return cases
|
|
# 从yaml中读取用例数据
|
|
elif CASE_FILE_TYPE == CaseFileType.YAML.value:
|
|
# 在用例数据"DATA_DIR"目录中寻找后缀是yaml, yml的文件
|
|
files = get_files(target=DATA_DIR, start="test_", end=".yaml") + get_files(target=DATA_DIR, start="test_",
|
|
end=".yml")
|
|
for file in files:
|
|
filename = os.path.basename(file)
|
|
# 读取yaml/yml文件中的用例数据，存储到data中
|
|
data = HandleYaml(file).read_yaml
|
|
# 将用例数据的名称作为测试用例文件名称
|
|
func_name = os.path.splitext(filename)[0]
|
|
# 测试用例test_demo.py的类名是TestDemo
|
|
class_name = func_name.split("_")[0].title() + func_name.split("_")[1].title()
|
|
# 调用gen_case方法生成测试用例test_demo.py
|
|
gen_case_file(func_name, data, class_name)
|
|
# 将excel中读取的用例数据放到cases列表中
|
|
cases.extend(data)
|
|
logger.debug(f"从yaml中读取到的用例数据是：{cases}")
|
|
return cases
|
|
else:
|
|
# 在用例数据"DATA_DIR"目录中寻找后缀是xlsx,xls, yaml, yml的文件
|
|
files = get_files(target=DATA_DIR, start="test_", end=".xlsx") + get_files(target=DATA_DIR, start="test_",
|
|
end=".xls") + get_files(
|
|
target=DATA_DIR, start="test_", end=".yaml") + get_files(target=DATA_DIR, start="test_",
|
|
end=".yml")
|
|
for file in files:
|
|
filename = os.path.basename(file)
|
|
if os.path.splitext(file)[1] == ".xlsx" or os.path.splitext(file)[1] == ".xls":
|
|
data = ReadExcel(file).read()
|
|
func_name = os.path.splitext(filename)[0]
|
|
cases.extend(data)
|
|
else:
|
|
data = HandleYaml(file).read_yaml
|
|
func_name = os.path.splitext(filename)[0]
|
|
cases.extend(data)
|
|
|
|
class_name = func_name.split("_")[0].title() + func_name.split("_")[1].title()
|
|
gen_case_file(func_name, data, class_name)
|
|
logger.debug(f"从excel以及yaml中读取到的用例数据是：{cases}")
|
|
return cases
|
|
|
|
|
|
def gen_case_file(func_name, case_data, class_name):
|
|
"""
|
|
根据定义的模板生成测试用例文件
|
|
"""
|
|
# 如果自动生成用例的目录不存在则自动创建一个
|
|
if not os.path.exists(AUTO_CASE_DIR):
|
|
os.makedirs(AUTO_CASE_DIR)
|
|
"""
|
|
string.Template是将一个string设置为模板，通过替换变量的方法，最终得到想要的string。
|
|
"""
|
|
# 定义生成的测试用例的模板
|
|
with open(file=os.path.join(CONF_DIR, "case_template.txt"), mode="r", encoding="utf-8") as f:
|
|
case_template = f.read()
|
|
my_case = Template(case_template).safe_substitute({"case_data": case_data,
|
|
"func_title": func_name,
|
|
"class_title": class_name})
|
|
with open(os.path.join(AUTO_CASE_DIR, func_name + '.py'), "w", encoding="utf-8") as fp:
|
|
fp.write(my_case)
|