소스 검색

Add tokio test for the basic chat application

Eric Zhang 4 년 전
부모
커밋
5ca9a0ff6c
1개의 변경된 파일19개의 추가작업 그리고 0개의 파일을 삭제
  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\"]]");
+    }
+}