rust-http-body/tests/is_end_stream.rs

use http::HeaderMap;
use http_body::{Body, SizeHint};
use std::pin::Pin;
use std::task::{Context, Poll};

struct Mock {
    size_hint: SizeHint,
}

impl Body for Mock {
    type Data = ::std::io::Cursor<Vec<u8>>;
    type Error = ();

    fn poll_data(
        self: Pin<&mut Self>,
        _cx: &mut Context<'_>,
    ) -> Poll<Option<Result<Self::Data, Self::Error>>> {
        Poll::Ready(None)
    }

    fn poll_trailers(
        self: Pin<&mut Self>,
        _cx: &mut Context<'_>,
    ) -> Poll<Result<Option<HeaderMap>, Self::Error>> {
        Poll::Ready(Ok(None))
    }

    fn size_hint(&self) -> SizeHint {
        self.size_hint.clone()
    }
}

#[test]
fn is_end_stream_true() {
    let combos = [
        (None, None, false),
        (Some(123), None, false),
        (Some(0), Some(123), false),
        (Some(123), Some(123), false),
        (Some(0), Some(0), false),
    ];

    for &(lower, upper, is_end_stream) in &combos {
        let mut size_hint = SizeHint::new();
        assert_eq!(0, size_hint.lower());
        assert!(size_hint.upper().is_none());

        if let Some(lower) = lower {
            size_hint.set_lower(lower);
        }

        if let Some(upper) = upper {
            size_hint.set_upper(upper);
        }

        let mut mock = Mock { size_hint };

        assert_eq!(
            is_end_stream,
            Pin::new(&mut mock).is_end_stream(),
            "size_hint = {:?}",
            mock.size_hint.clone()
        );
    }
}

#[test]
fn is_end_stream_default_false() {
    let mut mock = Mock {
        size_hint: SizeHint::default(),
    };

    assert_eq!(
        false,
        Pin::new(&mut mock).is_end_stream(),
        "size_hint = {:?}",
        mock.size_hint.clone()
    );
}