contrib/python/pytest-lazy-fixtures/pytest_lazy_fixtures/loader.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

import pytest

from .lazy_fixture import LazyFixtureWrapper
from .lazy_fixture_callable import LazyFixtureCallableWrapper


def load_lazy_fixtures(value, request: pytest.FixtureRequest):
    if isinstance(value, LazyFixtureCallableWrapper):
        return value.get_func(request)(
            *load_lazy_fixtures(value.args, request),
            **load_lazy_fixtures(value.kwargs, request),
        )
    if isinstance(value, LazyFixtureWrapper):
        return value.load_fixture(request)
    # we need to check exact type
    if type(value) is dict:
        return {load_lazy_fixtures(key, request): load_lazy_fixtures(val, request) for key, val in value.items()}
    # we need to check exact type
    if type(value) in {list, tuple, set}:
        return type(value)(load_lazy_fixtures(val, request) for val in value)
    return value