关键字 开发-15 mark标记用例

pytest可以支持对用例自定义标记，可以把用例按自己的需要归类标记，比如按用例优先级，标记一些smoke冒烟测试用例。

1. mark标记基本实现

import pytest

@pytest.mark.smoke
def test_login1():
    print('test1')


@pytest.mark.smoke
def test_login2():
    print('test2')


def test_login3():
    print('test3')

执行：pytest .\test_study_mark.py -m smoke,它会只执行被smoke标记过的用例。

# 对测试类下的每个用例都会生效
@pytest.mark.dack
class TestRun:
    def test_a3(self):
        pass

    def test_a4(self):
        pass

import pytest

# 对整个测试模块下的用例全部标记
pytestmark = pytest.mark.aaa

# 对对测试类下的每个用例都会生效
@pytest.mark.dack
class TestRun:
    def test_a3(self):
        pass

    def test_a4(self):
        pass

@pytest.mark.smoke
def test_login1():
    print('test1')


@pytest.mark.smoke
def test_login2():
    print('test2')


def test_login3():
    print('test3')

执行：pytest .\test_study_mark.py -m aaa,它会对整个测试模块全部生效。

2. 动态添加pytestmark属性

给运行函数添加pytestmark属性，并且通过pytest.Mark动态构造mark。

import pytest


def test_login1():
    print('test1')

def test_login2():
    print('test2')

def test_login3():
    print('test3')

# 给函数动态加pytestmark属性
test_login1.pytestmark = [
    pytest.Mark(name='aaa',
                args=(),
                kwargs={}
                )
]

给test_login1函数动态添加了pytestmark属性，于是执行：pytest .\test_study_mark.py -m aaa。

3. 在yaml中实现mark

在yaml文件中实现mark标记用例，需要在用例函数执行之前就添加pytestmark属性，并且添加到用例函数中。yaml文件如下所示。

# testcase/test_login2.yml

config:
  name: 登录用例
  variables:
    username: "admin"
    password: "eaton@2023"

test_login2:
    name: 登录2
    mark: smoke
    api: api/login.yml
    extract:
      code1: $.code
      code2: body.code
      token: $.data.token
    validate:
      - eq: [$.code, "000000"]
      - eq: [status_code, 200]

test_login3:
    name: 登录3
    api: api/login.yml
    extract:
      code1: $.code
      code2: body.code
      token: $.data.token
    validate:
      - eq: [$.code, "000000"]
      - eq: [status_code, 200]

在收集用例名称和用例内容的后，收集mark关键字内容。

# utils/run.py

            # ------------------ 收集 用例 mark 关键字------------
            case_mark = []
            if 'mark' in case[case_name][0]:
                case_mark = case[case_name][0].get('mark', [])
                if isinstance(case_mark, str):
                    case_mark = [item.strip(' ') for item in case_mark.split(',')]
                elif isinstance(case_mark, int):
                    case_mark = [str(case_name)]
                log.info(f'mark用例标记： {case_mark}')
            # ------------------ 收集 用例 mark 关键字 end------------

收集mark之后，在向module中加入test函数之前，需先将pytestmark动态添加进用例函数中。于是。

            # 给用例函数f，添加pytestmark属性
            if case_mark:
                f.pytestmark = [
                    pytest.Mark(name=mark_name,
                                args=(),
                                kwargs={}) for mark_name in case_mark
                ]
                # ------为用例添加mark标记示例 end--------
            # 向 module中加入test 函数
            setattr(self.module, case_name, f)

执行：pytest testcase/test_login2.yml -m smoke，如图可知，2个用例，只执行了被mark标记的用例。

3.1 在config中添加mark，全用例生效

config 中 mark 关键字添加给模块属性，对每个用例都生效。

        # -------------mark 标记config 下整个yaml 中全部标记--------------#
        config_mark = self.raw.get('config').get('mark', [])
        if isinstance(config_mark, str):
            config_mark = [item.strip(" ") for item in config_mark.split(',')]
        elif isinstance(config_mark, int):
            config_mark = [str(config_mark)]
        if config_mark:
            pytest_m = [
                    pytest.Mark(name=mark_name,
                                args=(),
                                kwargs={}) for mark_name in config_mark
            ]
            log.info(f'收集到的config mark：{config_mark}')
            setattr(self.module, "pytestmark", pytest_m)
        # ---------mark end---------------#

        case = {}  # 收集用例名称和执行内容

# testcase/test_login2.yml

config:
  name: 登录用例
  mark: aaa
  variables:
    username: "admin"
    password: "eaton@2023"

test_login2:
    name: 登录2
    mark: smoke
    api: api/login.yml
    extract:
      code1: $.code
      code2: body.code
      token: $.data.token
    validate:
      - eq: [$.code, "000000"]
      - eq: [status_code, 200]

test_login3:
    name: 登录3
    api: api/login.yml
    extract:
      code1: $.code
      code2: body.code
      token: $.data.token
    validate:
      - eq: [$.code, "000000"]
      - eq: [status_code, 200]

执行用例：pytest .\testcase\test_login2.yml -m aaa