summaryrefslogtreecommitdiffstats
path: root/third_party/rust/futures-0.1.31/tests/split.rs
diff options
context:
space:
mode:
Diffstat (limited to 'third_party/rust/futures-0.1.31/tests/split.rs')
-rw-r--r--third_party/rust/futures-0.1.31/tests/split.rs47
1 files changed, 47 insertions, 0 deletions
diff --git a/third_party/rust/futures-0.1.31/tests/split.rs b/third_party/rust/futures-0.1.31/tests/split.rs
new file mode 100644
index 0000000000..7a0667f135
--- /dev/null
+++ b/third_party/rust/futures-0.1.31/tests/split.rs
@@ -0,0 +1,47 @@
+extern crate futures;
+
+use futures::prelude::*;
+use futures::stream::iter_ok;
+
+struct Join<T, U>(T, U);
+
+impl<T: Stream, U> Stream for Join<T, U> {
+ type Item = T::Item;
+ type Error = T::Error;
+
+ fn poll(&mut self) -> Poll<Option<T::Item>, T::Error> {
+ self.0.poll()
+ }
+}
+
+impl<T, U: Sink> Sink for Join<T, U> {
+ type SinkItem = U::SinkItem;
+ type SinkError = U::SinkError;
+
+ fn start_send(&mut self, item: U::SinkItem)
+ -> StartSend<U::SinkItem, U::SinkError>
+ {
+ self.1.start_send(item)
+ }
+
+ fn poll_complete(&mut self) -> Poll<(), U::SinkError> {
+ self.1.poll_complete()
+ }
+
+ fn close(&mut self) -> Poll<(), U::SinkError> {
+ self.1.close()
+ }
+}
+
+#[test]
+fn test_split() {
+ let mut dest = Vec::new();
+ {
+ let j = Join(iter_ok(vec![10, 20, 30]), &mut dest);
+ let (sink, stream) = j.split();
+ let j = sink.reunite(stream).expect("test_split: reunite error");
+ let (sink, stream) = j.split();
+ sink.send_all(stream).wait().unwrap();
+ }
+ assert_eq!(dest, vec![10, 20, 30]);
+}