From 236629d1be7208612cbe5388e7ffebf85b73c157 Mon Sep 17 00:00:00 2001
From: Mikail Bagishov <bagishov.mikail@yandex.ru>
Date: Thu, 23 Apr 2020 21:19:56 +0300
Subject: stream: fix panic in Merge and Chain size_hint (#2430)

---
 tokio/tests/stream_merge.rs | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

(limited to 'tokio/tests/stream_merge.rs')

diff --git a/tokio/tests/stream_merge.rs b/tokio/tests/stream_merge.rs
index f0168d72..45ecdcb6 100644
--- a/tokio/tests/stream_merge.rs
+++ b/tokio/tests/stream_merge.rs
@@ -52,3 +52,27 @@ async fn merge_async_streams() {
     assert!(rx.is_woken());
     assert_eq!(None, assert_ready!(rx.poll_next()));
 }
+
+#[test]
+fn size_overflow() {
+    struct Monster;
+
+    impl tokio::stream::Stream for Monster {
+        type Item = ();
+        fn poll_next(
+            self: std::pin::Pin<&mut Self>,
+            _cx: &mut std::task::Context<'_>,
+        ) -> std::task::Poll<Option<()>> {
+            panic!()
+        }
+
+        fn size_hint(&self) -> (usize, Option<usize>) {
+            (usize::max_value(), Some(usize::max_value()))
+        }
+    }
+
+    let m1 = Monster;
+    let m2 = Monster;
+    let m = m1.merge(m2);
+    assert_eq!(m.size_hint(), (usize::max_value(), None));
+}
-- 
cgit v1.2.3