axol 0.2.0

Axol Web Framework
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

#![cfg(feature = "ws")]

use axol::{Message, Router, WebSocketUpgrade};
use axol_http::response::Response;

mod common;
use common::*;
use futures::{SinkExt, StreamExt};
use tokio_tungstenite::tungstenite::Message as TTMessage;

async fn simple_ws(ws: WebSocketUpgrade) -> Response {
    ws.on_upgrade(|mut c| async move {
        match c.recv().await.unwrap().unwrap() {
            Message::Text(text) => {
                c.send(Message::Text(text)).await.unwrap();
            }
            _ => panic!("invalid message at server"),
        }
    })
}

#[tokio::test]
async fn websocket_tests() {
    let server = spawn_router(Router::new().get("/ws", simple_ws)).await;

    let (mut stream, _) = tokio_tungstenite::connect_async(server.ws_url("/ws"))
        .await
        .unwrap();

    stream
        .send(TTMessage::Text("test message".into()))
        .await
        .unwrap();
    match stream.next().await.unwrap().unwrap() {
        TTMessage::Text(text) => {
            assert_eq!(text, "test message");
        }
        _ => panic!("invalid message at client"),
    }
}