deflate.rs

#![cfg(feature = "permessage-deflate")]
extern crate env_logger;
extern crate url;
extern crate ws;

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

#[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();
    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();
}