Fix unit tests.

tkhq · Sep 17, 2024 · c264d9c · c264d9c
1 parent 605447b
commit c264d9c
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/src/qos_crypto/src/shamir.rs b/src/qos_crypto/src/shamir.rs
@@ -186,36 +186,36 @@ mod test {
 		// Reconstruct with all the shares
 		let shares = all_shares.clone();
 		let reconstructed = shares_reconstruct(&shares).unwrap();
-		let old_reconstructed = deprecated_insecure_shares_reconstruct(&shares);
+		let old_reconstructed = shares_reconstruct(&shares).unwrap();
 		assert_eq!(secret.to_vec(), reconstructed);
 		assert_eq!(secret.to_vec(), old_reconstructed);
 
 		// Reconstruct with enough shares
 		let shares = &all_shares[..k];
 		let reconstructed = shares_reconstruct(shares).unwrap();
-		let old_reconstructed = deprecated_insecure_shares_reconstruct(shares);
+		let old_reconstructed = shares_reconstruct(shares).unwrap();
 		assert_eq!(secret.to_vec(), reconstructed);
 		assert_eq!(secret.to_vec(), old_reconstructed);
 
 		// Reconstruct with not enough shares
 		let shares = &all_shares[..(k - 1)];
 		let reconstructed = shares_reconstruct(shares).unwrap();
-		let old_reconstructed = deprecated_insecure_shares_reconstruct(shares);
+		let old_reconstructed = shares_reconstruct(shares).unwrap();
 		assert!(secret.to_vec() != reconstructed);
 		assert!(secret.to_vec() != old_reconstructed);
 
 		// Reconstruct with enough shuffled shares
 		let mut shares = all_shares.clone()[..k].to_vec();
 		shares.shuffle(&mut rand::thread_rng());
 		let reconstructed = shares_reconstruct(&shares).unwrap();
-		let old_reconstructed = deprecated_insecure_shares_reconstruct(&shares);
+		let old_reconstructed = shares_reconstruct(&shares).unwrap();
 		assert_eq!(secret.to_vec(), reconstructed);
 		assert_eq!(secret.to_vec(), old_reconstructed);
 
 		for combo in crate::n_choose_k::combinations(&all_shares, k) {
 			let reconstructed = shares_reconstruct(&combo).unwrap();
 			let old_reconstructed =
-				deprecated_insecure_shares_reconstruct(&shares);
+				shares_reconstruct(&shares).unwrap();
 			assert_eq!(secret.to_vec(), reconstructed);
 			assert_eq!(secret.to_vec(), old_reconstructed);
 		}