web3-proxy/web3_proxy/tests/common/rpc_key.rs
2023-07-14 18:31:00 -07:00

84 lines
2.2 KiB
Rust

use std::time::Duration;
use crate::TestApp;
use serde::Deserialize;
use tracing::info;
use ulid::Ulid;
use web3_proxy::{
errors::Web3ProxyResult,
frontend::users::authentication::LoginPostResponse,
rpcs::provider::{connect_http, EthersHttpProvider},
};
#[derive(Debug, Deserialize)]
pub struct RpcKeyResponse {
pub user_id: u64,
pub user_rpc_keys: std::collections::HashMap<String, RpcKey>,
}
#[derive(Debug, Deserialize)]
pub struct RpcKey {
pub active: bool,
pub allowed_ips: Option<serde_json::Value>,
pub allowed_origins: Option<serde_json::Value>,
pub allowed_referers: Option<serde_json::Value>,
pub allowed_user_agents: Option<serde_json::Value>,
pub description: Option<serde_json::Value>,
pub id: u64,
pub log_revert_chance: f64,
pub private_txs: bool,
pub role: String,
pub secret_key: Ulid,
pub user_id: u64,
}
/// Helper function to get the user's balance
#[allow(unused)]
pub async fn user_get_first_rpc_key(
x: &TestApp,
r: &reqwest::Client,
login_response: &LoginPostResponse,
) -> RpcKey {
// TODO: refactor to use login_response? or compare?
let get_keys = format!("{}user/keys", x.proxy_provider.url());
info!("Get balance");
let rpc_key_response = r
.get(get_keys)
.bearer_auth(login_response.bearer_token)
.send()
.await
.unwrap();
info!(?rpc_key_response);
let rpc_key_response = rpc_key_response.json::<serde_json::Value>().await.unwrap();
info!(?rpc_key_response);
info!("Parsing rpc key as json");
let rpc_key: RpcKeyResponse = serde_json::from_value(rpc_key_response).unwrap();
info!(?rpc_key);
rpc_key.user_rpc_keys.into_iter().next().unwrap().1
}
#[allow(unused)]
pub async fn user_get_provider(
x: &TestApp,
r: &reqwest::Client,
login_response: &LoginPostResponse,
) -> Web3ProxyResult<EthersHttpProvider> {
let first_key = login_response.rpc_keys.iter().next().unwrap().1;
let rpc_url = format!(
"{}rpc/{}",
x.proxy_provider.url(),
Ulid::from(first_key.secret_key)
);
connect_http(
rpc_url.parse().unwrap(),
Some(r.clone()),
Duration::from_secs(1),
)
}