Эх сурвалжийг харах

sdk/crypto/note: Fix test unit.

parazyd 3 жил өмнө
parent
commit
e2a2bd29ad

+ 5 - 2
src/sdk/src/crypto/note.rs

@@ -80,16 +80,19 @@ impl AeadEncryptedNote {
 #[cfg(test)]
 #[cfg(test)]
 mod tests {
 mod tests {
     use super::*;
     use super::*;
+    use crate::crypto::Keypair;
+
+    use rand::rngs::OsRng;
 
 
     #[test]
     #[test]
     fn test_aead_note() {
     fn test_aead_note() {
         let plaintext = "gm world";
         let plaintext = "gm world";
-        let keypair = Keypair::new(&mut OsRng);
+        let keypair = Keypair::random(&mut OsRng);
 
 
         let encrypted_note =
         let encrypted_note =
             AeadEncryptedNote::encrypt(&plaintext, &keypair.public, &mut OsRng).unwrap();
             AeadEncryptedNote::encrypt(&plaintext, &keypair.public, &mut OsRng).unwrap();
 
 
-        let plaintext2 = encrypted_note.decrypt(&keypair.secret).unwrap();
+        let plaintext2: String = encrypted_note.decrypt(&keypair.secret).unwrap();
 
 
         assert_eq!(plaintext, plaintext2);
         assert_eq!(plaintext, plaintext2);
     }
     }