diff --git a/src/error.rs b/src/error.rs index af8f863..c940b53 100644 --- a/src/error.rs +++ b/src/error.rs @@ -53,6 +53,12 @@ pub struct ApiError { message: String, } +impl ApiError { + pub fn status_code(&self) -> u16 { + self.status + } +} + impl Display for ApiError { fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { write!(f, "{}: {}", self.status, self.message) diff --git a/src/jwt.rs b/src/jwt.rs index 2b8be14..64f688b 100644 --- a/src/jwt.rs +++ b/src/jwt.rs @@ -26,4 +26,6 @@ pub(crate) fn decode_jwt(token: &str, hmac_secret: &str) -> Result, } diff --git a/src/tests/jwt.rs b/src/tests/jwt.rs index b2cb5a6..cc2ba66 100644 --- a/src/tests/jwt.rs +++ b/src/tests/jwt.rs @@ -36,6 +36,7 @@ fn decode_jwt_expired_token() { let claims = Claims { prompt: "expired prompt".to_string(), exp: 1, // Expired long ago (Unix epoch) + iss: Some("test".to_string()), }; let token = jsonwebtoken::encode( &jsonwebtoken::Header::default(), @@ -54,6 +55,7 @@ fn decode_jwt_valid_token() { let claims = Claims { prompt: "test prompt".to_string(), exp: 9999999999, + iss: Some("test".to_string()), }; let token = jsonwebtoken::encode( &jsonwebtoken::Header::default(), @@ -81,6 +83,7 @@ fn message_encrypted_variant_decrypts() { let claims = Claims { prompt: "encrypted prompt".to_string(), exp: 9999999999, + iss: Some("test".to_string()), }; let token = jsonwebtoken::encode( &jsonwebtoken::Header::default(),