run_api_test.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
# @Time    : 2024/5/31
# @File    : run_api_test.py
# @Software: IntelliJ IDEA

__author__ = 'xiaowu'

from common.dateTimeTool import DateTimeTool
from common.fileTool import FileTool
from common.logging import logger
from common.pytest import deal_pytest_ini_file
from init.api.api_init import clear as api_clear
import argparse
import os
import pytest

if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('-clr', '--clr', help='是否清空已有测试结果,1:是、0:否,默认为0', type=str)
    parser.add_argument('-d', '--dir', help='指定要测试的目录', type=str)
    parser.add_argument('-k', '--keyword', help='只执行匹配关键字的用例，会匹配文件名、类名、方法名', type=str)
    parser.add_argument('-lf', '--lf', help='是否运行上一次失败的用例,1:是、0:否,默认为0', type=str)
    parser.add_argument('-m', '--markexpr', help='只运行符合给定的mark表达式的测试', type=str)
    parser.add_argument('-s', '--capture', help='是否在标准输出流中输出日志,1:是、0:否,默认为0', type=str)
    parser.add_argument('-r', '--reruns', help='失败重跑次数,默认为0', type=str)
    parser.add_argument('-nc', '--need_clear', help='是否清除测试case产生的数据', type=int, default=0)
    parser.add_argument('-environment', '--cmd_environment', help='环境', type=str, default='other')

    args = parser.parse_args()

    # 处理pytest文件
    deal_pytest_ini_file()

    # 执行pytest前的参数准备
    pytest_execute_params = ['-c', 'config/pytest.ini', '-v', '--alluredir',
                             'output/api/report_data/' + args.cmd_environment + '/']
    # 判断目录参数
    dir = 'cases/api/'
    # 判断是否清空已有测试结果
    if args.clr:
        if int(args.clr):
            pytest_execute_params.append('--clean-alluredir')
    if args.dir:
        dir = args.dir
    # 判断关键字参数
    if args.keyword:
        pytest_execute_params.append('-k')
        pytest_execute_params.append(args.keyword)
    # 判断是否只运行上一次失败的用例
    if args.lf:
        if int(args.lf):
            pytest_execute_params.append('--lf')
    # 判断markexpr参数
    if args.markexpr:
        pytest_execute_params.append('-m')
        pytest_execute_params.append(args.markexpr)
    # 判断是否输出日志
    if args.capture:
        if int(args.capture):
            pytest_execute_params.append('-s')
    # 判断是否失败重跑
    if args.reruns:
        if int(args.reruns):
            pytest_execute_params.append('--reruns')
            pytest_execute_params.append(args.reruns)

    # 环境
    pytest_execute_params.append('--cmd_environment')
    pytest_execute_params.append(args.cmd_environment)

    pytest_execute_params.append(dir)

    logger.debug('%s开始测试......' % DateTimeTool.getNowTime())
    tmp_exit_code = pytest.main(pytest_execute_params)
    # 清理数据
    api_clear(args.need_clear, args.cmd_environment)