Skip to content

Commit 1daf8aa

Browse files
committed
test: reuse stackers from pox-4 as pox-5 stx-only stakers in epoch_4_0 e2e tests
1 parent 614c797 commit 1daf8aa

3 files changed

Lines changed: 118 additions & 139 deletions

File tree

stacks-node/src/tests/signer/v0/epoch_4_0_multi_miner_distribution.rs

Lines changed: 1 addition & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ use stacks::core::{StacksEpochId, STACKS_EPOCH_MAX};
5454
use stacks::types::chainstate::StacksPrivateKey;
5555
use stacks::util::secp256k1::Secp256k1PublicKey;
5656

57-
use super::{pox5_staker_initial_balances, pox5_staker_keys, MultipleMinerTest};
57+
use super::MultipleMinerTest;
5858
use crate::tests::nakamoto_integrations::enable_epoch_4_0;
5959
use crate::tests::neon_integrations::test_observer;
6060
use crate::tests::to_addr;
@@ -89,12 +89,8 @@ fn epoch_4_0_burn_distribution_chains_across_boundary() {
8989
}
9090

9191
let num_signers = 5;
92-
let seed = "epoch_4_0_multi_miner";
9392
let stake_amount: u128 = 100_000_000_000;
9493
let lock_cycles: u128 = 12;
95-
let stakers = pox5_staker_keys(num_signers, seed);
96-
let per_staker_balance: u64 = u64::try_from(stake_amount).unwrap() + 1_000_000;
97-
let staker_balances = pox5_staker_initial_balances(num_signers, seed, per_staker_balance);
9894

9995
let agg_pubkey: [u8; 33] = Secp256k1PublicKey::from_private(&StacksPrivateKey::from_seed(
10096
b"epoch-4-0-multi-miner-agg",
@@ -119,7 +115,6 @@ fn epoch_4_0_burn_distribution_chains_across_boundary() {
119115
let publisher_addr_str = publisher_addr.to_string();
120116
let token_contract_id_modifier = token_contract_id.clone();
121117
let registry_contract_id_modifier = registry_contract_id.clone();
122-
let staker_balances_modifier = staker_balances.clone();
123118

124119
let mut miners = MultipleMinerTest::new_with_config_modifications(
125120
num_signers,
@@ -132,9 +127,6 @@ fn epoch_4_0_burn_distribution_chains_across_boundary() {
132127
node_config.node.pox_5_sbtc_registry_contract =
133128
Some(registry_contract_id_modifier.clone());
134129
node_config.add_initial_balance(publisher_addr_str.clone(), 1_000_000);
135-
for (addr, balance) in staker_balances_modifier.iter() {
136-
node_config.add_initial_balance(addr.to_string(), *balance);
137-
}
138130
enable_epoch_4_0(node_config);
139131
},
140132
|node_config| {
@@ -161,7 +153,6 @@ fn epoch_4_0_burn_distribution_chains_across_boundary() {
161153
token_contract_name,
162154
registry_contract_name,
163155
&agg_pubkey,
164-
&stakers,
165156
stake_amount,
166157
lock_cycles,
167158
);

stacks-node/src/tests/signer/v0/epoch_4_0_waterfall.rs

Lines changed: 2 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ use stacks::util_lib::boot::boot_code_id;
3939
use stacks_common::deps_common::bitcoin::blockdata::transaction::Transaction as BitcoinTransaction;
4040
use stacks_signer::v0::SpawnedSigner;
4141

42-
use super::{pox5_staker_initial_balances, pox5_staker_keys, SignerTest};
42+
use super::SignerTest;
4343
use crate::tests::nakamoto_integrations::{enable_epoch_4_0, wait_for};
4444
use crate::tests::neon_integrations::{get_chain_info, next_block_and_wait};
4545
use crate::tests::to_addr;
@@ -94,11 +94,8 @@ fn epoch_4_0_block_commit_uses_single_sbtc_output() {
9494
}
9595

9696
let num_signers = 5;
97-
let seed = "epoch_4_0_basic";
9897
let stake_amount: u128 = 100_000_000_000;
9998
let lock_cycles: u128 = 12;
100-
let stakers = pox5_staker_keys(num_signers, seed);
101-
let per_staker_balance: u64 = u64::try_from(stake_amount).unwrap() + 1_000_000;
10299

103100
let agg_pubkey: [u8; 33] =
104101
Secp256k1PublicKey::from_private(&StacksPrivateKey::from_seed(b"epoch-4-0-waterfall-agg"))
@@ -119,12 +116,7 @@ fn epoch_4_0_block_commit_uses_single_sbtc_output() {
119116
ContractName::try_from(registry_contract_name.to_string()).expect("valid contract name"),
120117
);
121118

122-
let mut initial_balances = vec![(spender_addr, 1_000_000)];
123-
initial_balances.extend(pox5_staker_initial_balances(
124-
num_signers,
125-
seed,
126-
per_staker_balance,
127-
));
119+
let initial_balances = vec![(spender_addr, 1_000_000)];
128120

129121
let signer_test: SignerTest<SpawnedSigner> =
130122
SignerTest::new_with_config_modifications_and_snapshot(
@@ -157,7 +149,6 @@ fn epoch_4_0_block_commit_uses_single_sbtc_output() {
157149
token_contract_name,
158150
registry_contract_name,
159151
&agg_pubkey,
160-
&stakers,
161152
stake_amount,
162153
lock_cycles,
163154
);

0 commit comments

Comments
 (0)