From 01b0eca079f96ef348adaebc1079560f388cd888 Mon Sep 17 00:00:00 2001
From: Hannes <55623006+umgefahren@users.noreply.github.com>
Date: Wed, 23 Mar 2022 17:01:59 +0100
Subject: [PATCH] Extended tests to catch the now fixed bug

---
 tests/mod.rs | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/tests/mod.rs b/tests/mod.rs
index 1284a1b..2b7752c 100644
--- a/tests/mod.rs
+++ b/tests/mod.rs
@@ -27,9 +27,11 @@ fn ratchet_enc_skip() {
     let data = include_bytes!("../src/header.rs").to_vec();
     let (header1, encrypted1, nonce1) = alice_ratchet.ratchet_encrypt(&data, b"");
     let (header2, encrypted2, nonce2) = alice_ratchet.ratchet_encrypt(&data, b"");
+    let (header3, encrypted3, nonce3) = alice_ratchet.ratchet_encrypt(&data, b"");
+    let decrypted3 = bob_ratchet.ratchet_decrypt(&header3, &encrypted3, &nonce3, b"");
     let decrypted2 = bob_ratchet.ratchet_decrypt(&header2, &encrypted2, &nonce2, b"");
     let decrypted1 = bob_ratchet.ratchet_decrypt(&header1, &encrypted1, &nonce1, b"");
-    let comp_res = decrypted1 == data && decrypted2 == data;
+    let comp_res = decrypted1 == data && decrypted2 == data && decrypted3 == data;
     assert!(comp_res)
 }
 
@@ -101,9 +103,11 @@ fn ratchet_ench_enc_skip() {
     let data = include_bytes!("../src/header.rs").to_vec();
     let (header1, encrypted1, nonce1) = alice_ratchet.ratchet_encrypt(&data, b"");
     let (header2, encrypted2, nonce2) = alice_ratchet.ratchet_encrypt(&data, b"");
+    let (header3, encrypted3, nonce3) = alice_ratchet.ratchet_encrypt(&data, b"");
+    let decrypted3 = bob_ratchet.ratchet_decrypt(&header3, &encrypted3, &nonce3, b"");
     let decrypted2 = bob_ratchet.ratchet_decrypt(&header2, &encrypted2, &nonce2, b"");
     let decrypted1 = bob_ratchet.ratchet_decrypt(&header1, &encrypted1, &nonce1, b"");
-    let comp_res = decrypted1 == data && decrypted2 == data;
+    let comp_res = decrypted1 == data && decrypted2 == data && decrypted3 == data;
     assert!(comp_res)
 }
 
@@ -191,4 +195,4 @@ fn import_export() {
     let ex_alice_ratchet = alice_ratchet.export();
     let in_alice_ratchet = RatchetEncHeader::import(&ex_alice_ratchet);
     assert_eq!(in_alice_ratchet, alice_ratchet);
-}
\ No newline at end of file
+}