Disable assertion rewriting external modules

changelog/13403.bugfix.rst

@@ -0,0 +1 @@
+Disable assertion for modules outside current working dir(cwd)

src/_pytest/assertion/__init__.py

@@ -110,8 +110,14 @@ class AssertionState:
     def __init__(self, config: Config, mode) -> None:
         self.mode = mode
         self.trace = config.trace.root.get("assertion")
+        self.config = config
         self.hook: rewrite.AssertionRewritingHook | None = None
 
+    @property
+    def invocation_path(self):
+        """Get current root path (current working dir)"""
+        return str(self.config.invocation_params.dir)
+
 
 def install_importhook(config: Config) -> rewrite.AssertionRewritingHook:
     """Try to install the rewrite hook, raise SystemError if it fails."""

src/_pytest/assertion/rewrite.py

@@ -239,7 +239,9 @@ def _should_rewrite(self, name: str, fn: str, state: AssertionState) -> bool:
         # rewritten if they match the naming convention for test files
         fn_path = PurePath(fn)
         for pat in self.fnpats:
-            if fnmatch_ex(pat, fn_path):
+            if fnmatch_ex(pat, fn_path) and fn_path.is_relative_to(
+                state.invocation_path
+            ):
                 state.trace(f"matched test file {fn!r}")
                 return True
 

src/_pytest/pytester.py

@@ -12,6 +12,7 @@
 from collections.abc import Iterable
 from collections.abc import Sequence
 import contextlib
+import dataclasses
 from fnmatch import fnmatch
 import gc
 import importlib
@@ -749,6 +750,13 @@ def chdir(self) -> None:
         This is done automatically upon instantiation.
         """
         self._monkeypatch.chdir(self.path)
+        self._monkeypatch.setattr(
+            self._request.config,
+            "invocation_params",
+            dataclasses.replace(
+                self._request.config.invocation_params, dir=Path(self._path)
+            ),
+        )
 
     def _makefile(
         self,

testing/example_scripts/rewrite/pytest.ini

@@ -0,0 +1,2 @@
+[pytest]
+python_files = *.py

testing/example_scripts/rewrite/some_plugin/__init__.py

@@ -0,0 +1,11 @@
+from __future__ import annotations
+
+import pytest
+
+
+@pytest.fixture
+def special_asserter():
+    def special_assert(x, y):
+        assert {"x": x} == {"x": y}
+
+    return special_assert

testing/example_scripts/rewrite/src/main.py

@@ -0,0 +1,9 @@
+"""Stub file for testing"""
+
+from __future__ import annotations
+
+
+def func(x: int, y: int) -> int:
+    """Stub function"""
+    assert (x) > 0
+    return 0 if x == y else 1 if x > y else -1

testing/example_scripts/rewrite/tests/conftest.py

@@ -0,0 +1,16 @@
+from __future__ import annotations
+
+from _pytest.fixtures import fixture
+
+
+pytest_plugins = ["pytester", "some_plugin"]
+
+
+@fixture
+def b():
+    return 1
+
+
+@fixture
+def a():
+    return 2

testing/example_scripts/rewrite/tests/test_main.py

@@ -0,0 +1,13 @@
+from __future__ import annotations
+
+from typing import Callable
+
+from testing.example_scripts.rewrite.src.main import func
+
+
+def test_plugin(a: int, b: int, special_asserter: Callable[[int, int], bool]):
+    special_asserter(a, b)
+
+
+def test_func(a: int, b: int, special_asserter: Callable[[int, int], bool]):
+    assert {"res": func(a, b)} == {"res": 0}

testing/example_scripts/rewrite/venv/lib64/python3.11/site-packages/external_lib/external_lib.py

@@ -0,0 +1,3 @@
+def some_check(a: int):
+    assert abs(a)<2
+    return a in set(0, 1, -1)

testing/test_assertrewrite.py

@@ -12,6 +12,7 @@
 import inspect
 import marshal
 import os
+from os import mkdir
 from pathlib import Path
 import py_compile
 import re
@@ -35,6 +36,7 @@
 from _pytest.assertion.rewrite import rewrite_asserts
 from _pytest.config import Config
 from _pytest.config import ExitCode
+from _pytest.monkeypatch import MonkeyPatch
 from _pytest.pathlib import make_numbered_dir
 from _pytest.pytester import Pytester
 import pytest
@@ -370,6 +372,7 @@ def test_rewrites_plugin_as_a_package(self, pytester: Pytester) -> None:
         pytester.makeconftest('pytest_plugins = ["plugin"]')
         pytester.makepyfile("def test(special_asserter): special_asserter(1, 2)\n")
         result = pytester.runpytest()
+
         result.stdout.fnmatch_lines(["*assert 1 == 2*"])
 
     def test_honors_pep_235(self, pytester: Pytester, monkeypatch) -> None:
@@ -1294,6 +1297,34 @@ def test_meta_path():
         )
         assert pytester.runpytest().ret == 0
 
+    def test_invocation_dir(self, pytester: Pytester, monkeypatch: MonkeyPatch) -> None:
+        """Test get invocation param afrom AssertionState"""
+        from _pytest.assertion import AssertionState
+
+        config = pytester.parseconfig()
+        state = AssertionState(config, "rewrite")
+        assert state.invocation_path == str(config.invocation_params.dir)
+        new_rootpath = str(pytester.path / "test")
+        if not os.path.exists(new_rootpath):
+            os.mkdir(new_rootpath)
+        monkeypatch.setattr(
+            config,
+            "invocation_params",
+            Config.InvocationParams(
+                args=(),
+                plugins=(),
+                dir=Path(new_rootpath),
+            ),
+        )
+        state = AssertionState(config, "rewrite")
+        assert state.invocation_path == new_rootpath
+
+    @pytest.mark.skipif(
+        sys.platform.startswith("win32"), reason="cannot remove cwd on Windows"
+    )
+    @pytest.mark.skipif(
+        sys.platform.startswith("sunos5"), reason="cannot remove cwd on Solaris"
+    )
     def test_write_pyc(self, pytester: Pytester, tmp_path) -> None:
         from _pytest.assertion import AssertionState
         from _pytest.assertion.rewrite import _write_pyc
@@ -1971,6 +2002,36 @@ def test_simple_failure():
             assert hook.find_spec("file") is not None
             assert self.find_spec_calls == ["file"]
 
+    def test_assert_rewrites_only_rootpath(
+        self, pytester: Pytester, hook: AssertionRewritingHook, monkeypatch
+    ) -> None:
+        """Do not rewrite assertions in tests outside `AssertState.rootpath` (#13403)."""
+        pytester.makepyfile(
+            **{
+                "file.py": """\
+                    def test_simple_failure():
+                        assert 1 + 1 == 3
+                """
+            }
+        )
+        with mock.patch.object(hook, "fnpats", ["*.py"]):
+            assert hook.find_spec("file") is not None
+
+        rootpath = f"{os.getcwd()}/tests"
+        if not os.path.exists(rootpath):
+            mkdir(rootpath)
+        monkeypatch.setattr(
+            pytester._request.config,
+            "invocation_params",
+            Config.InvocationParams(
+                args=(),
+                plugins=(),
+                dir=Path(rootpath),
+            ),
+        )
+        with mock.patch.object(hook, "fnpats", ["*.py"]):
+            assert hook.find_spec("file") is None
+
     @pytest.mark.skipif(
         sys.platform.startswith("win32"), reason="cannot remove cwd on Windows"
     )