jsonwebtoken/tests/rsa.rs

52 lines
1.5 KiB
Rust
Raw Normal View History

2017-02-22 02:45:28 -05:00
extern crate jsonwebtoken;
#[macro_use]
extern crate serde_derive;
extern crate chrono;
2017-02-22 02:45:28 -05:00
use chrono::Utc;
2019-05-15 10:20:25 -04:00
use jsonwebtoken::{decode, encode, sign, verify, Algorithm, Der, Header, Pkcs8, Validation};
2017-02-22 02:45:28 -05:00
#[derive(Debug, PartialEq, Clone, Serialize, Deserialize)]
struct Claims {
sub: String,
company: String,
exp: i64,
2017-02-22 02:45:28 -05:00
}
2017-04-10 23:40:01 -04:00
#[test]
fn round_trip_sign_verification() {
2019-05-15 10:20:25 -04:00
let privkey = include_bytes!("private_rsa_key.der");
let encrypted = sign("hello world", Der::from(&&privkey[..]), Algorithm::RS256).unwrap();
2018-10-28 14:58:35 -04:00
let is_valid =
verify(&encrypted, "hello world", include_bytes!("public_rsa_key.der"), Algorithm::RS256)
.unwrap();
2017-04-10 23:40:01 -04:00
assert!(is_valid);
}
2017-02-22 02:45:28 -05:00
#[test]
fn round_trip_claim() {
let my_claims = Claims {
sub: "b@b.com".to_string(),
company: "ACME".to_string(),
exp: Utc::now().timestamp() + 10000,
2017-02-22 02:45:28 -05:00
};
2019-05-15 10:20:25 -04:00
let privkey = include_bytes!("private_rsa_key.der");
2018-10-28 14:58:35 -04:00
let token =
2019-05-15 10:20:25 -04:00
encode(&Header::new(Algorithm::RS256), &my_claims, Der::from(&&privkey[..])).unwrap();
2018-10-28 14:58:35 -04:00
let token_data = decode::<Claims>(
&token,
include_bytes!("public_rsa_key.der"),
&Validation::new(Algorithm::RS256),
)
.unwrap();
2017-02-22 02:45:28 -05:00
assert_eq!(my_claims, token_data.claims);
assert!(token_data.header.kid.is_none());
}
2019-05-15 10:20:25 -04:00
#[test]
#[should_panic(expected = "InvalidRsaKey")]
fn fails_with_different_key_format() {
let privkey = include_bytes!("private_rsa_key.der");
sign("hello world", Pkcs8::from(&&privkey[..]), Algorithm::RS256).unwrap();
}