diff --git a/src/commands/start/usdfc_funding/usdfc_funding_step.rs b/src/commands/start/usdfc_funding/usdfc_funding_step.rs index d287b9a8..299e0f1c 100644 --- a/src/commands/start/usdfc_funding/usdfc_funding_step.rs +++ b/src/commands/start/usdfc_funding/usdfc_funding_step.rs @@ -8,6 +8,7 @@ use super::funding_operations::{self, check_mock_usdfc_balance, transfer_mock_us use super::key_operations::get_user_private_key; use crate::commands::start::step::{SetupContext, Step}; use crate::commands::start::usdfc_funding::key_operations::get_user_eth_address; +use crate::constants::USER_ACCOUNT_COUNT; use crate::docker::containers::lotus_container_name; use crate::docker::core::container_is_running; use std::error::Error; @@ -41,7 +42,7 @@ impl USDFCFundingStep { let mut accounts_to_check = Vec::new(); // Add user accounts (base-1 numbering) - for user_num in 1..=3 { + for user_num in 1..=USER_ACCOUNT_COUNT { let account_name = format!("USER_{}", user_num); accounts_to_check.push((account_name, 100_000u64)); } @@ -250,7 +251,7 @@ impl Step for USDFCFundingStep { let mut token_transfers = Vec::new(); // Add user accounts (base-1 numbering) - for user_num in 1..=3 { + for user_num in 1..=USER_ACCOUNT_COUNT { let account_name = format!("USER_{}", user_num); let eth_address = get_user_eth_address(&account_name)?; let amount_tokens = 100_000u64; @@ -291,7 +292,7 @@ impl Step for USDFCFundingStep { // Build list of accounts to verify let mut accounts_to_verify = Vec::new(); - for user_num in 1..=3 { + for user_num in 1..=USER_ACCOUNT_COUNT { let account_name = format!("USER_{}", user_num); accounts_to_verify.push((account_name, 100_000u64)); } diff --git a/src/constants.rs b/src/constants.rs index 64532323..b6f356ab 100644 --- a/src/constants.rs +++ b/src/constants.rs @@ -41,6 +41,9 @@ pub const PORT_CHECK_TIMEOUT_MS: u64 = 5000; /// PDP Service Provider configuration pub const MAX_PDP_SP_COUNT: usize = 5; +/// Number of user test accounts (USER_1, USER_2, USER_3) +pub const USER_ACCOUNT_COUNT: usize = 3; + /// Service configuration pub const SERVICE_NAME: &str = "FOC DevNet Warm Storage"; pub const SERVICE_DESCRIPTION: &str = "Warm storage service for FOC local development network"; diff --git a/src/external_api/export.rs b/src/external_api/export.rs index 720f1182..3aa2d2d9 100644 --- a/src/external_api/export.rs +++ b/src/external_api/export.rs @@ -8,6 +8,7 @@ use std::path::Path; use chrono::Utc; use crate::commands::start::step::SetupContext; +use crate::constants::USER_ACCOUNT_COUNT; use crate::crypto::derive_ethereum_key; use crate::crypto::mnemonic::load_mnemonic; use crate::external_api::{ @@ -55,7 +56,7 @@ fn build_users(ctx: &SetupContext) -> Result, Box