Fix tests after adding strict kex

mkj · Dec 28, 2023 · a3bdfcc · a3bdfcc
1 parent 92975ad
commit a3bdfcc
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/encrypt.rs b/src/encrypt.rs
@@ -749,6 +749,7 @@ mod tests {
                     discard_next: false,
                     is_client: false,
                     send_ext_info: true,
+                    strict_kex: false,
                 })
             })
             // and plaintext

diff --git a/src/kex.rs b/src/kex.rs
@@ -991,8 +991,8 @@ mod tests {
         let serv_init = if let Packet::KexInit(k) = serv_init { k } else { panic!() };
         assert!(ts.next().is_none());
 
-        serv.handle_kexinit(cli_init, false, &serv_conf, &version, &mut ts.sender()).unwrap();
-        cli.handle_kexinit(serv_init, true, &cli_conf, &version, &mut tc.sender()).unwrap();
+        serv.handle_kexinit(cli_init, false, &serv_conf, &version, true, &mut ts.sender()).unwrap();
+        cli.handle_kexinit(serv_init, true, &cli_conf, &version, true, &mut tc.sender()).unwrap();
 
         let cli_dhinit = tc.next().unwrap();
         let cli_dhinit = if let Packet::KexDHInit(k) = cli_dhinit { k } else { panic!() };