Update loadtest for running with chain params

Signed-off-by: Danil <deniallugo@gmail.com>

Author: Deniallugo

zk_toolbox/crates/zk_supervisor/src/commands/test/loadtest.rs

@@ -0,0 +1,52 @@
+use anyhow::Context;
+use common::{cmd::Cmd, config::global_config, logger};
+use config::EcosystemConfig;
+use xshell::{cmd, Shell};
+
+use crate::messages::MSG_CHAIN_NOT_FOUND_ERR;
+
+pub fn run(shell: &Shell) -> anyhow::Result<()> {
+    let ecosystem_config = EcosystemConfig::from_file(shell)?;
+
+    let chain_config = ecosystem_config
+        .load_chain(global_config().chain_name.clone())
+        .context(MSG_CHAIN_NOT_FOUND_ERR)?;
+
+    let general_api = chain_config
+        .get_general_config()?
+        .api_config
+        .context("API config is not found")?;
+
+    let mut command = cmd!(shell, "cargo run --release --bin loadnext")
+        .env(
+            "L2_CHAIN_ID",
+            chain_config
+                .get_genesis_config()?
+                .l2_chain_id
+                .as_u64()
+                .to_string(),
+        )
+        .env(
+            "MAIN_TOKEN",
+            format!(
+                "{:?}",
+                ecosystem_config
+                    .get_erc20_tokens()
+                    .first()
+                    .context("NO Erc20 tokens were deployed")?
+                    .address
+            ),
+        )
+        .env("L2_RPC_ADDRESS", general_api.web3_json_rpc.http_url)
+        .env("L2_WS_RPC_ADDRESS", general_api.web3_json_rpc.ws_url);
+
+    if global_config().verbose {
+        command = command.env("RUST_LOG", "loadnext=debug")
+    }
+
+    Cmd::new(command).with_force_run().run()?;
+
+    logger::outro("Loadtest success");
+
+    Ok(())
+}

zk_toolbox/crates/zk_supervisor/src/commands/test/mod.rs

@@ -15,6 +15,7 @@ mod args;
 mod build;
 mod integration;
 mod l1_contracts;
+mod loadtest;
 mod prover;
 mod recovery;
 mod revert;
@@ -43,6 +44,8 @@ pub enum TestCommands {
     Prover,
     #[clap(about = MSG_TEST_WALLETS_INFO)]
     Wallet,
+    #[clap(about = MSG_TEST_WALLETS_INFO)]
+    Loadtest,
 }
 
 pub async fn run(shell: &Shell, args: TestCommands) -> anyhow::Result<()> {
@@ -56,5 +59,6 @@ pub async fn run(shell: &Shell, args: TestCommands) -> anyhow::Result<()> {
         TestCommands::L1Contracts => l1_contracts::run(shell),
         TestCommands::Prover => prover::run(shell),
         TestCommands::Wallet => wallet::run(shell),
+        TestCommands::Loadtest => loadtest::run(shell),
     }
 }