summaryrefslogtreecommitdiffstats
path: root/third_party/rust/http-body/tests
diff options
context:
space:
mode:
Diffstat (limited to 'third_party/rust/http-body/tests')
-rw-r--r--third_party/rust/http-body/tests/is_end_stream.rs79
1 files changed, 79 insertions, 0 deletions
diff --git a/third_party/rust/http-body/tests/is_end_stream.rs b/third_party/rust/http-body/tests/is_end_stream.rs
new file mode 100644
index 0000000000..beaeb0b1a0
--- /dev/null
+++ b/third_party/rust/http-body/tests/is_end_stream.rs
@@ -0,0 +1,79 @@
+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()
+ );
+}