2017-02-22 02:45:28 -05:00
|
|
|
extern crate jsonwebtoken;
|
|
|
|
#[macro_use]
|
|
|
|
extern crate serde_derive;
|
2018-07-25 09:42:00 -04:00
|
|
|
extern crate chrono;
|
2017-02-22 02:45:28 -05:00
|
|
|
|
2018-07-25 09:42:00 -04:00
|
|
|
use chrono::Utc;
|
2019-06-16 11:51:43 -04:00
|
|
|
use jsonwebtoken::{decode, encode, sign, verify, Algorithm, Key, Header, Validation};
|
2017-02-22 02:45:28 -05:00
|
|
|
|
|
|
|
#[derive(Debug, PartialEq, Clone, Serialize, Deserialize)]
|
|
|
|
struct Claims {
|
|
|
|
sub: String,
|
2018-07-25 09:42:00 -04:00
|
|
|
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");
|
2019-06-16 11:51:43 -04:00
|
|
|
let encrypted = sign("hello world", Key::Der(&privkey[..]), Algorithm::RS256).unwrap();
|
2018-10-28 14:58:35 -04:00
|
|
|
let is_valid =
|
2019-06-16 11:51:43 -04:00
|
|
|
verify(&encrypted, "hello world", Key::Der(include_bytes!("public_rsa_key.der")), Algorithm::RS256)
|
2018-10-28 14:58:35 -04:00
|
|
|
.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(),
|
2018-07-25 09:42:00 -04:00
|
|
|
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-06-16 11:51:43 -04:00
|
|
|
encode(&Header::new(Algorithm::RS256), &my_claims, Key::Der(&privkey[..])).unwrap();
|
2018-10-28 14:58:35 -04:00
|
|
|
let token_data = decode::<Claims>(
|
|
|
|
&token,
|
2019-06-16 11:51:43 -04:00
|
|
|
Key::Der(include_bytes!("public_rsa_key.der")),
|
2018-10-28 14:58:35 -04:00
|
|
|
&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");
|
2019-06-16 11:51:43 -04:00
|
|
|
sign("hello world", Key::Pkcs8(&privkey[..]), Algorithm::RS256).unwrap();
|
2019-05-15 10:20:25 -04:00
|
|
|
}
|