Add concurrent_send test

Author: nanoqsh

tests/communication.rs

@@ -116,3 +116,53 @@ async fn split_communication() {
     let messages = msg_rx.await.expect("Failed to receive messages");
     assert_eq!(messages.len(), 10);
 }
+
+#[async_std::test]
+async fn concurrent_send() {
+    let _ = env_logger::try_init();
+
+    let (con_tx, con_rx) = futures::channel::oneshot::channel();
+    let (msg_tx, msg_rx) = futures::channel::oneshot::channel();
+
+    let f = async move {
+        let listener = TcpListener::bind("127.0.0.1:12347").await.unwrap();
+        info!("Server ready");
+        con_tx.send(()).unwrap();
+        info!("Waiting on next connection");
+        let (connection, _) = listener.accept().await.expect("No connections to accept");
+        let stream = accept_async(connection).await;
+        let stream = stream.expect("Failed to handshake with connection");
+        run_connection(stream, msg_tx).await;
+    };
+
+    task::spawn(f);
+
+    info!("Waiting for server to be ready");
+
+    con_rx.await.expect("Server not ready");
+    let tcp = TcpStream::connect("127.0.0.1:12347")
+        .await
+        .expect("Failed to connect");
+    let url = url::Url::parse("ws://localhost:12347/").unwrap();
+    let (stream, _) = client_async(url, tcp)
+        .await
+        .expect("Client failed to connect");
+
+    let (tx, _rx) = stream.split();
+
+    // the `WebSocketSender::send` takes a shared `&self`, so you can call it concurrently.
+    // this test case checks that it works
+    let results = futures::future::join_all((1..10).map(|i| {
+        info!("Sending message");
+        tx.send(Message::text(format!("{}", i)))
+    }))
+    .await;
+
+    assert!(results.iter().all(Result::is_ok));
+
+    tx.close(None).await.expect("Failed to close");
+
+    info!("Waiting for response messages");
+    let messages = msg_rx.await.expect("Failed to receive messages");
+    assert_eq!(messages.len(), 10);
+}