diff options
author | Mikail Bagishov <bagishov.mikail@yandex.ru> | 2020-04-23 21:19:56 +0300 |
---|---|---|
committer | GitHub <noreply@github.com> | 2020-04-23 20:19:56 +0200 |
commit | 236629d1be7208612cbe5388e7ffebf85b73c157 (patch) | |
tree | 73ca0d102a5220d7693c4cacbdc0523fe7d1b1aa /tokio/tests | |
parent | f83f6388c42aa62c2096073b1dd80459189d7ea9 (diff) |
stream: fix panic in Merge and Chain size_hint (#2430)
Diffstat (limited to 'tokio/tests')
-rw-r--r-- | tokio/tests/stream_chain.rs | 24 | ||||
-rw-r--r-- | tokio/tests/stream_merge.rs | 24 |
2 files changed, 48 insertions, 0 deletions
diff --git a/tokio/tests/stream_chain.rs b/tokio/tests/stream_chain.rs index 0e14618b..98461a8c 100644 --- a/tokio/tests/stream_chain.rs +++ b/tokio/tests/stream_chain.rs @@ -69,3 +69,27 @@ async fn pending_first() { assert_eq!(stream.size_hint(), (0, None)); assert_eq!(None, assert_ready!(stream.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.chain(m2); + assert_eq!(m.size_hint(), (usize::max_value(), None)); +} 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)); +} |