Skip to content

Commit a7f4240

Browse files
authored
Merge pull request #6560 from fdefelici/feat/aac-compute-marf
feat: aac add marf computation in test harness
2 parents fbcba1b + 111595a commit a7f4240

8 files changed

+408
-180
lines changed

stacks-signer/src/client/mod.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -224,7 +224,7 @@ pub(crate) mod tests {
224224

225225
pub fn generate_random_consensus_hash() -> ConsensusHash {
226226
let rng = rand::thread_rng();
227-
let bytes: Vec<u8> = rng.sample_iter(Standard).take(20).collect();
227+
let bytes: Vec<u8> = rng.sample_iter::<u8, _>(Standard).take(20).collect();
228228
let mut hash = [0u8; 20];
229229
hash.copy_from_slice(&bytes);
230230
ConsensusHash(hash)

stackslib/src/chainstate/nakamoto/test_signers.rs

Lines changed: 12 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -44,8 +44,10 @@ pub struct TestSigners {
4444

4545
impl Default for TestSigners {
4646
fn default() -> Self {
47-
let aggregate_public_key: Vec<u8> =
48-
rand::thread_rng().sample_iter(Standard).take(33).collect();
47+
let aggregate_public_key: Vec<u8> = rand::thread_rng()
48+
.sample_iter::<u8, _>(Standard)
49+
.take(33)
50+
.collect();
4951
let num_signers = 5;
5052
let threshold = 5 * 7 / 10;
5153

@@ -70,8 +72,10 @@ impl TestSigners {
7072

7173
/// Internal function to generate aggregate key information
7274
fn default_with_signers(signer_keys: Vec<Secp256k1PrivateKey>) -> Self {
73-
let aggregate_public_key: Vec<u8> =
74-
rand::thread_rng().sample_iter(Standard).take(33).collect();
75+
let aggregate_public_key: Vec<u8> = rand::thread_rng()
76+
.sample_iter::<u8, _>(Standard)
77+
.take(33)
78+
.collect();
7579
let num_signers = signer_keys.len();
7680
let threshold = u32::try_from(num_signers * 7 / 10).unwrap();
7781
Self {
@@ -227,8 +231,10 @@ impl TestSigners {
227231
return self.aggregate_public_key.clone();
228232
}
229233

230-
let aggregate_public_key: Vec<u8> =
231-
rand::thread_rng().sample_iter(Standard).take(33).collect();
234+
let aggregate_public_key: Vec<u8> = rand::thread_rng()
235+
.sample_iter::<u8, _>(Standard)
236+
.take(33)
237+
.collect();
232238
self.aggregate_public_key.clone_from(&aggregate_public_key);
233239
aggregate_public_key
234240
}

stackslib/src/chainstate/nakamoto/tests/mod.rs

Lines changed: 24 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -2270,7 +2270,10 @@ fn parse_vote_for_aggregate_public_key_valid() {
22702270
let signer_index = thread_rng().next_u64();
22712271
let signer_index_arg = Value::UInt(signer_index as u128);
22722272

2273-
let aggregate_key: Vec<u8> = rand::thread_rng().sample_iter(Standard).take(33).collect();
2273+
let aggregate_key: Vec<u8> = rand::thread_rng()
2274+
.sample_iter::<u8, _>(Standard)
2275+
.take(33)
2276+
.collect();
22742277
let aggregate_key_arg = Value::buff_from(aggregate_key.clone()).expect("Failed to create buff");
22752278
let round = thread_rng().next_u64();
22762279
let round_arg = Value::UInt(round as u128);
@@ -2316,7 +2319,10 @@ fn parse_vote_for_aggregate_public_key_invalid() {
23162319

23172320
let signer_index = thread_rng().next_u32();
23182321
let signer_index_arg = Value::UInt(signer_index as u128);
2319-
let aggregate_key: Vec<u8> = rand::thread_rng().sample_iter(Standard).take(33).collect();
2322+
let aggregate_key: Vec<u8> = rand::thread_rng()
2323+
.sample_iter::<u8, _>(Standard)
2324+
.take(33)
2325+
.collect();
23202326
let aggregate_key_arg = Value::buff_from(aggregate_key).expect("Failed to create buff");
23212327
let round = thread_rng().next_u64();
23222328
let round_arg = Value::UInt(round as u128);
@@ -2498,7 +2504,10 @@ fn valid_vote_transaction() {
24982504
let signer_index = thread_rng().next_u32();
24992505
let signer_index_arg = Value::UInt(signer_index as u128);
25002506

2501-
let aggregate_key: Vec<u8> = rand::thread_rng().sample_iter(Standard).take(33).collect();
2507+
let aggregate_key: Vec<u8> = rand::thread_rng()
2508+
.sample_iter::<u8, _>(Standard)
2509+
.take(33)
2510+
.collect();
25022511
let aggregate_key_arg = Value::buff_from(aggregate_key).expect("Failed to create buff");
25032512
let round = thread_rng().next_u64();
25042513
let round_arg = Value::UInt(round as u128);
@@ -2548,7 +2557,10 @@ fn valid_vote_transaction_malformed_transactions() {
25482557
let signer_index = thread_rng().next_u32();
25492558
let signer_index_arg = Value::UInt(signer_index as u128);
25502559

2551-
let aggregate_key: Vec<u8> = rand::thread_rng().sample_iter(Standard).take(33).collect();
2560+
let aggregate_key: Vec<u8> = rand::thread_rng()
2561+
.sample_iter::<u8, _>(Standard)
2562+
.take(33)
2563+
.collect();
25522564
let aggregate_key_arg = Value::buff_from(aggregate_key).expect("Failed to create buff");
25532565
let round = thread_rng().next_u64();
25542566
let round_arg = Value::UInt(round as u128);
@@ -2782,7 +2794,10 @@ fn filter_one_transaction_per_signer_multiple_addresses() {
27822794
let signer_index = thread_rng().next_u32();
27832795
let signer_index_arg = Value::UInt(signer_index as u128);
27842796

2785-
let aggregate_key: Vec<u8> = rand::thread_rng().sample_iter(Standard).take(33).collect();
2797+
let aggregate_key: Vec<u8> = rand::thread_rng()
2798+
.sample_iter::<u8, _>(Standard)
2799+
.take(33)
2800+
.collect();
27862801
let aggregate_key_arg = Value::buff_from(aggregate_key).expect("Failed to create buff");
27872802
let round = thread_rng().next_u64();
27882803
let round_arg = Value::UInt(round as u128);
@@ -2910,7 +2925,10 @@ fn filter_one_transaction_per_signer_duplicate_nonces() {
29102925
let signer_index = thread_rng().next_u32();
29112926
let signer_index_arg = Value::UInt(signer_index as u128);
29122927

2913-
let aggregate_key: Vec<u8> = rand::thread_rng().sample_iter(Standard).take(33).collect();
2928+
let aggregate_key: Vec<u8> = rand::thread_rng()
2929+
.sample_iter::<u8, _>(Standard)
2930+
.take(33)
2931+
.collect();
29142932
let aggregate_key_arg = Value::buff_from(aggregate_key).expect("Failed to create buff");
29152933
let round = thread_rng().next_u64();
29162934
let round_arg = Value::UInt(round as u128);

0 commit comments

Comments
 (0)