use super::forms::RegistrationForm;
use super::responses::LoggedInUserData;
use crate::common::asserts::assert_json_ok_response;
use crate::common::contexts::user::responses::{
    AddedUserResponse, BannedUserResponse, SuccessfulLoginResponse, TokenRenewalData, TokenRenewalResponse, TokenVerifiedResponse,
};
use crate::common::responses::TextResponse;

pub fn assert_added_user_response(response: &TextResponse) {
    let _added_user_response: AddedUserResponse = serde_json::from_str(&response.body)
        .unwrap_or_else(|_| panic!("response {:#?} should be a AddedUserResponse", response.body));
    assert_json_ok_response(response);
}

pub fn assert_successful_login_response(response: &TextResponse, registered_user: &RegistrationForm) {
    let successful_login_response: SuccessfulLoginResponse = serde_json::from_str(&response.body)
        .unwrap_or_else(|_| panic!("response {:#?} should be a SuccessfulLoginResponse", response.body));

    let logged_in_user = successful_login_response.data;

    assert_eq!(logged_in_user.username, registered_user.username);

    assert_json_ok_response(response);
}

pub fn assert_token_verified_response(response: &TextResponse) {
    let token_verified_response: TokenVerifiedResponse = serde_json::from_str(&response.body)
        .unwrap_or_else(|_| panic!("response {:#?} should be a TokenVerifiedResponse", response.body));

    assert_eq!(token_verified_response.data, "Token is valid.");

    assert_json_ok_response(response);
}

pub fn assert_token_renewal_response(response: &TextResponse, logged_in_user: &LoggedInUserData) {
    let token_renewal_response: TokenRenewalResponse = serde_json::from_str(&response.body)
        .unwrap_or_else(|_| panic!("response {:#?} should be a TokenRenewalResponse", response.body));

    assert_eq!(
        token_renewal_response.data,
        TokenRenewalData {
            token: logged_in_user.token.clone(),
            username: logged_in_user.username.clone(),
            admin: logged_in_user.admin,
        }
    );

    assert_json_ok_response(response);
}

pub fn assert_banned_user_response(response: &TextResponse, registered_user: &RegistrationForm) {
    let banned_user_response: BannedUserResponse = serde_json::from_str(&response.body)
        .unwrap_or_else(|_| panic!("response {:#?} should be a BannedUserResponse", response.body));

    assert_eq!(
        banned_user_response.data,
        format!("Banned user: {}", registered_user.username)
    );

    assert_json_ok_response(response);
}