Ver código fonte

Add tokio test for the basic chat application

Eric Zhang 4 anos atrás
pai
commit
5ca9a0ff6c
1 arquivos alterados com 19 adições e 0 exclusões
  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\"]]");
+    }
+}