web3-proxy/web3_proxy/tests/common/create_user.rs

80 lines
2.1 KiB
Rust

use crate::TestApp;
use entities::{user, user_tier};
use ethers::prelude::{LocalWallet, Signer};
use ethers::types::Signature;
use migration::sea_orm::{
self, ActiveModelTrait, ColumnTrait, EntityTrait, IntoActiveModel, QueryFilter,
};
use tracing::info;
use web3_proxy::errors::Web3ProxyResult;
use web3_proxy::frontend::users::authentication::{LoginPostResponse, PostLogin};
use super::mysql::TestMysql;
/// Helper function to create an "ordinary" user
#[allow(unused)]
pub async fn create_user(
x: &TestApp,
r: &reqwest::Client,
user_wallet: &LocalWallet,
referral_code: Option<String>,
) -> (LoginPostResponse) {
let login_post_url = format!("{}user/login", x.proxy_provider.url());
let user_login_get_url = format!(
"{}user/login/{:?}",
x.proxy_provider.url(),
user_wallet.address()
);
let user_login_message = r.get(user_login_get_url).send().await.unwrap();
let user_login_message = user_login_message.text().await.unwrap();
// Sign the message and POST it to login as the user
let user_signed: Signature = user_wallet.sign_message(&user_login_message).await.unwrap();
info!(?user_signed);
let user_post_login_data = PostLogin {
msg: user_login_message,
sig: user_signed.to_string(),
referral_code,
};
info!(?user_post_login_data);
let mut user_login_response = r
.post(&login_post_url)
.json(&user_post_login_data)
.send()
.await
.unwrap()
.json::<LoginPostResponse>()
.await
.unwrap();
info!(?user_login_response);
user_login_response
}
/// TODO: use an admin user to do this instead
#[allow(unused)]
pub async fn set_user_tier(
x: &TestApp,
db: &TestMysql,
user: user::Model,
tier_name: &str,
) -> Web3ProxyResult<user_tier::Model> {
let db_conn = db.conn();
let ut = user_tier::Entity::find()
.filter(user_tier::Column::Title.like(tier_name))
.one(db_conn)
.await?
.unwrap();
let mut user = user.into_active_model();
user.user_tier_id = sea_orm::Set(ut.id);
user.save(db_conn).await?;
Ok(ut)
}