[go: up one dir, main page]

ws 0.7.3

Lightweight, event-driven WebSockets for Rust.
Documentation
#![cfg(feature="permessage-deflate")]
extern crate ws;
extern crate url;
extern crate env_logger;

use ws::{WebSocket, Sender, Message, Builder, Settings};
use ws::deflate::DeflateHandler;

#[test]
fn round_trip() {

    const MESSAGE: &'static str = "this is the message that will be sent as a message";

    let mut name = "Client";

    let mut ws = WebSocket::new(|output: Sender| {

        if name == "Client" {
            output.send(MESSAGE).unwrap();
        }

        let handler = move |msg: Message| {
            if name == "Server" {
                output.send(msg)
            } else {
                assert!(msg.as_text().unwrap() == MESSAGE);
                output.shutdown()
            }
        };

        name = "Server";

        DeflateHandler::new(handler)
    }).unwrap();

    let url = url::Url::parse("ws://127.0.0.1:3012").unwrap();

    ws.connect(url).unwrap();

    ws.listen("127.0.0.1:3012").unwrap();

}

#[test]
fn fragment() {

    env_logger::init().unwrap();
    const MESSAGE: &'static str = "Hello";

    let mut name = "Client";

    let mut ws = Builder::new().with_settings(Settings {
        fragment_size: 4,
        ..Default::default()
    }).build(|output: Sender| {

        if name == "Client" {
            output.send(MESSAGE).unwrap();
        }

        let handler = move |msg: Message| {
            if name == "Server" {
                output.send(msg)
            } else {
                assert!(msg.as_text().unwrap() == MESSAGE);
                output.shutdown()
            }
        };

        name = "Server";

        DeflateHandler::new(handler)

    }).unwrap();

    let url = url::Url::parse("ws://127.0.0.1:3024").unwrap();

    ws.connect(url).unwrap();

    ws.listen("127.0.0.1:3024").unwrap();
}