123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141 |
- import sys
- import warnings
- from contextlib import contextmanager
- from typing import Generator
- from typing import Optional
- from typing import TYPE_CHECKING
- import pytest
- from _pytest.config import apply_warning_filters
- from _pytest.config import Config
- from _pytest.config import parse_warning_filter
- from _pytest.main import Session
- from _pytest.nodes import Item
- from _pytest.terminal import TerminalReporter
- if TYPE_CHECKING:
- from typing_extensions import Literal
- def pytest_configure(config: Config) -> None:
- config.addinivalue_line(
- "markers",
- "filterwarnings(warning): add a warning filter to the given test. "
- "see https://docs.pytest.org/en/stable/how-to/capture-warnings.html#pytest-mark-filterwarnings ",
- )
- @contextmanager
- def catch_warnings_for_item(
- config: Config,
- ihook,
- when: "Literal['config', 'collect', 'runtest']",
- item: Optional[Item],
- ) -> Generator[None, None, None]:
- """Context manager that catches warnings generated in the contained execution block.
- ``item`` can be None if we are not in the context of an item execution.
- Each warning captured triggers the ``pytest_warning_recorded`` hook.
- """
- config_filters = config.getini("filterwarnings")
- cmdline_filters = config.known_args_namespace.pythonwarnings or []
- with warnings.catch_warnings(record=True) as log:
- # mypy can't infer that record=True means log is not None; help it.
- assert log is not None
- if not sys.warnoptions:
- # If user is not explicitly configuring warning filters, show deprecation warnings by default (#2908).
- warnings.filterwarnings("always", category=DeprecationWarning)
- warnings.filterwarnings("always", category=PendingDeprecationWarning)
- warnings.filterwarnings("error", category=pytest.PytestRemovedIn7Warning)
- apply_warning_filters(config_filters, cmdline_filters)
- # apply filters from "filterwarnings" marks
- nodeid = "" if item is None else item.nodeid
- if item is not None:
- for mark in item.iter_markers(name="filterwarnings"):
- for arg in mark.args:
- warnings.filterwarnings(*parse_warning_filter(arg, escape=False))
- yield
- for warning_message in log:
- ihook.pytest_warning_captured.call_historic(
- kwargs=dict(
- warning_message=warning_message,
- when=when,
- item=item,
- location=None,
- )
- )
- ihook.pytest_warning_recorded.call_historic(
- kwargs=dict(
- warning_message=warning_message,
- nodeid=nodeid,
- when=when,
- location=None,
- )
- )
- def warning_record_to_str(warning_message: warnings.WarningMessage) -> str:
- """Convert a warnings.WarningMessage to a string."""
- warn_msg = warning_message.message
- msg = warnings.formatwarning(
- str(warn_msg),
- warning_message.category,
- warning_message.filename,
- warning_message.lineno,
- warning_message.line,
- )
- return msg
- @pytest.hookimpl(hookwrapper=True, tryfirst=True)
- def pytest_runtest_protocol(item: Item) -> Generator[None, None, None]:
- with catch_warnings_for_item(
- config=item.config, ihook=item.ihook, when="runtest", item=item
- ):
- yield
- @pytest.hookimpl(hookwrapper=True, tryfirst=True)
- def pytest_collection(session: Session) -> Generator[None, None, None]:
- config = session.config
- with catch_warnings_for_item(
- config=config, ihook=config.hook, when="collect", item=None
- ):
- yield
- @pytest.hookimpl(hookwrapper=True)
- def pytest_terminal_summary(
- terminalreporter: TerminalReporter,
- ) -> Generator[None, None, None]:
- config = terminalreporter.config
- with catch_warnings_for_item(
- config=config, ihook=config.hook, when="config", item=None
- ):
- yield
- @pytest.hookimpl(hookwrapper=True)
- def pytest_sessionfinish(session: Session) -> Generator[None, None, None]:
- config = session.config
- with catch_warnings_for_item(
- config=config, ihook=config.hook, when="config", item=None
- ):
- yield
- @pytest.hookimpl(hookwrapper=True)
- def pytest_load_initial_conftests(
- early_config: "Config",
- ) -> Generator[None, None, None]:
- with catch_warnings_for_item(
- config=early_config, ihook=early_config.hook, when="config", item=None
- ):
- yield
|