Sha256: 60052b83b5eeb2395b77bc213f35098d2d5880529f0d83884582a8bbff78b139
Contents?: true
Size: 1.29 KB
Versions: 27
Compression:
Stored size: 1.29 KB
Contents
use futures::channel::mpsc;
use futures::executor::block_on;
use futures::stream::{abortable, Stream, StreamExt};
use futures::task::{Context, Poll};
use futures::SinkExt;
use futures_test::task::new_count_waker;
use std::pin::Pin;
#[test]
fn abortable_works() {
let (_tx, a_rx) = mpsc::channel::<()>(1);
let (mut abortable_rx, abort_handle) = abortable(a_rx);
abort_handle.abort();
assert!(abortable_rx.is_aborted());
assert_eq!(None, block_on(abortable_rx.next()));
}
#[test]
fn abortable_awakens() {
let (_tx, a_rx) = mpsc::channel::<()>(1);
let (mut abortable_rx, abort_handle) = abortable(a_rx);
let (waker, counter) = new_count_waker();
let mut cx = Context::from_waker(&waker);
assert_eq!(counter, 0);
assert_eq!(Poll::Pending, Pin::new(&mut abortable_rx).poll_next(&mut cx));
assert_eq!(counter, 0);
abort_handle.abort();
assert_eq!(counter, 1);
assert!(abortable_rx.is_aborted());
assert_eq!(Poll::Ready(None), Pin::new(&mut abortable_rx).poll_next(&mut cx));
}
#[test]
fn abortable_resolves() {
let (mut tx, a_rx) = mpsc::channel::<()>(1);
let (mut abortable_rx, _abort_handle) = abortable(a_rx);
block_on(tx.send(())).unwrap();
assert!(!abortable_rx.is_aborted());
assert_eq!(Some(()), block_on(abortable_rx.next()));
}
Version data entries
27 entries across 27 versions & 1 rubygems