update test to expect status and kernel info reply

kafonek · Nov 23, 2023 · 20f73b9 · 20f73b9
1 parent 094f91f
commit 20f73b9
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_messages.rs b/tests/test_messages.rs
@@ -74,5 +74,8 @@ async fn test_kernel_info(_ipykernel_process: Option<Child>) {
     let action = client.kernel_info_request(handlers).await;
     action.await;
     let counts = handler.counts.lock().await;
-    assert_eq!(counts.get("kernel_info"), Some(&1));
+    let mut expected = HashMap::new();
+    expected.insert("kernel_info".to_string(), 1);
+    expected.insert("status".to_string(), 2);
+    assert_eq!(*counts, expected);
 }