|
1 | 1 | use std::time::Duration; |
2 | 2 |
|
3 | | -use async_std::{prelude::*, task, sync::channel}; |
| 3 | +use async_std::prelude::*; |
| 4 | + |
| 5 | +use async_std::channel; |
| 6 | +use async_std::task; |
4 | 7 |
|
5 | 8 | use stop_token::StopSource; |
6 | 9 |
|
7 | 10 | #[test] |
8 | 11 | fn smoke() { |
9 | 12 | task::block_on(async { |
10 | | - let (sender, receiver) = channel::<i32>(10); |
| 13 | + let (sender, receiver) = channel::bounded::<i32>(10); |
11 | 14 | let stop_source = StopSource::new(); |
12 | 15 | let task = task::spawn({ |
13 | 16 | let stop_token = stop_source.stop_token(); |
14 | 17 | let receiver = receiver.clone(); |
15 | 18 | async move { |
16 | | - let mut xs = Vec::new(); |
17 | | - let mut stream = stop_token.stop_stream(receiver); |
18 | | - while let Some(x) = stream.next().await { |
19 | | - xs.push(x) |
| 19 | + let mut xs = Vec::new(); |
| 20 | + let mut stream = stop_token.stop_stream(receiver); |
| 21 | + while let Some(x) = stream.next().await { |
| 22 | + xs.push(x) |
| 23 | + } |
| 24 | + xs |
20 | 25 | } |
21 | | - xs |
22 | | - }}); |
23 | | - sender.send(1).await; |
24 | | - sender.send(2).await; |
25 | | - sender.send(3).await; |
| 26 | + }); |
| 27 | + sender.send(1).await.unwrap(); |
| 28 | + sender.send(2).await.unwrap(); |
| 29 | + sender.send(3).await.unwrap(); |
26 | 30 |
|
27 | 31 | task::sleep(Duration::from_millis(250)).await; |
28 | 32 | drop(stop_source); |
29 | 33 | task::sleep(Duration::from_millis(250)).await; |
30 | 34 |
|
31 | | - sender.send(4).await; |
32 | | - sender.send(5).await; |
33 | | - sender.send(6).await; |
| 35 | + sender.send(4).await.unwrap(); |
| 36 | + sender.send(5).await.unwrap(); |
| 37 | + sender.send(6).await.unwrap(); |
34 | 38 | assert_eq!(task.await, vec![1, 2, 3]); |
35 | 39 | }) |
36 | 40 | } |
0 commit comments