#[cfg(test)]
mod tests {
use std::error::Error;
use futures_util::io::{AsyncBufReadExt, AsyncReadExt};
use glib::Bytes;
use crate::{prelude::*, MemoryInputStream};
#[test]
fn new() {
let strm = MemoryInputStream::new();
let ret = strm.skip(1, crate::Cancellable::NONE);
assert!(ret.is_ok());
assert_eq!(ret.unwrap(), 0);
let mut buf = vec![0; 10];
let ret = strm.read(&mut buf, crate::Cancellable::NONE).unwrap();
assert_eq!(ret, 0);
}
#[test]
fn from_bytes() {
let b = Bytes::from_owned(vec![1, 2, 3]);
let strm = MemoryInputStream::from_bytes(&b);
let mut buf = vec![0; 10];
let ret = strm.read(&mut buf, crate::Cancellable::NONE).unwrap();
assert_eq!(ret, 3);
assert_eq!(buf[0], 1);
assert_eq!(buf[1], 2);
assert_eq!(buf[2], 3);
let ret = strm.skip(10, crate::Cancellable::NONE).unwrap();
assert_eq!(ret, 0);
}
#[test]
fn add_bytes() {
let strm = MemoryInputStream::new();
let b = Bytes::from_owned(vec![1, 2, 3]);
strm.add_bytes(&b);
let mut buf = vec![0; 10];
let ret = strm.read(&mut buf, crate::Cancellable::NONE).unwrap();
assert_eq!(ret, 3);
assert_eq!(buf[0], 1);
assert_eq!(buf[1], 2);
assert_eq!(buf[2], 3);
let ret = strm.skip(10, crate::Cancellable::NONE).unwrap();
assert_eq!(ret, 0);
}
#[test]
fn read_future() {
use futures_util::future::TryFutureExt;
let c = glib::MainContext::new();
let buf = vec![0; 10];
let b = glib::Bytes::from_owned(vec![1, 2, 3]);
let strm = MemoryInputStream::from_bytes(&b);
let res = c
.block_on(
strm.read_future(buf, glib::Priority::default())
.map_err(|(_buf, err)| err)
.map_ok(move |(mut buf, len)| {
buf.truncate(len);
buf
}),
)
.unwrap();
assert_eq!(res, vec![1, 2, 3]);
}
#[test]
fn async_read() {
async fn run() -> Result<(), Box<dyn Error>> {
let b = Bytes::from_owned(vec![1, 2, 3]);
let mut read = MemoryInputStream::from_bytes(&b).into_async_buf_read(8);
let mut buf = [0u8; 4];
assert_eq!(read.read(&mut buf).await?, 3);
assert_eq!(buf, [1, 2, 3, 0]);
assert_eq!(read.read(&mut buf).await?, 0);
let mut read = MemoryInputStream::from_bytes(&b).into_async_buf_read(8);
let mut buf = [0u8; 1];
assert_eq!(read.read(&mut buf).await?, 1);
assert_eq!(buf, [1]);
assert_eq!(read.read(&mut buf).await?, 1);
assert_eq!(buf, [2]);
assert_eq!(read.read(&mut buf).await?, 1);
assert_eq!(buf, [3]);
assert_eq!(read.read(&mut buf).await?, 0);
let mut read = MemoryInputStream::from_bytes(&b).into_async_buf_read(2);
let mut buf = [0u8; 4];
assert_eq!(read.read(&mut buf).await?, 2);
assert_eq!(buf, [1, 2, 0, 0]);
assert_eq!(read.read(&mut buf).await?, 1);
assert_eq!(buf[0], 3);
assert_eq!(read.read(&mut buf).await?, 0);
let mut read = MemoryInputStream::from_bytes(&b).into_async_buf_read(2);
let mut buf = [0u8; 1];
assert_eq!(read.read(&mut buf).await?, 1);
assert_eq!(buf, [1]);
assert_eq!(read.read(&mut buf).await?, 1);
assert_eq!(buf, [2]);
assert_eq!(read.read(&mut buf).await?, 1);
assert_eq!(buf, [3]);
assert_eq!(read.read(&mut buf).await?, 0);
Ok(())
}
let main_context = glib::MainContext::new();
main_context.block_on(run()).unwrap();
}
#[test]
fn async_buf_read() {
async fn run() -> Result<(), Box<dyn Error>> {
let b = Bytes::from_owned(vec![1, 2, 3]);
let mut read = MemoryInputStream::from_bytes(&b).into_async_buf_read(16);
let mut buf = String::new();
assert_eq!(read.read_line(&mut buf).await?, 3);
assert_eq!(buf.as_bytes(), [1, 2, 3]);
assert_eq!(read.read_line(&mut buf).await?, 0);
let mut read = MemoryInputStream::from_bytes(&b).into_async_buf_read(2);
let mut buf = String::new();
assert_eq!(read.read_line(&mut buf).await?, 3);
assert_eq!(buf.as_bytes(), [1, 2, 3]);
assert_eq!(read.read_line(&mut buf).await?, 0);
Ok(())
}
let main_context = glib::MainContext::new();
main_context.block_on(run()).unwrap();
}
}