use tayvo_authifier::{
models::{Account, ValidatedTicket},
Authifier, Result,
};
use rocket::State;
use tayvo_rocket_empty::EmptyResponse;
#[openapi(tag = "Account")]
#[post("/disable")]
pub async fn disable_account(
authifier: &State<Authifier>,
mut account: Account,
_ticket: ValidatedTicket,
) -> Result<EmptyResponse> {
account.disable(authifier).await.map(|_| EmptyResponse)
}
#[cfg(test)]
#[cfg(feature = "test")]
mod tests {
use crate::test::*;
#[async_std::test]
async fn success() {
use rocket::http::Header;
let (authifier, session, account, receiver) =
for_test_authenticated("disable_account::success").await;
let ticket = MFATicket::new(account.id.to_string(), true);
ticket.save(&authifier).await.unwrap();
let client = bootstrap_rocket_with_auth(
authifier.clone(),
routes![crate::routes::account::disable_account::disable_account],
)
.await;
let res = client
.post("/disable")
.header(Header::new("X-Session-Token", session.token))
.header(Header::new("X-MFA-Ticket", ticket.token))
.header(ContentType::JSON)
.dispatch()
.await;
assert_eq!(res.status(), Status::NoContent);
assert!(
authifier
.database
.find_account(&account.id)
.await
.unwrap()
.disabled
);
assert_eq!(
authifier
.database
.find_session(&session.id)
.await
.unwrap_err(),
Error::UnknownUser
);
let event = receiver.try_recv().expect("an event");
if let AuthifierEvent::DeleteAllSessions { user_id, .. } = event {
assert_eq!(user_id, session.user_id);
} else {
panic!("Received incorrect event type. {:?}", event);
}
}
}