Как я могу найти, какие приборы используются в тесте?

Question

Я хочу автоматически отмечать тесты в зависимости от того, какие приборы они используют. Например, если в тесте используется прибор с именем spark, я бы хотел добавить маркер с именем uses_spark, чтобы я мог автоматически их игнорировать.

Я знаю, что могу использовать pytest_collection_modifyitems в conftest.py для добавления маркеров.

def pytest_collection_modifyitems(items):
  for item in items:
    if uses_spark_fixture(item):
      item.add_marker(pytest.mark.spark)

def uses_spark_fixture(item):
  ???

Как мне реализовать uses_spark_fixture?

Answer 1

Каждый элемент хранит список используемых приборов в атрибуте fixturenames. таким образом, проверка довольно проста:

def pytest_collection_modifyitems(items):
    for item in items:
        if 'spark_fixture' in item.fixturenames:
            item.add_marker(pytest.mark.spark)