1 files changed, 77 insertions, 0 deletions
diff --git a/third_party/rust/futures/tests/split.rs b/third_party/rust/futures/tests/split.rs
new file mode 100644
index 0000000000..9f4f1a07e2
--- /dev/null
+++ b/third_party/rust/futures/tests/split.rs
@@ -0,0 +1,77 @@
+use futures::executor::block_on;
+use futures::sink::{Sink, SinkExt};
+use futures::stream::{self, Stream, StreamExt};
+use futures::task::{Context, Poll};
+use pin_utils::unsafe_pinned;
+use std::pin::Pin;
+
+struct Join<T, U> {
+    stream: T,
+    sink: U
+}
+
+impl<T, U> Join<T, U> {
+    unsafe_pinned!(stream: T);
+    unsafe_pinned!(sink: U);
+}
+
+impl<T: Stream, U> Stream for Join<T, U> {
+    type Item = T::Item;
+
+    fn poll_next(
+        self: Pin<&mut Self>,
+        cx: &mut Context<'_>,
+    ) -> Poll<Option<T::Item>> {
+        self.stream().poll_next(cx)
+    }
+}
+
+impl<T, U: Sink<Item>, Item> Sink<Item> for Join<T, U> {
+    type Error = U::Error;
+
+    fn poll_ready(
+        self: Pin<&mut Self>,
+        cx: &mut Context<'_>,
+    ) -> Poll<Result<(), Self::Error>> {
+        self.sink().poll_ready(cx)
+    }
+
+    fn start_send(
+        self: Pin<&mut Self>,
+        item: Item,
+    ) -> Result<(), Self::Error> {
+        self.sink().start_send(item)
+    }
+
+    fn poll_flush(
+        self: Pin<&mut Self>,
+        cx: &mut Context<'_>,
+    ) -> Poll<Result<(), Self::Error>> {
+        self.sink().poll_flush(cx)
+    }
+
+    fn poll_close(
+        self: Pin<&mut Self>,
+        cx: &mut Context<'_>,
+    ) -> Poll<Result<(), Self::Error>> {
+        self.sink().poll_close(cx)
+    }
+}
+
+#[test]
+fn test_split() {
+    let mut dest: Vec<i32> = Vec::new();
+    {
+       let join = Join {
+            stream: stream::iter(vec![10, 20, 30]),
+            sink: &mut dest
+        };
+
+        let (sink, stream) = join.split();
+        let join = sink.reunite(stream).expect("test_split: reunite error");
+        let (mut sink, stream) = join.split();
+        let mut stream = stream.map(Ok);
+        block_on(sink.send_all(&mut stream)).unwrap();
+    }
+    assert_eq!(dest, vec![10, 20, 30]);
+}