Pytest中实现自动生成测试用例脚本代码

最新推荐文章于 2024-06-19 23:17:04 发布

2401_84141210

最新推荐文章于 2024-06-19 23:17:04 发布

阅读量233

点赞数 3

文章标签：单元测试功能测试学习

本文链接：https://blog.csdn.net/2401_84141210/article/details/138942821

版权

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

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

需要这份系统化的资料的朋友，可以戳这里获取

2024软件测试面试刷题，这个小程序（永久刷题），靠它快速找到工作了！（刷题APP的天花板）_软件测试刷题小程序-CSDN博客文章浏览阅读2.8k次，点赞85次，收藏12次。你知不知道有这么一个软件测试面试的刷题小程序。里面包含了面试常问的软件测试基础题，web自动化测试、app自动化测试、接口测试、性能测试、自动化测试、安全测试及一些常问到的人力资源题目。最主要的是他还收集了像阿里、华为这样的大厂面试真题，还有互动交流板块……_软件测试刷题小程序外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传 https://blog.csdn.net/AI_Green/article/details/134931243?spm=1001.2014.3001.5502在Python的测试框架中，我们通常会针对某个系统进行测试用例的维护，在对庞大系统进行用例维护时，往往会发现很多测试用例是差不多的，甚至大多数代码是一样的。

故为了提高我们测试用例维护的效率，在本文中，我们将探讨如何自己设计脚本，自动生成测试用例脚本。

自动发现测试原理

pytest的一个重要特性就是它可以自动发现并运行测试。所以我们只需要按照一定的命名规则来命名测试文件和测试函数，然后pytest就可以自动找到并运行这些测试。

默认情况下，pytest会运行所有名称以test_开头或者以_test结尾的.py文件中的测试。在这些文件中，pytest会运行所有名称以test_开头的函数或者方法。

例如，以下代码定义了一个测试函数：

def test_addition():
    assert 1 + 1 == 2

你可以把这个函数保存在一个名为test_example.py的文件中，然后在命令行中运行pytest命令，pytest就会自动找到并运行这个测试。

接口测试数据维护

为了达到目的，所以我们约定以特定规则进行接口测试数据的维护，我们选择json类型文件进行维护，文件目录结构如下：

需要注意的是，考虑到维护的便利性，我们推荐根据模块进行分层管理接口数据文件。

读取接口测试文件信息方法

有了接口数据文件，那么我们就可以设计一个方法，根据传入的接口模块、接口名称，进行相应的接口文件地址拼接，从而去读取json文件内容，代码如下：

class GetTestData:
    """
    获取测试数据封装模块
    """

    def __init__(self, logger):
        self.test_data_path = FilePathSetting.test_data_path  # 获取测试数据配置路径
        self.logger = logger

    def get_api_data(self, api_module, api_name):
        """
        根据接口模块和接口名称 获取接口配置信息
        :param api_module:接口模块
        :param api_name:接口名称
        :return:
        """
        # 拼接需要读取接口模板信息的*.json文件路径地址
        api_data_path = os.path.join(os.path.join(self.test_data_path, api_module), api_name)
        self.logger.info("{init_api_data} 接口模板文件地址：%s" % api_data_path)

        # 读取接口模板路径下的*.json信息
        with open(file=api_data_path, mode='r', encoding='utf-8') as f:
            api_data = json.load(f)

        self.logger.info("{init_api_data} 接口模板信息：%s" % json.dumps(api_data, ensure_ascii=False))

        return api_data

    def get_case_data(self, pytest_current_test):
        """
        根据传入的test_*.py路径 获取case配置信息
        :param pytest_current_test:
        :return:
        """
        # self.logger.info('~~~~~~~~~~~~ {init_api_data} 初始化当前接口请求模板数据 ~~~~~~~~~~~~')
        self.logger.info('{init_api_data} 当前test_case位置信息:%s' % pytest_current_test)
        # 根据传入的test_*.py路径 获取接口所属模块文件名称
        file_path = pytest_current_test.split("::Test")[0]
        self.logger.info('{init_api_data} 当前接口test_case文件地址:%s' % file_path)

        # 根据传入的test_*.py路径 获取接口所属模块文件名称
        api_module = os.path.split(os.path.split(file_path)[0])[1]
        self.logger.info("{init_api_data} 接口所属模块：%s" % api_module)
        # 根据传入的test_*.py路径 获取接口模板（*.json）名称
        # # api_name = os.path.split(file_path)[1].replace('.py', '.json').replace('test_', '')
        api_name = os.path.split(file_path)[1][5:].replace('.py', '.json')  # 1、截取第6个字符到结尾；2、将.py后缀替换为.json
        self.logger.info("{init_api_data} 接口模板文件名称：%s" % api_name)

        return self.get_api_data(api_module, api_name)

基础的测试用例模板

因为我们基础的测试用例是通用的，如以下代码实例：

import allure
import pytest

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

api_module = '接口模块***'
api_name = '接口名称***'


@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)

故我们就可以设计一个用例模板的生成器：

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)


![img](https://img-blog.csdnimg.cn/img_convert/28628fb4e9838066d44527db47789d71.png)
![img](https://img-blog.csdnimg.cn/img_convert/6c2069e057ecaaa47663ef68419fc24a.png)
![img](https://img-blog.csdnimg.cn/img_convert/29fe9308e288d3dfca0ea6049bb58734.png)

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

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

**[需要这份系统化的资料的朋友，可以戳这里获取](https://bbs.csdn.net/topics/618631832)**

(img-tlojnnuY-1715821279964)]
[外链图片转存中...(img-moMo5r3e-1715821279964)]
[外链图片转存中...(img-UKo9UAvj-1715821279965)]

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

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

**[需要这份系统化的资料的朋友，可以戳这里获取](https://bbs.csdn.net/topics/618631832)**

2401_84141210

关注

3
点赞
踩
6

收藏

觉得还不错? 一键收藏
0
评论
Pytest中实现自动生成测试用例脚本代码

在Python的测试框架中，我们通常会针对某个系统进行测试用例的维护，在对庞大系统进行用例维护时，往往会发现很多测试用例是差不多的，甚至大多数代码是一样的。故为了提高我们测试用例维护的效率，在本文中，我们将探讨如何自己设计脚本，自动生成测试用例脚本。
复制链接

扫一扫