8d31680ff1
* working on test for referral * refactored logic in tests to helper functions * added getbalance helper * added more helper functions to test for referral * gotta be able to submit to rpc now * adding provider instead * referral tests are not passing, may be because we need for stats to be updated * balance doesnt decrease * will look into deposits in the meantime * finding some first bugs * seems to have fixed a bunch of stuff * feedback, lint, and make it compile --------- Co-authored-by: Bryan Stitt <bryan@llamanodes.com>
56 lines
1.5 KiB
Rust
56 lines
1.5 KiB
Rust
use crate::TestApp;
|
|
use serde::Deserialize;
|
|
use tracing::info;
|
|
use ulid::Ulid;
|
|
use web3_proxy::frontend::users::authentication::LoginPostResponse;
|
|
|
|
#[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 {
|
|
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!("Rpc Key");
|
|
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
|
|
}
|