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

48 lines
1.7 KiB
Rust
Raw Normal View History

use crate::TestApp;
use ethers::prelude::{LocalWallet, Signer};
2023-07-10 06:13:03 +03:00
use migration::sea_orm::prelude::Decimal;
use tracing::info;
use web3_proxy::frontend::admin::AdminIncreaseBalancePost;
use web3_proxy::frontend::users::authentication::LoginPostResponse;
/// Helper function to increase the balance of a user, from an admin
#[allow(unused)]
pub async fn admin_increase_balance(
x: &TestApp,
r: &reqwest::Client,
admin_login_response: &LoginPostResponse,
target_wallet: &LocalWallet,
amount: Decimal,
) -> serde_json::Value {
let increase_balance_post_url = format!("{}admin/increase_balance", x.proxy_provider.url());
info!("Increasing balance");
// Login the user
// Use the bearer token of admin to increase user balance
let increase_balance_data = AdminIncreaseBalancePost {
user_address: target_wallet.address(), // set user address to increase balance
amount, // set amount to increase
note: Some("Test increasing balance".to_string()),
};
info!(?increase_balance_post_url);
info!(?increase_balance_data);
info!(?admin_login_response.bearer_token);
let increase_balance_response = r
.post(increase_balance_post_url)
.json(&increase_balance_data)
.bearer_auth(admin_login_response.bearer_token)
.send()
.await
.unwrap();
info!(?increase_balance_response, "http response");
More balance tests (#182) * fix popularity contest * more info in the Debug for Web3Rpc * add frontend_requests and cache_misses to the Balance query * add more to balance and stats flushing and improved test coverage * it compiles * deserializer for Ulid to Uuid I think a wrapper type on Ulid that implements sea_orm::Value is probably better * rename variable to match struct name * add deserializer for Address -> Vec<u8> * sql sum returns a Decimal. need to convert to u64 * assert more * one log and assert more * log more * use a helper to get the user's rpc provider * this should be 2 now that we have a public and authed call * this should be zero. the public has the cache miss * instrument cu calcs * trace the value we took, not the default that replaced it * move usd_per_chain into config * remove some extra logging * use Arc::into_inner to maybe avoid a race * off by 1 * pass paid credits used instead of returning it this lets us use it to write to our user balance cache first. importantly, this keeps us from holding a write lock while writing to mysql * no cache misses expected in this test * actually check the admin * put the balance checks back now that the rest of the test works * archive request is being set incorrectly * wow howd we manage flipping the greater than sign on archive depth * move latest_balance and premium_credits_used to before any stats are emitted * lint * and build undoes the linting. fun i didnt even want to lint them in the first place, so this is fine * missed incrementing total_spent when not incrementing total_spent_paid_credits * use the credits on self * use the credits on self (pt 2) * fix type for 10 cu query * convert the requestmetadata on the other side of the channel * logs * viewing stats is allowed even without a balance * move paid_credits_used to AuthorizationChecks * wip * test_sum_credits_used finally passes * UserBalanceCache::get_or_insert * re-enable rpc_secret_key_cache * move invalidate to a helper function and always call it **after** the db is commited * fix PartialEq and Eq on RpcSecretKey * cargo upgrade
2023-07-12 10:35:07 +03:00
// TODO: use a struct here
let increase_balance_response = increase_balance_response
.json::<serde_json::Value>()
.await
.unwrap();
info!(?increase_balance_response, "json response");
increase_balance_response
}