werkzeug 2.3.3 compatibility

pallets · May 2, 2023 · 3fbfbad · 3fbfbad
1 parent 726d3f4
commit 3fbfbad
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 3 deletions.
diff --git a/CHANGES.rst b/CHANGES.rst
@@ -3,6 +3,8 @@ Version 2.2.5
 
 Unreleased
 
+-   Update for compatibility with Werkzeug 2.3.3.
+
 
 Version 2.2.4
 -------------

diff --git a/src/flask/testing.py b/src/flask/testing.py
@@ -168,10 +168,21 @@ def session_transaction(
             app.session_interface.save_session(app, sess, resp)
 
         if hasattr(self, "_update_cookies_from_response"):
-            self._update_cookies_from_response(
-                ctx.request.host.partition(":")[0], resp.headers.getlist("Set-Cookie")
-            )
+            try:
+                # Werkzeug>=2.3.3
+                self._update_cookies_from_response(
+                    ctx.request.host.partition(":")[0],
+                    ctx.request.path,
+                    resp.headers.getlist("Set-Cookie"),
+                )
+            except TypeError:
+                # Werkzeug>=2.3.0,<2.3.3
+                self._update_cookies_from_response(  # type: ignore[call-arg]
+                    ctx.request.host.partition(":")[0],
+                    resp.headers.getlist("Set-Cookie"),  # type: ignore[arg-type]
+                )
         else:
+            # Werkzeug<2.3.0
             self.cookie_jar.extract_wsgi(  # type: ignore[union-attr]
                 ctx.request.environ, resp.headers
             )