Add jwt headers in JsonClient if exist

PnX-SI · Sep 13, 2023 · b8860dc · b8860dc
1 parent a80e4d6
commit b8860dc
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/utils_flask_sqla/tests/utils.py b/src/utils_flask_sqla/tests/utils.py
@@ -24,5 +24,11 @@ def open(self, *args, **kwargs):
                     }
                 )
             )
+        # the set_logged_user_jwt add a jwt attribute to client with the current logged user
+        jwt = getattr(self, "jwt")
+        if jwt:
+            jwt_headers = Headers({"Authorization": f"Bearer {jwt}"})
+            headers.extend(jwt_headers)
+
         kwargs["headers"] = headers
         return super().open(*args, **kwargs)