From f86eb819ad7489133e3d45516dc7bfc8ad092a44 Mon Sep 17 00:00:00 2001 From: v <218731485+vrt-boss-bfx@users.noreply.github.com> Date: Sat, 16 Aug 2025 23:48:06 +0700 Subject: [PATCH] Make test client accept base environ from flask_test_client --- src/flask_socketio/test_client.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/src/flask_socketio/test_client.py b/src/flask_socketio/test_client.py index 312bac1d..f19d7d4b 100644 --- a/src/flask_socketio/test_client.py +++ b/src/flask_socketio/test_client.py @@ -128,7 +128,15 @@ def connect(self, namespace=None, query_string=None, headers=None, if query_string[0] != '?': query_string = '?' + query_string url += query_string - environ = EnvironBuilder(url, headers=headers).get_environ() + + environ_base = ( + None + if not self.flask_test_client + else self.flask_test_client.environ_base + ) + environ = EnvironBuilder( + url, headers=headers, environ_base=environ_base + ).get_environ() environ['flask.app'] = self.app if self.flask_test_client: # inject cookies from Flask