floraachy
/
pytest2
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/_pytest/config/__init__.py
View File

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