访问@ pytest.fixture

时间:2016-10-19 19:27:04

标签: python pytest

我正在编写@ pytest.fixture,我需要一种方法来访问使用fixture的测试用例名称的信息。

我刚刚发现了一篇涵盖主题的文章:http://programeveryday.com/post/pytest-creating-and-using-fixtures-for-streamlined-testing/ - 谢谢Dan!

import pytest


@pytest.fixture(scope='session')
def my_fixture(request):
    print request.function.__name__
    # I like the module name, too!
    # request.module.__name__
    yield


def test_name(my_fixture):
    assert False

问题是它不适用于会话范围: E AttributeError: function not available in session-scoped context

1 个答案:

答案 0 :(得分:1)

我认为没有必要设置会话作用域,因为@placebo_session(来自here)每个函数调用都可以工作。所以我建议简单地这样做:

@pytest.fixture(scope='function')
def placebo_session(request):
    session_kwargs = {
        'region_name': os.environ.get('AWS_DEFAULT_REGION', 'us-east-1')
    }
    profile_name = os.environ.get('PLACEBO_PROFILE', None)
    if profile_name:
        session_kwargs['profile_name'] = profile_name

    session = boto3.Session(**session_kwargs)

    prefix = request.function.__name__

    base_dir = os.environ.get(
        "PLACEBO_DIR", os.path.join(os.getcwd(), "placebo"))
    record_dir = os.path.join(base_dir, prefix)

    if not os.path.exists(record_dir):
        os.makedirs(record_dir)

    pill = placebo.attach(session, data_path=record_dir)

    if os.environ.get('PLACEBO_MODE') == 'record':
        pill.record()
    else:
        pill.playback()

    return session

但是如果你想要每个会话和每个测试用例都完成某些事情,你可以分成两个灯具(然后使用func_session fixture)。

@pytest.fixture(scope='session')
def session_fixture():
  # do something one per session
  yield someobj

@pytest.fixture(scope='function')
def func_session(session_fixture, request):
  # do something with object created in session_fixture and
  # request.function
  yield some_val