From bc6d18041d652502f83d44aedbea234e02b59f6f Mon Sep 17 00:00:00 2001 From: yenicelik Date: Fri, 21 Jul 2023 18:02:07 -0400 Subject: [PATCH] added single proxy test --- web3_proxy/tests/common/app.rs | 4 +- web3_proxy/tests/test_single_proxy.rs | 281 ++++++++++++++++++++++++++ 2 files changed, 283 insertions(+), 2 deletions(-) create mode 100644 web3_proxy/tests/test_single_proxy.rs diff --git a/web3_proxy/tests/common/app.rs b/web3_proxy/tests/common/app.rs index 3c050f07..8e95fcdd 100644 --- a/web3_proxy/tests/common/app.rs +++ b/web3_proxy/tests/common/app.rs @@ -51,7 +51,7 @@ impl TestApp { influx: Option<&TestInflux>, ) -> Self { let chain_id = anvil.instance.chain_id(); - let num_workers = 2; + let num_workers = 4; // TODO: move basic setup into a test fixture let path = env::var("PATH").unwrap(); @@ -128,7 +128,7 @@ impl TestApp { thread::spawn(move || { let runtime = Builder::new_multi_thread() .enable_all() - .worker_threads(2) + .worker_threads(num_workers) .build() .unwrap(); diff --git a/web3_proxy/tests/test_single_proxy.rs b/web3_proxy/tests/test_single_proxy.rs new file mode 100644 index 00000000..33ed4e3b --- /dev/null +++ b/web3_proxy/tests/test_single_proxy.rs @@ -0,0 +1,281 @@ +mod common; + +use crate::common::create_provider_with_rpc_key::create_provider_for_user; +use crate::common::influx::TestInflux; +use crate::common::rpc_key::user_get_first_rpc_key; +use crate::common::stats_accounting::{user_get_influx_stats_aggregated, user_get_mysql_stats}; +use crate::common::user_balance::user_get_balance; +use crate::common::{ + admin_increases_balance::admin_increase_balance, anvil::TestAnvil, + create_admin::create_user_as_admin, create_user::create_user, mysql::TestMysql, TestApp, +}; +use futures::future::try_join_all; +use rust_decimal::Decimal; +use std::str::FromStr; +use std::sync::Arc; +use std::time::Duration; +use tokio::time::sleep; +use tracing::{info, warn}; +use web3_proxy::rpcs::blockchain::ArcBlock; + +#[cfg_attr(not(feature = "tests-needing-docker"), ignore)] +#[test_log::test(tokio::test)] +async fn test_single_proxy_stats_add_up() { + let chain_id = 999_001_999; + let a = TestAnvil::spawn(chain_id).await; + + let db = TestMysql::spawn().await; + + let influx = TestInflux::spawn().await; + + let r = reqwest::Client::builder() + .timeout(Duration::from_secs(20)) + .build() + .unwrap(); + + // Since when do indices start with 1 + let x = TestApp::spawn(&a, Some(&db), Some(&influx)).await; + + // make a user and give them credits + let user_0_wallet = a.wallet(0); + let user_1_wallet = a.wallet(1); + let admin_wallet = a.wallet(2); + info!(?admin_wallet); + + // Login both users + let user_0_login = create_user(&x, &r, &user_0_wallet, None).await; + let user_1_login = create_user(&x, &r, &user_1_wallet, None).await; + let admin_login = create_user_as_admin(&x, &db, &r, &admin_wallet).await; + + // Load up balances + admin_increase_balance(&x, &r, &admin_login, &user_0_wallet, Decimal::from(1000)).await; + admin_increase_balance(&x, &r, &admin_login, &user_1_wallet, Decimal::from(2000)).await; + + let user_0_balance = user_get_balance(&x, &r, &user_0_login).await; + let user_1_balance = user_get_balance(&x, &r, &user_1_login).await; + + let user_0_balance_pre = user_0_balance.remaining(); + let user_1_balance_pre = user_1_balance.remaining(); + + assert_eq!(user_0_balance_pre, Decimal::from(1000)); + assert_eq!(user_1_balance_pre, Decimal::from(2000)); + + // Generate the proxies + let number_requests = 1000; + let mut handles = Vec::new(); + + // Get the RPC key from the user + let user_0_secret_key = user_get_first_rpc_key(&x, &r, &user_0_login) + .await + .secret_key; + + let proxy_0_user_0_provider = + create_provider_for_user(x.proxy_provider.url(), &user_0_secret_key).await; + let proxy_1_user_0_provider = + create_provider_for_user(x.proxy_provider.url(), &user_0_secret_key).await; + + let proxy_0_user_0_provider = Arc::new(proxy_0_user_0_provider); + let proxy_1_user_0_provider = Arc::new(proxy_1_user_0_provider); + + warn!("Created users, generated providers"); + + info!("Proxy 1: {:?}", proxy_0_user_0_provider); + info!("Proxy 2: {:?}", proxy_1_user_0_provider); + + for _ in 0..number_requests { + // send 2 to proxy 0 user 0 + let proxy_0_user_0_provider_clone = proxy_0_user_0_provider.clone(); + handles.push(tokio::spawn(async move { + proxy_0_user_0_provider_clone + .request::<_, Option>("eth_getBlockByNumber", ("latest", false)) + .await + .unwrap() + .unwrap() + })); + } + + try_join_all(handles).await.unwrap(); + + // Flush all stats here + // TODO: the test should maybe pause time so that stats definitely flush from our queries. + let flush_0_count_0 = x.flush_stats().await.unwrap(); + + // Wait a bit + // TODO: instead of waiting a bit, make flush_stats wait until all stats are handled before returning + sleep(Duration::from_secs(5)).await; + warn!("Counts 0 are: {:?}", flush_0_count_0); + assert_eq!(flush_0_count_0.relational, 1); + assert_eq!(flush_0_count_0.timeseries, 2); + + // no more stats should arrive + let flush_0_count_1 = x.flush_stats().await.unwrap(); + warn!("Counts 0 are: {:?}", flush_0_count_1); + assert_eq!(flush_0_count_1.relational, 0); + assert_eq!(flush_0_count_1.timeseries, 0); + + // get stats now + // todo!("Need to validate all the stat accounting now"); + // Get the stats from here + let mysql_stats = user_get_mysql_stats(&x, &r, &user_0_login).await; + info!("mysql stats are: {:?}", mysql_stats); + + let influx_aggregate_stats = + user_get_influx_stats_aggregated(&x, &r, &user_0_login, chain_id).await; + info!( + "influx_aggregate_stats stats are: {:?}", + influx_aggregate_stats + ); + + // Get the balance + let user_0_balance_post = user_get_balance(&x, &r, &user_0_login).await; + + let influx_stats = influx_aggregate_stats["result"].get(0).unwrap(); + let mysql_stats = mysql_stats["stats"].get(0).unwrap(); + + info!("Influx and mysql stats are"); + info!(?influx_stats); + info!(?mysql_stats); + + assert_eq!(user_0_balance_post.total_frontend_requests, number_requests); + assert_eq!( + mysql_stats["error_response"], + influx_stats["error_response"] + ); + assert_eq!( + mysql_stats["archive_needed"], + influx_stats["archive_needed"] + ); + assert_eq!( + Decimal::from_str(&mysql_stats["chain_id"].to_string().replace('"', "")).unwrap(), + Decimal::from_str(&influx_stats["chain_id"].to_string().replace('"', "")).unwrap() + ); + assert_eq!( + Decimal::from_str(&mysql_stats["no_servers"].to_string()).unwrap(), + Decimal::from_str(&influx_stats["no_servers"].to_string()).unwrap() + ); + assert_eq!( + Decimal::from_str(&mysql_stats["cache_hits"].to_string()).unwrap(), + Decimal::from_str(&influx_stats["total_cache_hits"].to_string()).unwrap() + ); + assert_eq!( + Decimal::from_str(&mysql_stats["cache_misses"].to_string()).unwrap(), + Decimal::from_str(&influx_stats["total_cache_misses"].to_string()).unwrap() + ); + assert_eq!( + Decimal::from_str(&mysql_stats["frontend_requests"].to_string()).unwrap(), + Decimal::from_str(&influx_stats["total_frontend_requests"].to_string()).unwrap() + ); + assert_eq!( + Decimal::from_str(&mysql_stats["sum_credits_used"].to_string().replace('"', "")).unwrap(), + Decimal::from_str( + &influx_stats["total_credits_used"] + .to_string() + .replace('"', "") + ) + .unwrap() + ); + assert_eq!( + Decimal::from_str( + &mysql_stats["sum_incl_free_credits_used"] + .to_string() + .replace('"', "") + ) + .unwrap(), + Decimal::from_str( + &influx_stats["total_incl_free_credits_used"] + .to_string() + .replace('"', "") + ) + .unwrap() + ); + assert_eq!( + Decimal::from_str(&mysql_stats["sum_request_bytes"].to_string()).unwrap(), + Decimal::from_str(&influx_stats["total_request_bytes"].to_string()).unwrap() + ); + assert_eq!( + Decimal::from_str(&mysql_stats["sum_response_bytes"].to_string()).unwrap(), + Decimal::from_str( + &influx_stats["total_response_bytes"] + .to_string() + .replace('"', "") + ) + .unwrap() + ); + assert_eq!( + Decimal::from_str(&mysql_stats["sum_response_millis"].to_string()).unwrap(), + Decimal::from_str( + &influx_stats["total_response_millis"] + .to_string() + .replace('"', "") + ) + .unwrap() + ); + assert_eq!( + Decimal::from_str(&mysql_stats["backend_requests"].to_string()).unwrap(), + Decimal::from_str(&influx_stats["total_backend_requests"].to_string()).unwrap() + ); + + // We don't have gauges so we cant really fix this in influx. will get back to this later + // assert_eq!( + // Decimal::from(user_0_balance_post.remaining()), + // Decimal::from_str(&influx_stats["balance"].to_string()).unwrap() + // ); + + // The fields we skip for mysql + // backend_retries, id, no_servers, period_datetime, rpc_key_id + + // The fields we skip for influx + // collection, rpc_key, time, + + // let user_get_influx_stats_detailed = + // user_get_influx_stats_detailed(&x_0, &r, &user_0_login).await; + // info!( + // "user_get_influx_stats_detailed stats are: {:?}", + // user_get_influx_stats_detailed + // ); +} + +// Gotta compare stats with influx: +// "stats": [ +// { +// "archive_needed": false, +// "backend_requests": 2, +// "backend_retries": 0, +// "cache_hits": 148, +// "cache_misses": 2, +// "chain_id": 999001999, +// "error_response": false, +// "frontend_requests": 150, +// "id": 1, +// "no_servers": 0, +// "period_datetime": "2023-07-13T00:00:00Z", +// "rpc_key_id": 1, +// "sum_credits_used": "180.8000000000", +// "sum_incl_free_credits_used": "180.8000000000", +// "sum_request_bytes": 12433, +// "sum_response_bytes": 230533, +// "sum_response_millis": 194 +// } +// ] + +// influx +// "result": [ +// { +// "archive_needed": false, +// "balance": 939.6, +// "chain_id": "999001999", +// "collection": "opt-in", +// "error_response": false, +// "no_servers": 0, +// "rpc_key": "01H5E9HRZW2S73F1996KPKMYCE", +// "time": "2023-07-16 02:47:56 +00:00", +// "total_backend_requests": 1, +// "total_cache_hits": 49, +// "total_cache_misses": 1, +// "total_credits_used": 60.4, +// "total_frontend_requests": 50, +// "total_incl_free_credits_used": 60.4, +// "total_request_bytes": 4141, +// "total_response_bytes": 76841, +// "total_response_millis": 72 +// }