tayvo_rocket_authifier 1.0.15

rocket.rs implementation of authifier
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77

//! Fetch available MFA methods.
//! GET /mfa/methods
use tayvo_authifier::models::{Account, MFAMethod};
use rocket::serde::json::Json;

/// # Get MFA Methods
///
/// Fetch available MFA methods.
#[openapi(tag = "MFA")]
#[get("/methods")]
pub async fn get_mfa_methods(account: Account) -> Json<Vec<MFAMethod>> {
    Json(account.mfa.get_methods())
}

#[cfg(test)]
#[cfg(feature = "test")]
mod tests {
    use tayvo_authifier::models::totp::Totp;

    use crate::test::*;

    #[async_std::test]
    async fn success() {
        use rocket::http::Header;

        let (authifier, session, _, _) = for_test_authenticated("get_mfa_methods::success").await;
        let client = bootstrap_rocket_with_auth(
            authifier,
            routes![crate::routes::mfa::get_mfa_methods::get_mfa_methods],
        )
        .await;

        let res = client
            .get("/methods")
            .header(Header::new("X-Session-Token", session.token))
            .dispatch()
            .await;

        assert_eq!(res.status(), Status::Ok);
        assert_eq!(
            serde_json::from_str::<Vec<MFAMethod>>(&res.into_string().await.unwrap()).unwrap(),
            vec![MFAMethod::Password]
        );
    }

    #[async_std::test]
    async fn success_has_recovery_and_totp() {
        use rocket::http::Header;

        let (authifier, session, mut account, _) =
            for_test_authenticated("get_mfa_methods::success_has_recovery_and_totp").await;

        account.mfa.totp_token = Totp::Enabled {
            secret: "some".to_string(),
        };
        account.mfa.generate_recovery_codes();
        account.save(&authifier).await.unwrap();

        let client = bootstrap_rocket_with_auth(
            authifier,
            routes![crate::routes::mfa::get_mfa_methods::get_mfa_methods],
        )
        .await;

        let res = client
            .get("/methods")
            .header(Header::new("X-Session-Token", session.token))
            .dispatch()
            .await;

        assert_eq!(res.status(), Status::Ok);
        assert_eq!(
            serde_json::from_str::<Vec<MFAMethod>>(&res.into_string().await.unwrap()).unwrap(),
            vec![MFAMethod::Totp, MFAMethod::Recovery]
        );
    }
}