|
2 | 2 |
|
3 | 3 | from collections.abc import Sequence |
4 | 4 | from importlib.metadata import version |
| 5 | +from typing import Any |
5 | 6 |
|
6 | | -from _pytest.fixtures import FixtureDef, FixtureManager |
| 7 | +from _pytest.fixtures import FixtureDef, FixtureManager, FixtureRequest |
7 | 8 | from _pytest.nodes import Node |
8 | | -from packaging.version import Version |
9 | 9 | from packaging.version import parse as parse_version |
10 | 10 |
|
11 | 11 | pytest_version = parse_version(version("pytest")) |
12 | 12 |
|
| 13 | +__all__ = ["getfixturedefs", "inject_fixture"] |
13 | 14 |
|
14 | | -if pytest_version >= Version("8.1"): |
| 15 | +if pytest_version.release >= (8, 1): |
15 | 16 |
|
16 | 17 | def getfixturedefs(fixturemanager: FixtureManager, fixturename: str, node: Node) -> Sequence[FixtureDef] | None: |
17 | 18 | return fixturemanager.getfixturedefs(fixturename, node) |
18 | 19 |
|
| 20 | + def inject_fixture(request: FixtureRequest, arg: str, value: Any) -> None: |
| 21 | + """Inject fixture into pytest fixture request. |
| 22 | +
|
| 23 | + :param request: pytest fixture request |
| 24 | + :param arg: argument name |
| 25 | + :param value: argument value |
| 26 | + """ |
| 27 | + |
| 28 | + request._fixturemanager._register_fixture( |
| 29 | + name=arg, |
| 30 | + func=lambda: value, |
| 31 | + nodeid=request.node.nodeid, |
| 32 | + ) |
| 33 | + |
19 | 34 | else: |
20 | 35 |
|
21 | 36 | def getfixturedefs(fixturemanager: FixtureManager, fixturename: str, node: Node) -> Sequence[FixtureDef] | None: |
22 | 37 | return fixturemanager.getfixturedefs(fixturename, node.nodeid) |
| 38 | + |
| 39 | + def inject_fixture(request: FixtureRequest, arg: str, value: Any) -> None: |
| 40 | + """Inject fixture into pytest fixture request. |
| 41 | +
|
| 42 | + :param request: pytest fixture request |
| 43 | + :param arg: argument name |
| 44 | + :param value: argument value |
| 45 | + """ |
| 46 | + fd = FixtureDef( |
| 47 | + fixturemanager=request._fixturemanager, |
| 48 | + baseid=None, |
| 49 | + argname=arg, |
| 50 | + func=lambda: value, |
| 51 | + scope="function", |
| 52 | + params=None, |
| 53 | + ) |
| 54 | + fd.cached_result = (value, 0, None) |
| 55 | + |
| 56 | + old_fd = request._fixture_defs.get(arg) |
| 57 | + add_fixturename = arg not in request.fixturenames |
| 58 | + |
| 59 | + def fin() -> None: |
| 60 | + request._fixturemanager._arg2fixturedefs[arg].remove(fd) |
| 61 | + |
| 62 | + if old_fd is not None: |
| 63 | + request._fixture_defs[arg] = old_fd |
| 64 | + |
| 65 | + if add_fixturename: |
| 66 | + request._pyfuncitem._fixtureinfo.names_closure.remove(arg) |
| 67 | + |
| 68 | + request.addfinalizer(fin) |
| 69 | + |
| 70 | + # inject fixture definition |
| 71 | + request._fixturemanager._arg2fixturedefs.setdefault(arg, []).append(fd) |
| 72 | + |
| 73 | + # inject fixture value in request cache |
| 74 | + request._fixture_defs[arg] = fd |
| 75 | + if add_fixturename: |
| 76 | + request._pyfuncitem._fixtureinfo.names_closure.append(arg) |
0 commit comments