From 33528870ae2bada74769e0cda9791f71576aa947 Mon Sep 17 00:00:00 2001
From: allegroai <>
Date: Mon, 18 Mar 2024 15:45:09 +0200
Subject: [PATCH] Request cookies processing enhanced for more flexibility

---
 apiserver/server_init/request_handlers.py | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/apiserver/server_init/request_handlers.py b/apiserver/server_init/request_handlers.py
index 992a5f9..a8c9182 100644
--- a/apiserver/server_init/request_handlers.py
+++ b/apiserver/server_init/request_handlers.py
@@ -148,6 +148,11 @@ class RequestHandlers:
         call.set_error_result(msg=msg, code=code, subcode=subcode)
         return call
 
+    def _get_session_auth_cookie(self, req):
+        return req.cookies.get(
+            config.get("apiserver.auth.session_auth_cookie_name")
+        )
+
     def _create_api_call(self, req):
         call = None
         try:
@@ -161,9 +166,7 @@ class RequestHandlers:
 
             # Resolve authorization: if cookies contain an authorization token, use it as a starting point.
             # in any case, request headers always take precedence.
-            auth_cookie = req.cookies.get(
-                config.get("apiserver.auth.session_auth_cookie_name")
-            )
+            auth_cookie = self._get_session_auth_cookie(req)
             headers = (
                 {}
                 if not auth_cookie