使用python反射，实现pytest读取yaml并发送请求

pytest + yaml

yaml

- feature: 用户模块
  story: 登录
  title: 添加用户
  request:
    method: POST
    url: /system/user/list
    headers: null
    params: null
  validate: null

read_yaml_all

def read_yaml_all(path):
    with open(path, 'r', encoding='utf-8') as f:
        value = yaml.safe_load(f)
        return value

dataclass.py

from dataclasses import dataclass


@dataclass
class CaseInfo:
    feature: str
    story: str
    title: str
    request: dict
    validate: dict


def verify_yaml(case_info: dict):
    """
    通过解包的方式，校验yaml格式是否正确
    :param case_info:
    :return:
    """
    try:
        case = CaseInfo(**case_info)
        return case
    except Exception:
        raise Exception("YAML测试用例不符合规范！")

test_yaml_class.py

from pathlib import Path

import pytest

from lib import read_yaml_all, verify_yaml


class TestYamlCases:
    pass


def create_case_by_yaml(yaml_path):
    # 读取yaml
    @pytest.mark.parametrize('case', read_yaml_all(yaml_path))
    def yaml_function(self, session, case):
        """
        :param self: TestYamlCases类对象
        :param session: 夹具
        :param case: 参数化
        :return:
        """
        # 校验yaml格式是否与数据类字段一致
        case_info = verify_yaml(case)
        # 发送请求
        session.request(**case_info.request)
    # return方法与def平级，切记
    return yaml_function


test_case_yaml_paths = Path(__file__).parent
case_yaml_list = test_case_yaml_paths.glob("**/*.yaml")

for yaml_file in case_yaml_list:
    print(yaml_file)
    print(yaml_file.stem)
    # python 反射，通过反射的方式将pytest的测试用例传入TestYamlCases中
    setattr(TestYamlCases, "test_" + yaml_file.stem, create_case_by_yaml(yaml_file))

成功