Browse Source

Add tokio test for the basic chat application

Eric Zhang 4 năm trước cách đây
mục cha
commit
5ca9a0ff6c
1 tập tin đã thay đổi với 19 bổ sung0 xóa
  1. 19 0
      rustpad-server/src/server.rs

+ 19 - 0
rustpad-server/src/server.rs

@@ -123,3 +123,22 @@ impl Rustpad {
         self.notify.notify_waiters();
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[tokio::test]
+    async fn test_single_message() {
+        let filter = routes();
+        let mut client = warp::test::ws()
+            .path("/socket")
+            .handshake(filter)
+            .await
+            .expect("handshake");
+        client.send_text("hello world").await;
+        let msg = client.recv().await.expect("recv");
+        let msg = msg.to_str().expect("string");
+        assert_eq!(msg, "[[0,\"hello world\"]]");
+    }
+}