From 3d280d18089ac4571d3fb6c8a2e0557ce1d7ae1d Mon Sep 17 00:00:00 2001
From: Serhii Volovyk <SergeyVolovyk@gmail.com>
Date: Tue, 30 Jul 2024 20:25:22 +0300
Subject: [PATCH] Load test fixes (#765)

* update acc for load tests

* fmt

* tune load tests balance

* fmt

* load test log check fixed

* fmt
---
 chain-signatures/contract/src/lib.rs           | 5 ++++-
 load-tests/src/fastauth/mod.rs                 | 3 ++-
 load-tests/src/multichain/mod.rs               | 7 +++----
 load-tests/src/multichain/test_plans/short.txt | 2 +-
 4 files changed, 10 insertions(+), 7 deletions(-)

diff --git a/chain-signatures/contract/src/lib.rs b/chain-signatures/contract/src/lib.rs
index 01315435d..330aac1ec 100644
--- a/chain-signatures/contract/src/lib.rs
+++ b/chain-signatures/contract/src/lib.rs
@@ -728,7 +728,10 @@ impl VersionedMpcContract {
     ) -> Result<SignatureResponse, MpcContractError> {
         match self {
             Self::V0(_) => match signature {
-                SignatureResult::Ok(signature) => Ok(signature),
+                SignatureResult::Ok(signature) => {
+                    log!("Signature is ready.");
+                    Ok(signature)
+                }
                 SignatureResult::Err(_) => Err(MpcContractError::SignError(SignError::Timeout)),
             },
         }
diff --git a/load-tests/src/fastauth/mod.rs b/load-tests/src/fastauth/mod.rs
index cb59c784c..02d0f7a54 100644
--- a/load-tests/src/fastauth/mod.rs
+++ b/load-tests/src/fastauth/mod.rs
@@ -47,7 +47,8 @@ pub async fn prepare_user_credentials(user: &mut GooseUser) -> TransactionResult
 
     let subaccount = root_account
         .create_subaccount(&format!("user-{}", rand::random::<u64>()))
-        .initial_balance(NearToken::from_yoctonear(200000000000000000000000u128))
+        // Balance this values depending on how many users you want to create and available balance
+        .initial_balance(NearToken::from_near(10))
         .transact()
         .await
         .unwrap()
diff --git a/load-tests/src/multichain/mod.rs b/load-tests/src/multichain/mod.rs
index d9e3af9be..4baafcac2 100644
--- a/load-tests/src/multichain/mod.rs
+++ b/load-tests/src/multichain/mod.rs
@@ -28,8 +28,7 @@ pub async fn multichain_sign(user: &mut GooseUser) -> TransactionResult {
         .get_session_data::<UserSession>()
         .expect("Session Data must be set");
 
-    let multichain_contract_id =
-        AccountId::try_from("v5.multichain-mpc-dev.testnet".to_string()).unwrap();
+    let multichain_contract_id = AccountId::try_from("v1.signer-dev.testnet".to_string()).unwrap();
     let testnet_rpc_url = "https://rpc.testnet.near.org".to_string();
 
     let signer = InMemorySigner {
@@ -69,7 +68,7 @@ pub async fn multichain_sign(user: &mut GooseUser) -> TransactionResult {
             }))
             .unwrap(),
             gas: 300_000_000_000_000,
-            deposit: 1,
+            deposit: 1, // attach more if you are ok with going above "CHEAP_REQUESTS" limit
         })],
     };
 
@@ -107,7 +106,7 @@ pub async fn multichain_sign(user: &mut GooseUser) -> TransactionResult {
 
     tracing::info!("goose_response: {:?}", rsp);
 
-    let expected_log = "sign_helper: signature ready";
+    let expected_log = "Signature is ready.";
 
     let validate = &Validate::builder()
         .status(200)
diff --git a/load-tests/src/multichain/test_plans/short.txt b/load-tests/src/multichain/test_plans/short.txt
index 276260e51..79e9ad0e7 100644
--- a/load-tests/src/multichain/test_plans/short.txt
+++ b/load-tests/src/multichain/test_plans/short.txt
@@ -1 +1 @@
-0,0s;5,30s;5,20m;0,0s
\ No newline at end of file
+0,0s;1,30s;1,10m;0,0s
\ No newline at end of file