berkus
/
actix-web
mirror of https://github.com/fafhrd91/actix-web
1
0
Fork 0
Code Issues Releases Wiki Activity

Websocket continuation tests

This commit is contained in:
Sebastian Urban 2018-11-08 16:25:06 +01:00
parent 2c0ef6b5fe
commit d2a160a0d3
1 changed files with 50 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

50
tests/test_ws.rs
View File

@ -147,6 +147,31 @@ fn test_large_text() {
}
}
#[test]
fn test_continuation_text() {
let data1 = rand::thread_rng()
.sample_iter(&Alphanumeric)
.take(10_000)
.collect::<String>();
let data2 = rand::thread_rng()
.sample_iter(&Alphanumeric)
.take(10_000)
.collect::<String>();
let data = data1.clone() + &data2;
let mut srv = test::TestServer::new(|app| app.handler(|req| ws::start(req, Ws)));
let (mut reader, mut writer) = srv.ws().unwrap();
for _ in 0..100 {
writer.text_part(data1.clone(), false);
writer.text_part(data2.clone(), true);
let (item, r) = srv.execute(reader.into_future()).unwrap();
reader = r;
assert_eq!(item, Some(ws::Message::Text(data.clone())));
}
}
#[test]
fn test_large_bin() {
let data = rand::thread_rng()
@ -165,6 +190,31 @@ fn test_large_bin() {
}
}
#[test]
fn test_continuation_binary() {
let data1 = rand::thread_rng()
.sample_iter(&Alphanumeric)
.take(10_000)
.collect::<String>();
let data2 = rand::thread_rng()
.sample_iter(&Alphanumeric)
.take(10_000)
.collect::<String>();
let data = data1.clone() + &data2;
let mut srv = test::TestServer::new(|app| app.handler(|req| ws::start(req, Ws)));
let (mut reader, mut writer) = srv.ws().unwrap();
for _ in 0..100 {
writer.binary_part(data1.clone(), false);
writer.binary_part(data2.clone(), true);
let (item, r) = srv.execute(reader.into_future()).unwrap();
reader = r;
assert_eq!(item, Some(ws::Message::Binary(Binary::from(data.clone()))));
}
}
#[test]
fn test_client_frame_size() {
let data = rand::thread_rng()