Skip to content

Commit 9360397

Browse files
committed
Do not use a test_observer in boot_to_epoch_3 to enable use with multi node tests
Signed-off-by: Jacinta Ferrant <[email protected]>
1 parent 8ddddae commit 9360397

File tree

3 files changed

+12
-8
lines changed

3 files changed

+12
-8
lines changed

testnet/stacks-node/src/tests/signer/mod.rs

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,6 @@ use wsts::state_machine::PublicKeys;
6363

6464
use super::nakamoto_integrations::wait_for;
6565
use crate::config::{Config as NeonConfig, EventKeyType, EventObserverConfig, InitialBalance};
66-
use crate::event_dispatcher::MinedNakamotoBlockEvent;
6766
use crate::neon::{Counters, TestFlag};
6867
use crate::run_loop::boot_nakamoto;
6968
use crate::tests::bitcoin_regtest::BitcoinCoreController;
@@ -364,9 +363,10 @@ impl<S: Signer<T> + Send + 'static, T: SignerEventTrait + 'static> SignerTest<Sp
364363
reward_cycle_height.saturating_sub(current_block_height)
365364
}
366365

367-
fn mine_nakamoto_block(&mut self, timeout: Duration) -> MinedNakamotoBlockEvent {
366+
fn mine_nakamoto_block(&mut self, timeout: Duration) {
368367
let commits_submitted = self.running_nodes.commits_submitted.clone();
369368
let mined_block_time = Instant::now();
369+
let info_before = self.stacks_client.get_peer_info().unwrap();
370370
next_block_and_mine_commit(
371371
&mut self.running_nodes.btc_regtest_controller,
372372
timeout.as_secs(),
@@ -376,15 +376,15 @@ impl<S: Signer<T> + Send + 'static, T: SignerEventTrait + 'static> SignerTest<Sp
376376
.unwrap();
377377

378378
wait_for(timeout.as_secs(), || {
379-
Ok(!test_observer::get_mined_nakamoto_blocks().is_empty())
379+
let info_after = self.stacks_client.get_peer_info().unwrap();
380+
Ok(info_after.stacks_tip_height > info_before.stacks_tip_height)
380381
})
381382
.unwrap();
382383
let mined_block_elapsed_time = mined_block_time.elapsed();
383384
info!(
384385
"Nakamoto block mine time elapsed: {:?}",
385386
mined_block_elapsed_time
386387
);
387-
test_observer::get_mined_nakamoto_blocks().pop().unwrap()
388388
}
389389

390390
fn mine_block_wait_on_processing(

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

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2954,7 +2954,8 @@ fn signer_set_rollover() {
29542954
let transfer_tx =
29552955
make_stacks_transfer(&sender_sk, sender_nonce, send_fee, &recipient, send_amt);
29562956
submit_tx(&http_origin, &transfer_tx);
2957-
let mined_block = signer_test.mine_nakamoto_block(short_timeout);
2957+
signer_test.mine_nakamoto_block(short_timeout);
2958+
let mined_block = test_observer::get_mined_nakamoto_blocks().pop().unwrap();
29582959
let block_sighash = mined_block.signer_signature_hash;
29592960
let signer_signatures = mined_block.signer_signature;
29602961

@@ -3068,7 +3069,8 @@ fn signer_set_rollover() {
30683069
let transfer_tx =
30693070
make_stacks_transfer(&sender_sk, sender_nonce, send_fee, &recipient, send_amt);
30703071
submit_tx(&http_origin, &transfer_tx);
3071-
let mined_block = signer_test.mine_nakamoto_block(short_timeout);
3072+
signer_test.mine_nakamoto_block(short_timeout);
3073+
let mined_block = test_observer::get_mined_nakamoto_blocks().pop().unwrap();
30723074

30733075
info!("---- Verifying that the new signers signed the block -----");
30743076
let signer_signatures = mined_block.signer_signature;

testnet/stacks-node/src/tests/signer/v1.rs

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -202,7 +202,8 @@ impl SignerTest<SpawnedSigner> {
202202
agg_key: &Point,
203203
timeout: Duration,
204204
) -> MinedNakamotoBlockEvent {
205-
let new_block = self.mine_nakamoto_block(timeout);
205+
self.mine_nakamoto_block(timeout);
206+
let new_block = test_observer::get_mined_nakamoto_blocks().pop().unwrap();
206207
let signer_sighash = new_block.signer_signature_hash.clone();
207208
let signature = self.wait_for_confirmed_block_v1(&signer_sighash, timeout);
208209
assert!(signature.0.verify(&agg_key, signer_sighash.as_bytes()));
@@ -1130,7 +1131,8 @@ fn sign_after_signer_reboot() {
11301131

11311132
info!("------------------------- Test Mine Block after restart -------------------------");
11321133

1133-
let last_block = signer_test.mine_nakamoto_block(timeout);
1134+
signer_test.mine_nakamoto_block(timeout);
1135+
let last_block = test_observer::get_mined_nakamoto_blocks().pop().unwrap();
11341136
let proposed_signer_signature_hash = signer_test
11351137
.wait_for_validate_ok_response(short_timeout)
11361138
.signer_signature_hash;

0 commit comments

Comments
 (0)