Jason Macnak | 09c3688 | 2020-04-01 16:22:56 +0000 | [diff] [blame^] | 1 | use futures::channel::mpsc; |
| 2 | use futures::executor::block_on; |
| 3 | use futures::future::poll_fn; |
| 4 | use futures::stream::StreamExt; |
| 5 | use futures::sink::SinkExt; |
| 6 | use std::sync::atomic::{AtomicUsize, Ordering}; |
| 7 | use std::thread; |
| 8 | |
| 9 | #[test] |
| 10 | fn sequence() { |
| 11 | let (tx, rx) = mpsc::channel(1); |
| 12 | |
| 13 | let amt = 20; |
| 14 | let t = thread::spawn(move || { |
| 15 | block_on(send_sequence(amt, tx)) |
| 16 | }); |
| 17 | let list: Vec<_> = block_on(rx.collect()); |
| 18 | let mut list = list.into_iter(); |
| 19 | for i in (1..=amt).rev() { |
| 20 | assert_eq!(list.next(), Some(i)); |
| 21 | } |
| 22 | assert_eq!(list.next(), None); |
| 23 | |
| 24 | t.join().unwrap(); |
| 25 | } |
| 26 | |
| 27 | async fn send_sequence(n: u32, mut sender: mpsc::Sender<u32>) { |
| 28 | for x in 0..n { |
| 29 | sender.send(n - x).await.unwrap(); |
| 30 | } |
| 31 | } |
| 32 | |
| 33 | #[test] |
| 34 | fn drop_sender() { |
| 35 | let (tx, mut rx) = mpsc::channel::<u32>(1); |
| 36 | drop(tx); |
| 37 | let f = poll_fn(|cx| { |
| 38 | rx.poll_next_unpin(cx) |
| 39 | }); |
| 40 | assert_eq!(block_on(f), None) |
| 41 | } |
| 42 | |
| 43 | #[test] |
| 44 | fn drop_rx() { |
| 45 | let (mut tx, rx) = mpsc::channel::<u32>(1); |
| 46 | block_on(tx.send(1)).unwrap(); |
| 47 | drop(rx); |
| 48 | assert!(block_on(tx.send(1)).is_err()); |
| 49 | } |
| 50 | |
| 51 | #[test] |
| 52 | fn drop_order() { |
| 53 | static DROPS: AtomicUsize = AtomicUsize::new(0); |
| 54 | let (mut tx, rx) = mpsc::channel(1); |
| 55 | |
| 56 | struct A; |
| 57 | |
| 58 | impl Drop for A { |
| 59 | fn drop(&mut self) { |
| 60 | DROPS.fetch_add(1, Ordering::SeqCst); |
| 61 | } |
| 62 | } |
| 63 | |
| 64 | block_on(tx.send(A)).unwrap(); |
| 65 | assert_eq!(DROPS.load(Ordering::SeqCst), 0); |
| 66 | drop(rx); |
| 67 | assert_eq!(DROPS.load(Ordering::SeqCst), 1); |
| 68 | assert!(block_on(tx.send(A)).is_err()); |
| 69 | assert_eq!(DROPS.load(Ordering::SeqCst), 2); |
| 70 | } |