pytest的一个demo

上周看了元类的实现方式后,对「代码块」的理解好像有了不一样的认知。

  • 函数是一段代码块
  • 类也是一段代码块

根据现在的工作中使用的pytest写一个demo

demo:

import pytest
import requests

@pytest.fixture(scope='module')
def get_request():
    request_list = []

    def _get_request(data, flag=True):
        if flag:
            x = requests.post('http://httpbin.org/post', data=data)
            request_list.append(x.json().get('data'))
        return request_list

    yield _get_request

@pytest.mark.parametrize("i", [1, 2, 3])
def test_customer_records(i, get_request):
    print('\n')
    print(f'第{i}次测试')
    customer = get_request(f'{{"a{i}", "b{i}"}}')
    print(customer)

结果:


第1次测试
['{"a1", "b1"}']
.

第2次测试
['{"a1", "b1"}', '{"a2", "b2"}']
.

第3次测试
['{"a1", "b1"}', '{"a2", "b2"}', '{"a3", "b3"}']

解释:

  1. 使用module级别的fixture可以定义一个存放多次数据的空的列表request_list

  2. 在fixture中写一个函数,并用yield将它返回以供调用

  3. 测试的时候使用参数化构建case会执行三次

  4. 从测试结果可以看出,列表内容持续增多

写的时候感觉好像很神奇,后来想想其实和以下方式实现一样:

request_list = []
def _get_request2(request_list, data, flag=True):
    if flag:
        x = requests.post('http://httpbin.org/post', data=data)
        request_list.append(x.json().get('data'))
    return request_list

@pytest.fixture(scope='module')
def get_request2():
    yield _get_request2

@pytest.mark.parametrize("i", [1, 2, 3])
def test_customer_records(i, get_request2):
    print('\n')
    print(f'第{i}次测试')
    customer = get_request2(request_list, f'{{"a{i}", "b{i}"}}')
    print(customer)

也不就是:

request_list = []
def _get_request2(request_list, data, flag=True):
    if flag:
        x = requests.post('http://httpbin.org/post', data=data)
        request_list.append(x.json().get('data'))
    return request_list

@pytest.mark.parametrize("i", [1, 2, 3])
def test_customer_records(i):
    print('\n')
    print(f'第{i}次测试')
    customer = _get_request2(request_list, f'{{"a{i}", "b{i}"}}')
    print(customer)

额,好Low

不过其实也不是毫无用处,当scope等级改变之后

  • session:项目级传递数据相当于global
  • class:class级别传递数据,并可以使用pytest的turndown机制
  • function:函数级别,没啥用

你可能感兴趣的:(pytest的一个demo)