problem 24 part 2

1 files changed, 9 insertions, 0 deletions

diff --git a/tests/lib.rs b/tests/lib.rs
index d38e3e3..3961b84 100644
--- a/tests/lib.rs
+++ b/tests/lib.rs
@@ -512,3 +512,12 @@ fn problem_24 () {
     ).unwrap();
     assert_eq!(got, key);
 }
+
+#[test]
+fn problem_24_part_2 () {
+    let seed = time::now().to_timespec().sec as u32;
+    let mut mt = matasano::MersenneTwister::from_seed(seed);
+    let token: Vec<u8> = mt.gen_iter().take(16).collect();
+    let got = matasano::recover_mt19937_key_from_time(&token[..]).unwrap();
+    assert_eq!(got, seed);
+}