use chrono::{SubsecRound, Utc};
use laboratory::expect;
use tokio::runtime::Runtime;

use sylvia_iot_auth::models::refresh_token::{QueryCond, RefreshToken, RefreshTokenModel};

/// Test `add()`.
pub fn add(runtime: &Runtime, model: &dyn RefreshTokenModel) -> Result<(), String> {
    let token = RefreshToken {
        refresh_token: "token_add_none".to_string(),
        expires_at: Utc::now().trunc_subsecs(3),
        scope: Some("scope_add".to_string()),
        client_id: "client_id_add".to_string(),
        redirect_uri: "redirect_uri_add".to_string(),
        user_id: "user_id_add".to_string(),
    };
    if let Err(e) = runtime.block_on(async { model.add(&token).await }) {
        return Err(format!("model.add() none error: {}", e));
    }

    let get_token = match runtime.block_on(async { model.get(&token.refresh_token).await }) {
        Err(e) => return Err(format!("model.get() none error: {}", e)),
        Ok(token) => match token {
            None => return Err("should get none one".to_string()),
            Some(token) => token,
        },
    };
    expect(get_token).to_equal(token)?;

    let token = RefreshToken {
        refresh_token: "token_add_some".to_string(),
        expires_at: Utc::now().trunc_subsecs(3),
        scope: Some("scope_add".to_string()),
        client_id: "client_id_add".to_string(),
        redirect_uri: "redirect_uri_add".to_string(),
        user_id: "user_id_add".to_string(),
    };
    if let Err(e) = runtime.block_on(async { model.add(&token).await }) {
        return Err(format!("model.add() some error: {}", e));
    }

    let get_token = match runtime.block_on(async { model.get(&token.refresh_token).await }) {
        Err(e) => return Err(format!("model.get() some error: {}", e)),
        Ok(token) => match token {
            None => return Err("should get some one".to_string()),
            Some(token) => token,
        },
    };
    expect(get_token).to_equal(token)
}

/// Test `add()` with duplicate key.
pub fn add_dup(runtime: &Runtime, model: &dyn RefreshTokenModel) -> Result<(), String> {
    let token = RefreshToken {
        refresh_token: "token_add".to_string(),
        expires_at: Utc::now().trunc_subsecs(3),
        scope: Some("scope_add".to_string()),
        client_id: "client_id_add".to_string(),
        redirect_uri: "redirect_uri_add".to_string(),
        user_id: "user_id_add".to_string(),
    };
    if let Err(e) = runtime.block_on(async { model.add(&token).await }) {
        return Err(format!("model.add() error: {}", e));
    }
    if let Ok(_) = runtime.block_on(async { model.add(&token).await }) {
        return Err("model.add() duplicate should error".to_string());
    }
    Ok(())
}

/// Test `del()` by specifying a refresh token.
pub fn del_by_token(runtime: &Runtime, model: &dyn RefreshTokenModel) -> Result<(), String> {
    let token_del = "token_del";
    let token_not_del = "token_not_del";
    let mut token = RefreshToken {
        refresh_token: token_del.to_string(),
        expires_at: Utc::now().trunc_subsecs(3),
        scope: Some("scope_del".to_string()),
        client_id: "client_id_del".to_string(),
        redirect_uri: "redirect_uri_del".to_string(),
        user_id: "user_id_del".to_string(),
    };
    let cond = QueryCond {
        refresh_token: Some(token_del),
        ..Default::default()
    };
    if let Err(e) = runtime.block_on(async {
        model.add(&token).await?;
        token.refresh_token = token_not_del.to_string();
        model.add(&token).await?;
        model.del(&cond).await
    }) {
        return Err(format!("model.add/del error: {}", e));
    }
    match runtime.block_on(async { model.get(token_del).await }) {
        Err(e) => return Err(format!("model.get() error: {}", e)),
        Ok(token) => match token {
            None => (),
            Some(_) => return Err("delete fail".to_string()),
        },
    }
    match runtime.block_on(async { model.get(token_not_del).await }) {
        Err(e) => Err(format!("model.get() not delete one error: {}", e)),
        Ok(token) => match token {
            None => Err("delete wrong one".to_string()),
            Some(_) => Ok(()),
        },
    }
}

/// Test `del()` twice.
pub fn del_twice(runtime: &Runtime, model: &dyn RefreshTokenModel) -> Result<(), String> {
    let token_del = "token_del";
    let token = RefreshToken {
        refresh_token: token_del.to_string(),
        expires_at: Utc::now().trunc_subsecs(3),
        scope: Some("scope_del".to_string()),
        client_id: "client_id_del".to_string(),
        redirect_uri: "redirect_uri_del".to_string(),
        user_id: "user_id_del".to_string(),
    };
    let cond = QueryCond {
        refresh_token: Some(token_del),
        ..Default::default()
    };
    if let Err(e) = runtime.block_on(async {
        model.add(&token).await?;
        model.del(&cond).await?;
        model.del(&cond).await
    }) {
        return Err(format!("model.add/del error: {}", e));
    }
    Ok(())
}

/// Test `del()` by specifying a client ID.
pub fn del_by_client_id(runtime: &Runtime, model: &dyn RefreshTokenModel) -> Result<(), String> {
    let token_del1 = "token_del1";
    let token_del2 = "token_del2";
    let token_not_del = "token_not_del";
    let mut token = RefreshToken {
        refresh_token: token_del1.to_string(),
        expires_at: Utc::now().trunc_subsecs(3),
        scope: Some("scope_del".to_string()),
        client_id: "client_id_del".to_string(),
        redirect_uri: "redirect_uri_del".to_string(),
        user_id: "user_id_del".to_string(),
    };
    let cond = QueryCond {
        client_id: Some("client_id_del"),
        ..Default::default()
    };
    if let Err(e) = runtime.block_on(async {
        model.add(&token).await?;
        token.refresh_token = token_del2.to_string();
        model.add(&token).await?;
        token.refresh_token = token_not_del.to_string();
        token.client_id = "client_id_not_del".to_string();
        model.add(&token).await?;
        model.del(&cond).await
    }) {
        return Err(format!("model.add/del error: {}", e));
    }
    match runtime.block_on(async { model.get(token_del1).await }) {
        Err(e) => return Err(format!("model.get() delete token1 error: {}", e)),
        Ok(token) => match token {
            None => (),
            Some(_) => return Err("delete token1 fail".to_string()),
        },
    }
    match runtime.block_on(async { model.get(token_del2).await }) {
        Err(e) => return Err(format!("model.get() delete token2 error: {}", e)),
        Ok(token) => match token {
            None => (),
            Some(_) => return Err("delete token2 fail".to_string()),
        },
    }
    match runtime.block_on(async { model.get(token_not_del).await }) {
        Err(e) => Err(format!("model.get() not delete one error: {}", e)),
        Ok(token) => match token {
            None => Err("delete wrong one".to_string()),
            Some(_) => Ok(()),
        },
    }
}

/// Test `del()` by specifying a user ID.
pub fn del_by_user_id(runtime: &Runtime, model: &dyn RefreshTokenModel) -> Result<(), String> {
    let token_del1 = "token_del1";
    let token_del2 = "token_del2";
    let token_not_del = "token_not_del";
    let mut token = RefreshToken {
        refresh_token: token_del1.to_string(),
        expires_at: Utc::now().trunc_subsecs(3),
        scope: Some("scope_del".to_string()),
        client_id: "client_id_del".to_string(),
        redirect_uri: "redirect_uri_del".to_string(),
        user_id: "user_id_del".to_string(),
    };
    let cond = QueryCond {
        user_id: Some("user_id_del"),
        ..Default::default()
    };
    if let Err(e) = runtime.block_on(async {
        model.add(&token).await?;
        token.refresh_token = token_del2.to_string();
        model.add(&token).await?;
        token.refresh_token = token_not_del.to_string();
        token.user_id = "user_id_not_del".to_string();
        model.add(&token).await?;
        model.del(&cond).await
    }) {
        return Err(format!("model.add/del error: {}", e));
    }
    match runtime.block_on(async { model.get(token_del1).await }) {
        Err(e) => return Err(format!("model.get() delete token1 error: {}", e)),
        Ok(token) => match token {
            None => (),
            Some(_) => return Err("delete token1 fail".to_string()),
        },
    }
    match runtime.block_on(async { model.get(token_del2).await }) {
        Err(e) => return Err(format!("model.get() delete token2 error: {}", e)),
        Ok(token) => match token {
            None => (),
            Some(_) => return Err("delete token2 fail".to_string()),
        },
    }
    match runtime.block_on(async { model.get(token_not_del).await }) {
        Err(e) => Err(format!("model.get() not delete one error: {}", e)),
        Ok(token) => match token {
            None => Err("delete wrong one".to_string()),
            Some(_) => Ok(()),
        },
    }
}

/// Test `del()` by specifying a pair of user ID and client ID.
pub fn del_by_user_client(runtime: &Runtime, model: &dyn RefreshTokenModel) -> Result<(), String> {
    let token_del = "token_del";
    let token_not_del = "token_not_del";
    let mut token = RefreshToken {
        refresh_token: token_del.to_string(),
        expires_at: Utc::now().trunc_subsecs(3),
        scope: Some("scope_del".to_string()),
        client_id: "client_id_del".to_string(),
        redirect_uri: "redirect_uri_del".to_string(),
        user_id: "user_id_del".to_string(),
    };
    let cond = QueryCond {
        client_id: Some("client_id_del"),
        user_id: Some("user_id_del"),
        ..Default::default()
    };
    if let Err(e) = runtime.block_on(async {
        model.add(&token).await?;
        token.refresh_token = token_not_del.to_string();
        token.user_id = "user_id_not_del".to_string();
        model.add(&token).await?;
        model.del(&cond).await
    }) {
        return Err(format!("model.add/del error: {}", e));
    }
    match runtime.block_on(async { model.get(token_del).await }) {
        Err(e) => return Err(format!("model.get() delete one error: {}", e)),
        Ok(token) => match token {
            None => (),
            Some(_) => return Err("delete one fail".to_string()),
        },
    }
    match runtime.block_on(async { model.get(token_not_del).await }) {
        Err(e) => Err(format!("model.get() not delete one error: {}", e)),
        Ok(token) => match token {
            None => Err("delete wrong one".to_string()),
            Some(_) => Ok(()),
        },
    }
}