Added environment variable PYTEST_VERSION which specifies pytest version, this is set when pytest is running and unset just before pytest is stopped

This commit is contained in:
dheerajck 2024-03-23 21:07:03 +05:30
parent 4e3dd21506
commit c3e8ec8989
1 changed files with 4 additions and 0 deletions

View File

@ -50,6 +50,7 @@ from .compat import PathAwareHookProxy
from .exceptions import PrintHelp as PrintHelp
from .exceptions import UsageError as UsageError
from .findpaths import determine_setup
from _pytest import __version__
import _pytest._code
from _pytest._code import ExceptionInfo
from _pytest._code import filter_traceback
@ -152,6 +153,7 @@ def main(
:returns: An exit code.
"""
try:
os.environ["PYTEST_VERSION"] = __version__
try:
config = _prepareconfig(args, plugins)
except ConftestImportFailure as e:
@ -186,6 +188,8 @@ def main(
for msg in e.args:
tw.line(f"ERROR: {msg}\n", red=True)
return ExitCode.USAGE_ERROR
finally:
os.environ.pop("PYTEST_VERSION", None)
def console_main() -> int: