@@ -249,12 +249,14 @@ def test_authenticate_success(self, session_constants, mock_user_management):
249249 "feature_flags" : ["flag1" , "flag2" ],
250250 }
251251
252- with patch .object (Session , "unseal_data" , return_value = mock_session ), patch (
253- "jwt.decode" , return_value = mock_jwt_payload
254- ), patch .object (
255- session .jwks ,
256- "get_signing_key_from_jwt" ,
257- return_value = Mock (key = session_constants ["PUBLIC_KEY" ]),
252+ with (
253+ patch .object (Session , "unseal_data" , return_value = mock_session ),
254+ patch ("jwt.decode" , return_value = mock_jwt_payload ),
255+ patch .object (
256+ session .jwks ,
257+ "get_signing_key_from_jwt" ,
258+ return_value = Mock (key = session_constants ["PUBLIC_KEY" ]),
259+ ),
258260 ):
259261 response = session .authenticate ()
260262
@@ -319,12 +321,14 @@ def test_authenticate_success_with_roles(
319321 "feature_flags" : ["flag1" , "flag2" ],
320322 }
321323
322- with patch .object (Session , "unseal_data" , return_value = mock_session ), patch (
323- "jwt.decode" , return_value = mock_jwt_payload
324- ), patch .object (
325- session .jwks ,
326- "get_signing_key_from_jwt" ,
327- return_value = Mock (key = session_constants ["PUBLIC_KEY" ]),
324+ with (
325+ patch .object (Session , "unseal_data" , return_value = mock_session ),
326+ patch ("jwt.decode" , return_value = mock_jwt_payload ),
327+ patch .object (
328+ session .jwks ,
329+ "get_signing_key_from_jwt" ,
330+ return_value = Mock (key = session_constants ["PUBLIC_KEY" ]),
331+ ),
328332 ):
329333 response = session .authenticate ()
330334
0 commit comments