clean up random key test

burdges · Jun 23, 2019 · 9f174bd · 9f174bd
1 parent c5c0e72
commit 9f174bd
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -234,9 +234,9 @@ mod tests {
     #[test]
     fn simple_encrypt_decrypt_test() {
         const TEST_PLAINTEXT: &'static [u8] = b"Hello there world, I'm just a test string";
-        let key = thread_rng().gen_iter::<u8>().take(RAW_KEY_SIZE).collect::<Vec<u8>>();
-        let l = Lioness::<VarBlake2b,ChaCha>::new_raw(array_ref!(key,0,RAW_KEY_SIZE));
-        //let l = LionessDefault::new_raw(array_ref!(key,0,RAW_KEY_SIZE));
+        let mut key = [0u8; RAW_KEY_SIZE];
+        thread_rng().fill_bytes(&mut key);
+        let l = Lioness::<VarBlake2b,ChaCha>::new_raw(&key);
         let mut v: Vec<u8> = TEST_PLAINTEXT.to_owned();
         assert_eq!(v,TEST_PLAINTEXT);
         l.encrypt(&mut v).unwrap();