|
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 as _FixtureDef |
7 |
| -from _pytest.fixtures import FixtureManager |
| 7 | +from _pytest.fixtures import FixtureDef, FixtureManager, FixtureRequest |
8 | 8 | from _pytest.nodes import Node
|
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", "FixtureDef"] |
| 13 | +__all__ = ["getfixturedefs", "inject_fixture"] |
14 | 14 |
|
15 | 15 | if pytest_version.release >= (8, 1):
|
16 | 16 |
|
17 |
| - def getfixturedefs(fixturemanager: FixtureManager, fixturename: str, node: Node) -> Sequence[_FixtureDef] | None: |
| 17 | + def getfixturedefs(fixturemanager: FixtureManager, fixturename: str, node: Node) -> Sequence[FixtureDef] | None: |
18 | 18 | return fixturemanager.getfixturedefs(fixturename, node)
|
19 | 19 |
|
20 |
| - def FixtureDef(fixturemanager, **kwargs): |
21 |
| - kwargs.setdefault("config", fixturemanager.config) |
22 |
| - return _FixtureDef(**kwargs) |
| 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 | + ) |
23 | 33 |
|
24 | 34 | else:
|
25 | 35 |
|
26 | 36 | def getfixturedefs(fixturemanager: FixtureManager, fixturename: str, node: Node) -> Sequence[FixtureDef] | None:
|
27 | 37 | return fixturemanager.getfixturedefs(fixturename, node.nodeid)
|
28 | 38 |
|
29 |
| - FixtureDef = _FixtureDef |
| 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