Skip to content

Commit b1e0305

Browse files
committed
refactor: enhance readability of MKTree tests
1 parent 4be7d99 commit b1e0305

File tree

1 file changed

+10
-10
lines changed

1 file changed

+10
-10
lines changed

mithril-common/src/crypto_helper/merkle_tree.rs

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -346,7 +346,7 @@ mod tests {
346346

347347
#[test]
348348
fn test_should_accept_valid_proof_generated_by_merkle_tree() {
349-
let leaves = generate_leaves(100000);
349+
let leaves = generate_leaves(10);
350350
let leaves_to_verify = &[leaves[0].to_owned(), leaves[3].to_owned()];
351351
let proof =
352352
MKProof::from_leaves(leaves_to_verify).expect("MKProof generation should not fail");
@@ -355,17 +355,17 @@ mod tests {
355355

356356
#[test]
357357
fn test_should_reject_invalid_proof_generated_by_merkle_tree() {
358-
let leaves = generate_leaves(100000);
358+
let leaves = generate_leaves(10);
359359
let leaves_to_verify = &[leaves[0].to_owned(), leaves[3].to_owned()];
360360
let mut proof =
361361
MKProof::from_leaves(leaves_to_verify).expect("MKProof generation should not fail");
362-
proof.inner_root = Arc::new(leaves[10].to_owned());
362+
proof.inner_root = Arc::new(leaves[1].to_owned());
363363
proof.verify().expect_err("The MKProof should be invalid");
364364
}
365365

366366
#[test]
367367
fn test_should_list_leaves() {
368-
let leaves = generate_leaves(1000);
368+
let leaves: Vec<MKTreeNode> = vec!["test-0".into(), "test-1".into(), "test-2".into()];
369369
let mktree = MKTree::new(&leaves).expect("MKTree creation should not fail");
370370
let leaves_retrieved = mktree.leaves();
371371

@@ -376,8 +376,8 @@ mod tests {
376376
}
377377

378378
#[test]
379-
fn test_should_clone() {
380-
let leaves = generate_leaves(1000);
379+
fn test_should_clone_and_compute_same_root() {
380+
let leaves = generate_leaves(10);
381381
let mktree = MKTree::new(&leaves).expect("MKTree creation should not fail");
382382
let mktree_clone = mktree.clone();
383383

@@ -389,15 +389,15 @@ mod tests {
389389

390390
#[test]
391391
fn test_should_support_append_leaves() {
392-
let leaves = generate_leaves(1000);
393-
let leaves_creation = &leaves[..900];
394-
let leaves_to_append = &leaves[900..];
392+
let leaves = generate_leaves(10);
393+
let leaves_creation = &leaves[..9];
394+
let leaves_to_append = &leaves[9..];
395395
let mut mktree = MKTree::new(leaves_creation).expect("MKTree creation should not fail");
396396
mktree
397397
.append(leaves_to_append)
398398
.expect("MKTree append leaves should not fail");
399399

400-
assert_eq!(1000, mktree.total_leaves());
400+
assert_eq!(10, mktree.total_leaves());
401401
}
402402

403403
#[test]

0 commit comments

Comments
 (0)