De-dupe some test code

This commit is contained in:
Thomas Gideon 2017-02-17 12:54:50 -05:00
parent 3c95e1b163
commit a57e582337
3 changed files with 10 additions and 18 deletions

View file

@ -10,7 +10,7 @@ use medallion::{
Token,
};
fn load_key(keypath: &str) -> Result<String, Error> {
fn load_pem(keypath: &str) -> Result<String, Error> {
let mut key_file = File::open(keypath)?;
let mut key = String::new();
key_file.read_to_string(&mut key)?;
@ -34,13 +34,13 @@ fn new_token(user_id: &str, password: &str) -> Option<String> {
};
let token = Token::new(header, claims);
token.signed(load_key("./privateKey.pem").unwrap().as_bytes()).ok()
token.signed(load_pem("./privateKey.pem").unwrap().as_bytes()).ok()
}
fn login(token: &str) -> Option<String> {
let token = Token::<DefaultHeader, Registered>::parse(token).unwrap();
if token.verify(load_key("./publicKey.pub").unwrap().as_bytes()).unwrap() {
if token.verify(load_pem("./publicKey.pub").unwrap().as_bytes()).unwrap() {
token.claims.sub
} else {
None