4242_TEST_JWT_AUDIENCE = "https://api.example.com"
4343_TEST_BEARER_TOKEN = "eyJhbGciOiJSUzI1NiIsImtpZCI6InRlc3QifQ.test.test" # noqa: S105
4444_TEST_KID = "test-kid"
45+ _FETCH_JWKS_PATH = "aignostics_foundry_core.api.auth._fetch_jwks"
4546
4647
4748@pytest .mark .unit
@@ -397,7 +398,7 @@ async def test_validate_jwt_returns_none_when_kid_not_in_jwks(self) -> None:
397398 jwks_without_kid : dict = {"keys" : []}
398399 settings = AuthSettings (jwt_enabled = True , domain = _TEST_DOMAIN , jwt_audience = _TEST_JWT_AUDIENCE )
399400
400- with patch ("aignostics_foundry_core.api.auth._fetch_jwks" , AsyncMock (return_value = jwks_without_kid )):
401+ with patch (_FETCH_JWKS_PATH , AsyncMock (return_value = jwks_without_kid )):
401402 import jwt
402403
403404 with patch .object (jwt , "get_unverified_header" , return_value = {"kid" : _TEST_KID , "alg" : "RS256" }):
@@ -409,9 +410,7 @@ async def test_validate_jwt_returns_none_on_fetch_failure(self) -> None:
409410 """_validate_jwt returns None when JWKS fetch raises an exception."""
410411 settings = AuthSettings (jwt_enabled = True , domain = _TEST_DOMAIN , jwt_audience = _TEST_JWT_AUDIENCE )
411412
412- with patch (
413- "aignostics_foundry_core.api.auth._fetch_jwks" , AsyncMock (side_effect = RuntimeError ("network error" ))
414- ):
413+ with patch (_FETCH_JWKS_PATH , AsyncMock (side_effect = RuntimeError ("network error" ))):
415414 result = await _validate_jwt (_TEST_BEARER_TOKEN , settings )
416415
417416 assert result is None
@@ -424,7 +423,7 @@ async def test_validate_jwt_returns_none_for_invalid_token(self, mock_jwks: dict
424423 settings = AuthSettings (jwt_enabled = True , domain = _TEST_DOMAIN , jwt_audience = _TEST_JWT_AUDIENCE )
425424
426425 with (
427- patch ("aignostics_foundry_core.api.auth._fetch_jwks" , AsyncMock (return_value = mock_jwks )),
426+ patch (_FETCH_JWKS_PATH , AsyncMock (return_value = mock_jwks )),
428427 patch .object (jwt , "get_unverified_header" , return_value = {"kid" : _TEST_KID , "alg" : "RS256" }),
429428 patch .object (RSAAlgorithm , "from_jwk" , return_value = MagicMock ()),
430429 patch .object (jwt , "decode" , side_effect = jwt .ExpiredSignatureError ("expired" )),
@@ -442,7 +441,7 @@ async def test_validate_jwt_returns_payload_for_valid_token(self, mock_jwks: dic
442441 settings = AuthSettings (jwt_enabled = True , domain = _TEST_DOMAIN , jwt_audience = _TEST_JWT_AUDIENCE )
443442
444443 with (
445- patch ("aignostics_foundry_core.api.auth._fetch_jwks" , AsyncMock (return_value = mock_jwks )),
444+ patch (_FETCH_JWKS_PATH , AsyncMock (return_value = mock_jwks )),
446445 patch .object (jwt , "get_unverified_header" , return_value = {"kid" : _TEST_KID , "alg" : "RS256" }),
447446 patch .object (RSAAlgorithm , "from_jwk" , return_value = MagicMock ()),
448447 patch .object (jwt , "decode" , return_value = expected_payload ),
0 commit comments