jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

test: add comprehensive unit tests for PassportService with exception handling optimization (#22268)

Browse source
This commit is contained in:
Jason Young 2025-07-12 19:56:20 +08:00 • committed by GitHub
commit 253d8e5a5f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 209 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
api/libs/passport.py
View file

@ -14,9 +14,11 @@ class PassportService:
def verify(self, token):
try:
return jwt.decode(token, self.sk, algorithms=["HS256"])
except jwt.exceptions.ExpiredSignatureError:
raise Unauthorized("Token has expired.")
except jwt.exceptions.InvalidSignatureError:
raise Unauthorized("Invalid token signature.")
except jwt.exceptions.DecodeError:
raise Unauthorized("Invalid token.")
except jwt.exceptions.ExpiredSignatureError:
raise Unauthorized("Token has expired.")
except jwt.exceptions.PyJWTError: # Catch-all for other JWT errors
raise Unauthorized("Invalid token.")