Pass context when using select

diff --git a/CHANGES.rst b/CHANGES.rst
index e31de857d3f6e939f9eb56d038efe159b633eab7..521f5a08a3984b6e14b2c5d03f177f1d778bbf6a 100644 (file)
--- a/CHANGES.rst
+++ b/CHANGES.rst
@@ -41,6 +41,8 @@ Unreleased
     contain the templates directory. :issue:`1705`
 -   Improve annotations for methods returning copies. :pr:`1880`
 -   ``urlize`` does not add ``mailto:`` to values like `@a@b`. :pr:`1870`
+-   Tests decorated with `@pass_context`` can be used with the ``|select``
+    filter. :issue:`1624`
 
 
 Version 3.1.4

diff --git a/src/jinja2/filters.py b/src/jinja2/filters.py
index a92832a34063c9ae13845432e407e7db3bf8e261..e5b5a00c5e8cb31f69c07c807594db1c492fd116 100644 (file)
--- a/src/jinja2/filters.py
+++ b/src/jinja2/filters.py
@@ -1780,7 +1780,7 @@ def prepare_select_or_reject(
         args = args[1 + off :]
 
         def func(item: t.Any) -> t.Any:
-            return context.environment.call_test(name, item, args, kwargs)
+            return context.environment.call_test(name, item, args, kwargs, context)
 
     except LookupError:
         func = bool  # type: ignore

diff --git a/tests/test_regression.py b/tests/test_regression.py
index 7bd4d15649af566ab5a715609eac8398b25013dc..10df2d1bd0b72e001729497bbeb37e4683c3aea1 100644 (file)
--- a/tests/test_regression.py
+++ b/tests/test_regression.py
@@ -737,6 +737,18 @@ End"""
         )
         assert tmpl.render() == "hellohellohello"
 
+    def test_pass_context_with_select(self, env):
+        @pass_context
+        def is_foo(ctx, s):
+            assert ctx is not None
+            return s == "foo"
+
+        env.tests["foo"] = is_foo
+        tmpl = env.from_string(
+            "{% for x in ['one', 'foo'] | select('foo') %}{{ x }}{% endfor %}"
+        )
+        assert tmpl.render() == "foo"
+
 
 @pytest.mark.parametrize("unicode_char", ["\N{FORM FEED}", "\x85"])
 def test_unicode_whitespace(env, unicode_char):