From 4d2e1fd4b3620ab7e95b744e87f059cde07d0cdc Mon Sep 17 00:00:00 2001 From: floraachy <1622042529@qq.com> Date: Mon, 22 May 2023 09:37:06 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E6=A0=B9=E6=8D=AE=E9=85=8D?= =?UTF-8?q?=E7=BD=AE=E8=AF=BB=E5=8F=96=E7=94=A8=E4=BE=8B=E6=95=B0=E6=8D=AE?= =?UTF-8?q?=EF=BC=8C=E8=87=AA=E5=8A=A8=E7=94=9F=E6=88=90=E6=B5=8B=E8=AF=95?= =?UTF-8?q?=E7=94=A8=E4=BE=8B=E6=96=87=E4=BB=B6=E7=9A=84=E6=96=B9=E6=B3=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- case_utils/case_handle.py | 124 +++++++++++++++++++++++++------------- 1 file changed, 82 insertions(+), 42 deletions(-) diff --git a/case_utils/case_handle.py b/case_utils/case_handle.py index 2ba7cc7..3a1a311 100644 --- a/case_utils/case_handle.py +++ b/case_utils/case_handle.py @@ -7,9 +7,9 @@ # @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.project_path import CASE_TEMPLATE_DIR, DATA_DIR, AUTO_CASE_DIR +from common_utils.excel_handle import ExcelHandle +from common_utils.yaml_handle import YamlHandle from config.global_vars import CaseFileType from config.settings import CASE_FILE_TYPE from string import Template @@ -17,6 +17,26 @@ from loguru import logger from common_utils.files_handle import get_files +def get_yaml_data(file_path): + """ + 从yaml/yml文件中获取用例数据 + :param file_path: yaml/yml文件绝对路径 + """ + if os.path.isfile(file_path): + # 读取yaml/yml文件中的用例数据，存储到data中 + return YamlHandle(file_path).read_yaml + + +def get_excel_data(file_path): + """ + 从xlsx/xls文件中获取用例数据 + :param file_path: xlsx/xls文件绝对路径 + """ + if os.path.isfile(file_path): + # 读取xlsx/xls文件中的用例数据，存储到data中 + return ExcelHandle(file_path).read() + + def get_case_data(): """ 根据配置文件，从指定类型文件中读取用例数据，并调用生成用例文件方法，生成用例文件 @@ -29,19 +49,21 @@ def get_case_data(): 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}") + data = ExcelHandle(file).read() + for _data in data: + # 将excel读取到的用例数据，适配allure格式 + excel_data = { + 'case_common': {'allure_epic': 'GitLink接口', 'allure_feature': _data["sheet_name"], + 'allure_story': _data["sheet_name"]}, + 'case_info': _data["data"] + } + # 调用gen_case方法生成测试用例, 例如：test_demo.py + gen_case_file(case_file_path=file, case_template_path=CASE_TEMPLATE_DIR, case_data=excel_data, + target_case_path=AUTO_CASE_DIR) + # 将获取到的用例数据统一保存到cases中 + cases.extend([excel_data]) + logger.debug(f"从{file}中读取到的用例数据是：{excel_data}") return cases # 从yaml中读取用例数据 elif CASE_FILE_TYPE == CaseFileType.YAML.value: @@ -49,18 +71,14 @@ def get_case_data(): 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}") + # 从yaml/yml中读取用例数据 + yaml_data = get_yaml_data(file) + # 调用gen_case方法生成测试用例, 例如：test_demo.py + gen_case_file(case_file_path=file, case_template_path=CASE_TEMPLATE_DIR, case_data=yaml_data, + target_case_path=AUTO_CASE_DIR) + # 将获取到的用例数据统一保存到cases中 + cases.extend([yaml_data]) + logger.debug(f"从{file}中读取到的用例数据是：{yaml_data}") return cases else: # 在用例数据"DATA_DIR"目录中寻找后缀是xlsx,xls, yaml, yml的文件 @@ -69,25 +87,40 @@ def get_case_data(): 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) + # 读取excel文件中的用例数据，存储到data中 + data = ExcelHandle(file).read() + for _data in data: + # 将excel读取到的用例数据，适配allure格式 + excel_data = { + 'case_common': {'allure_epic': 'GitLink接口', 'allure_feature': _data["sheet_name"], + 'allure_story': _data["sheet_name"]}, + 'case_info': _data["data"] + } + # 调用gen_case方法生成测试用例, 例如：test_demo.py + gen_case_file(case_file_path=file, case_template_path=CASE_TEMPLATE_DIR, case_data=excel_data, + target_case_path=AUTO_CASE_DIR) + # 将获取到的用例数据统一保存到cases中 + cases.extend([excel_data]) + logger.debug(f"从{file}中读取到的用例数据是：{excel_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}") + # 从yaml/yml中读取用例数据 + yaml_data = get_yaml_data(file) + # 调用gen_case方法生成测试用例, 例如：test_demo.py + gen_case_file(case_file_path=file, case_template_path=CASE_TEMPLATE_DIR, case_data=yaml_data, + target_case_path=AUTO_CASE_DIR) + # 将获取到的用例数据统一保存到cases中 + cases.extend([yaml_data]) return cases -def gen_case_file(func_name, case_data, class_name): +def gen_case_file(case_file_path, case_template_path, case_data, target_case_path): """ - 根据定义的模板生成测试用例文件 + 根据测试用例文件(yaml/yml/xlsx/xls)，以及事先定义的测试用例模板，实际用例数据，生成测试用例方法(.py) + :param case_file_path: 测试用例文件(yaml/yml/xlsx/xls)的绝对路径 + :param case_template_path: 测试用例模板的绝对路径 + :param case_data: 实际用例数据 + :param target_case_path: 测试用例方法(.py)的绝对路径 """ # 如果自动生成用例的目录不存在则自动创建一个 if not os.path.exists(AUTO_CASE_DIR): @@ -95,11 +128,18 @@ def gen_case_file(func_name, case_data, class_name): """ string.Template是将一个string设置为模板，通过替换变量的方法，最终得到想要的string。 """ + filename = os.path.basename(case_file_path) + # 将用例数据的名称作为测试用例文件名称 + func_name = os.path.splitext(filename)[0] + # 测试用例test_demo.py的类名是TestDemo + class_name = func_name.split("_")[0].title() + func_name.split("_")[1].title() # 定义生成的测试用例的模板 - with open(file=os.path.join(CONF_DIR, "case_template.txt"), mode="r", encoding="utf-8") as f: + with open(file=case_template_path, 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: + # 将测试用例方法写入py文件中 + with open(os.path.join(target_case_path, func_name + '.py'), "w", encoding="utf-8") as fp: fp.write(my_case)