Pytest中实现自动生成测试用例脚本代码！_使用注解生成pytest自动化测试代码

先自我介绍一下，小编浙江大学毕业，去过华为、字节跳动等大厂，目前阿里P7

深知大多数程序员，想要提升技能，往往是自己摸索成长，但自己不成体系的自学效果低效又漫长，而且极易碰到天花板技术停滞不前！

因此收集整理了一份《2024年最新软件测试全套学习资料》，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友。

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，涵盖了95%以上软件测试知识点，真正体系化！

由于文件比较多，这里只是将部分目录截图出来，全套包含大厂面经、学习笔记、源码讲义、实战项目、大纲路线、讲解视频，并且后续会持续更新

如果你需要这些资料，可以添加V获取：vip1024b （备注软件测试）

正文

import datetime
from typing import Text

def write_case(file_path: Text, page: Text) -> None:
“”" 写入用例数据 “”"
with open(file_path, ‘w’, encoding=“utf-8”) as file:
file.write(page)

def write_init_file(*, file_path: Text) -> None:
“”"
初始化文件写入
:param file_path: __init__文件地址
:return:
“”"
now = datetime.datetime.now().strftime(‘%Y/%m/%d %H:%M’)

page = f’‘’#!/usr/bin/env python

-- coding: utf-8 --

@Author : gmluo1988

@Software : PyCharm

@File : init.py.py

@Time : {now}

‘’’
write_case(file_path=file_path, page=page)

def write_test_case_file(*,
api_module: Text,
api_name: Text,
filename: Text,
case_path: Text) -> None:
“”"
标准测试用例文件写入
:param api_module: 接口所属模块
:param api_name: 接口名称名称
:param filename: 用例文件名称
:param case_path: 用例文件地址
“”"
now = datetime.datetime.now().strftime(‘%Y/%m/%d %H:%M’)

page = f’‘’#!/usr/bin/env python

-- coding: utf-8 --

@Author : gmluo1988

@Software : PyCharm

@File : {filename}

@Time : {now}

import allure
import pytest

from utils.assertion.assert_controller import AssertController
from utils.requests_tools.requests_client import RequestClient

@allure.epic(“{api_module}”)
@allure.feature(“{api_module} {api_name}”)
class TestCaseGroup:

@allure.title(“接口健康检查”)
@allure.description(“接口返回信息基础验证”)
@pytest.mark.smoke
def test_smoke(self, init_api_data):

获取环境配置信息&接口测试数据

env_data, case_data = init_api_data

获取接口返回信息

res = RequestClient(case_data).get_api_response()

进行断言验证

AssertController(case_data.assert_data).assert_equality(res.response_data)
‘’’
write_case(file_path=case_path, page=page)

有了测试用例生成器，那么我们就可以根据接口模块，接口名称，以及接口相关信息生成通用的接口用例文件了。

自动生成测试用例

接下来要做事情就是读取test_data目录下所有模块下的接口文件，遍历文件列表进行测试用例的字段生成。

要注意的是：

1、该封装方法可以根据需求设置需要排除的文件列表

2、test_case目录下已经存在的用例会跳过生成

3、该封装方法生成的是一个基本通用的用例，可在生成的用例基础上进行改造

import os

from common.file_path_setting import FilePathSetting
from utils.case_automatic.test_case_template import write_test_case_file, write_init_file
from utils.get_test_data_tools.get_case_data import GetCaseData

class CaseAutoGenerate:
“”"
测试用例自动生成工具
“”"

def init(self):
“”“设置需要排除的文件列表，一般是Cloud登录以及子系统跳转登录等前置条件接口文档”“”
self.excluded_file_list = [
(“users”, “login.json”),
(“users”, “change_password.json”)
]

def get_test_data_files(self) -> list:
“”"
获取所有测试数据文件路径信息列表
每条测试数据由一个tuple组成，包含3个值（filename：文件名；parent_dir：父级目录名称；file_path：文件地址）

需要排除不需要自动生成测试用例的文件
:return:测试数据文件路径信息列表
“”"
file_info_list = [] # 所有测试数据文件路径信息列表

获取测试数据目录所有文件下的子文件名称

for root, dirs, filenames in os.walk(FilePathSetting.test_data_path):
for filename in filenames:
parent_dir = os.path.basename(root) # 测试数据文件父级目录名称
file_path = os.path.join(root, filename) # 测试数据文件地址

需要排除不需要自动生成测试用例的文件

if (parent_dir, filename) not in self.excluded_file_list:

每条测试数据由一个tuple组成，包含3个值（filename：文件名；parent_dir：父级目录名称；file_path：文件地址）

file_info_list.append((filename, parent_dir, file_path))

返回所有测试数据文件路径信息列表

return file_info_list

@staticmethod
def get_test_case_files() -> tuple:
“”"
获取所有测试用例父级目录列表，测试用例文件地址列表

需要将’init.py’, 'conftest.py’文件进行过滤
:return:
“”"

case_dir_list = [] # 用例父级目录列表
case_path_list = [] # 测试用例文件地址列表

获取测试用例目录下所有文件下的子文件名称

for root, dirs, filenames in os.walk(FilePathSetting.test_case_path):
for filename in filenames:
parent_dir = os.path.basename(root)
file_path = os.path.join(root, filename)

需要将’init.py’, 'conftest.py’文件进行过滤

if filename not in [‘init.py’, ‘conftest.py’]:
case_path_list.append(file_path)

如果父级目录还未收录，则进行父级目录列表更新，并将’pycache’排除

if parent_dir not in case_dir_list and parent_dir not in [‘pycache’]:
case_dir_list.append(parent_dir)

分别返回测试用例父级目录列表，测试用例文件地址列表

return case_dir_list, case_path_list

@staticmethod
def mk_dir(test_data_file_info: tuple, existing_case_dir_list: list) -> None:
“”"
创建测试用例父级目录

如果测试数据文件的父级目录在测试用例目录中是不存在，
则在test_case目录下创建相同名称的package文件以及’init.py’文件
:param test_data_file_info: 测试数据文件信息，包含3个值（filename：文件名；parent_dir：父级目录名称；file_path：文件地址）
:param existing_case_dir_list: 已有的所有测试用例父级目录列表
:return:
“”"

test_data_dir = test_data_file_info[1] # 测试数据文件，父级目录名称

如果测试数据文件的父级目录在测试用例目录中是不存在，则在test_case目录下创建相同名称的package文件以及’init.py’文件

if test_data_dir not in existing_case_dir_list:
print(“test_case下缺少:{} 文件目录,进行文件新增操作…”.format(test_data_dir))
“”“创建相同名称的package文件”“”
dir_path = os.path.join(FilePathSetting.test_case_path, test_data_dir)
os.makedirs(dir_path, exist_ok=True)

“”“新增’init.py’文件”“”
init_file_path = os.path.join(dir_path, ‘init.py’)

open(init_file_path, ‘w’).close()

write_init_file(file_path=init_file_path)

@staticmethod
def is_case_exist(test_data_file_info: tuple, existing_case_path_list: list) -> tuple:
“”"
判断测试用例是否存在
如果不存在，则返回需要生成测试用例的文件名和文件地址信息
如果已经存在对应的测试用例，则返回None

:param test_data_file_info:
:param existing_case_path_list:
:return:target_case_filename, target_case_file_path
“”"
test_data_filename = test_data_file_info[0] # 测试数据文件名称
test_data_dir_name = test_data_file_info[1] # 测试数据文件所属父级目录名称

拼接目标测试用例所属父级目录地址

target_case_dir_path = os.path.join(FilePathSetting.test_case_path, test_data_dir_name)

组装目标测试用例文件名称：测试用例以“test_”开头，将文件格式“.json”替换成“.py”

target_case_filename = ‘test_’ + test_data_filename.replace(‘.json’, ‘.py’)

拼接目标测试用例文件地址

target_case_file_path = os.path.join(target_case_dir_path, target_case_filename)

if target_case_file_path in existing_case_path_list:

print(target_case_file_path)

if target_case_file_path not in existing_case_path_list:

返回需要生成测试用例的文件名和文件地址信息

return target_case_filename, target_case_file_path
else:
return None, None

def case_automatic(self) -> None:
“”"
测试用例自动生成入口
:return:
“”"

“”“获取所有测试数据文件路径信息列表”“”
test_data_file_info_list = self.get_test_data_files()
“”“获取所有测试用例父级目录列表，测试用例文件地址列表”“”
existing_case_dir_list, existing_case_path_list = self.get_test_case_files()

“”“遍历每个测试数据文件,如果对应的测试用例不存在，则进行测试用例自动生成操作”“”
for test_data_file_info in test_data_file_info_list:
“”“创建测试用例父级目录”“”
self.mk_dir(test_data_file_info, existing_case_dir_list)
“”“返回需要生成测试用例的文件名和文件地址信息”“”
target_case_filename, target_case_file_path = self.is_case_exist(test_data_file_info,
existing_case_path_list)
“”“创建对应的测试用例文件”“”
if target_case_filename and target_case_file_path:
print(“测试用例文件：{} 不存在,进行测试文件新增…”.format(target_case_file_path))
test_data_file_path = test_data_file_info[2]
case_data = GetCaseData().get_api_data_by_path(test_data_file_path)
api_module = case_data.api_module
api_name = case_data.api_name
write_test_case_file(api_module=api_module,
api_name=api_name,
filename=target_case_filename,
case_path=target_case_file_path)

if name == ‘main’:
print(“Generating test case…”)
case_auto_generate = CaseAutoGenerate()
case_auto_generate.case_automatic()

总结

Python中通过自己设计脚本，从而能够自动生成测试用例脚本，提高我们测试用例维护的效率，当然不同的系统可能有自己特别的方式，在此是为大家作个参考。

网上学习资料一大堆，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。

需要这份系统化的资料的朋友，可以添加V获取：vip1024b （备注软件测试）

一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！
维护的效率，当然不同的系统可能有自己特别的方式，在此是为大家作个参考。

网上学习资料一大堆，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。

需要这份系统化的资料的朋友，可以添加V获取：vip1024b （备注软件测试）
[外链图片转存中…(img-Ms0Ovo9r-1713345717509)]

一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！