compat.rs

use std::io::Cursor;

use compio_io::compat::AsyncStream;
use futures_executor::block_on;
use futures_util::{AsyncBufReadExt, AsyncReadExt, AsyncWriteExt};

#[test]
fn async_compat_read() {
    block_on(async {
        let src = &[1u8, 1, 4, 5, 1, 4, 1, 9, 1, 9, 8, 1, 0][..];
        let stream = AsyncStream::new(src);
        let mut stream = std::pin::pin!(stream);

        let mut buf = [0; 6];
        let len = stream.read(&mut buf).await.unwrap();

        assert_eq!(len, 6);
        assert_eq!(buf, [1, 1, 4, 5, 1, 4]);

        let mut buf = [0; 20];
        let len = stream.read(&mut buf).await.unwrap();
        assert_eq!(len, 7);
        assert_eq!(&buf[..7], [1, 9, 1, 9, 8, 1, 0]);
    })
}

#[test]
fn async_compat_bufread() {
    block_on(async {
        let src = &[1u8, 1, 4, 5, 1, 4, 1, 9, 1, 9, 8, 1, 0][..];
        let stream = AsyncStream::new(src);
        let mut stream = std::pin::pin!(stream);

        let slice = stream.fill_buf().await.unwrap();
        assert_eq!(slice, [1, 1, 4, 5, 1, 4, 1, 9, 1, 9, 8, 1, 0]);
        stream.consume_unpin(6);

        let mut buf = [0; 7];
        let len = stream.read(&mut buf).await.unwrap();

        assert_eq!(len, 7);
        assert_eq!(buf, [1, 9, 1, 9, 8, 1, 0]);
    })
}

#[test]
fn async_compat_write() {
    block_on(async {
        let dst = Cursor::new([0u8; 10]);
        let stream = AsyncStream::new(dst);
        let mut stream = std::pin::pin!(stream);

        let len = stream.write(&[1, 1, 4, 5, 1, 4]).await.unwrap();
        stream.flush().await.unwrap();

        assert_eq!(len, 6);
        assert_eq!(stream.get_ref().position(), 6);
        assert_eq!(stream.get_ref().get_ref(), &[1, 1, 4, 5, 1, 4, 0, 0, 0, 0]);

        let dst = Cursor::new([0u8; 10]);
        let stream = AsyncStream::with_capacity(10, dst);
        let mut stream = std::pin::pin!(stream);

        let len = stream
            .write(&[1, 1, 4, 5, 1, 4, 1, 9, 1, 9, 8, 1, 0])
            .await
            .unwrap();
        assert_eq!(len, 13);

        let err = stream.flush().await.unwrap_err();
        assert_eq!(err.kind(), std::io::ErrorKind::WriteZero);
    })
}

#[test]
fn async_compat_flush_fail() {
    block_on(async {
        let dst = Cursor::new([0u8; 10]);
        let stream = AsyncStream::new(dst);
        let mut stream = std::pin::pin!(stream);
        let len = stream
            .write(&[1, 1, 4, 5, 1, 4, 1, 9, 1, 9, 8, 1, 0])
            .await
            .unwrap();
        assert_eq!(len, 13);
        let err = stream.flush().await.unwrap_err();
        assert_eq!(err.kind(), std::io::ErrorKind::WriteZero);
    })
}