io_cursor.rs - mozsearch

Enable keyboard shortcuts

use assert_matches::assert_matches;

use futures::executor::block_on;

use futures::future::lazy;

use futures::io::{AsyncWrite, Cursor};

use futures::task::Poll;

use std::pin::Pin;

#[test]

fn cursor_asyncwrite_vec() {

    let mut cursor = Cursor::new(vec![0; 5]);

    block_on(lazy(|cx| {

        assert_matches!(Pin::new(&mut cursor).poll_write(cx, &[1, 2]), Poll::Ready(Ok(2)));

        assert_matches!(Pin::new(&mut cursor).poll_write(cx, &[3, 4]), Poll::Ready(Ok(2)));

        assert_matches!(Pin::new(&mut cursor).poll_write(cx, &[5, 6]), Poll::Ready(Ok(2)));

        assert_matches!(Pin::new(&mut cursor).poll_write(cx, &[6, 7]), Poll::Ready(Ok(2)));

    }));

    assert_eq!(cursor.into_inner(), [1, 2, 3, 4, 5, 6, 6, 7]);

#[test]

fn cursor_asyncwrite_box() {

    let mut cursor = Cursor::new(vec![0; 5].into_boxed_slice());

    block_on(lazy(|cx| {

        assert_matches!(Pin::new(&mut cursor).poll_write(cx, &[1, 2]), Poll::Ready(Ok(2)));

        assert_matches!(Pin::new(&mut cursor).poll_write(cx, &[3, 4]), Poll::Ready(Ok(2)));

        assert_matches!(Pin::new(&mut cursor).poll_write(cx, &[5, 6]), Poll::Ready(Ok(1)));

        assert_matches!(Pin::new(&mut cursor).poll_write(cx, &[6, 7]), Poll::Ready(Ok(0)));

    }));

    assert_eq!(&*cursor.into_inner(), [1, 2, 3, 4, 5]);