51 lines
1.5 KiB
Rust
51 lines
1.5 KiB
Rust
use crate::collab_stream_test::test_util::{random_i64, stream_client};
|
|
use collab_stream::model::StreamBinary;
|
|
|
|
#[tokio::test]
|
|
async fn read_single_message_test() {
|
|
let oid = format!("o{}", random_i64());
|
|
let client_2 = stream_client().await;
|
|
let mut stream_2 = client_2.stream("w1", &oid).await;
|
|
|
|
let (tx, mut rx) = tokio::sync::mpsc::channel(1);
|
|
tokio::spawn(async move {
|
|
let msg = stream_2.next().await.unwrap();
|
|
tx.send(msg).await.unwrap();
|
|
});
|
|
|
|
let msg = StreamBinary(vec![1, 2, 3]);
|
|
{
|
|
let client_1 = stream_client().await;
|
|
let mut stream_1 = client_1.stream("w1", &oid).await;
|
|
stream_1.insert_message(msg).await.unwrap();
|
|
}
|
|
|
|
let msg = rx.recv().await.unwrap().unwrap();
|
|
assert_eq!(msg.data, vec![1, 2, 3]);
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn read_multiple_messages_test() {
|
|
let oid = format!("o{}", random_i64());
|
|
let client_2 = stream_client().await;
|
|
let mut stream_2 = client_2.stream("w1", &oid).await;
|
|
stream_2.clear().await.unwrap();
|
|
|
|
{
|
|
let client_1 = stream_client().await;
|
|
let mut stream_1 = client_1.stream("w1", &oid).await;
|
|
let messages = vec![
|
|
StreamBinary(vec![1, 2, 3]),
|
|
StreamBinary(vec![4, 5, 6]),
|
|
StreamBinary(vec![7, 8, 9]),
|
|
];
|
|
stream_1.insert_messages(messages).await.unwrap();
|
|
}
|
|
|
|
let msg = stream_2.read_all_message().await.unwrap();
|
|
assert_eq!(msg.len(), 3);
|
|
assert_eq!(*msg[0], vec![1, 2, 3]);
|
|
assert_eq!(*msg[1], vec![4, 5, 6]);
|
|
assert_eq!(*msg[2], vec![7, 8, 9]);
|
|
}
|