This repository has been archived by the owner on Jan 2, 2020. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 0
/
awakener.rs
97 lines (75 loc) · 2.61 KB
/
awakener.rs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
use std::thread;
use std::time::Duration;
use mio_st::event::{Event, EventedId, Ready};
use mio_st::poll::Awakener;
mod util;
use self::util::{expect_events, init_with_poller};
#[test]
fn awakener() {
let (mut poller, mut events) = init_with_poller();
let event_id = EventedId(10);
// Keep `awakener` alive on this thread and create a new awakener to move
// to the other thread.
let awakener = Awakener::new(&mut poller, event_id)
.expect("unable to create awakener");
let awakener1 = awakener.try_clone()
.expect("unable to clone awakener");
let handle = thread::spawn(move || {
awakener1.wake().expect("unable to wake");
});
expect_events(&mut poller, &mut events, vec![
Event::new(event_id, Ready::READABLE),
]);
handle.join().unwrap();
}
#[test]
fn awakener_try_clone() {
let (mut poller, mut events) = init_with_poller();
let event_id = EventedId(10);
// Keep `awakener` alive on this thread and create two new awakeners to move
// to the other threads.
let awakener = Awakener::new(&mut poller, event_id)
.expect("unable to create awakener");
let awakener1 = awakener.try_clone()
.expect("unable to clone awakener");
let awakener2 = awakener1.try_clone()
.expect("unable to clone awakener");
let handle1 = thread::spawn(move || {
awakener1.wake().expect("unable to wake");
});
let handle2 = thread::spawn(move || {
thread::sleep(Duration::from_millis(500));
awakener2.wake().expect("unable to wake");
});
expect_events(&mut poller, &mut events, vec![
Event::new(event_id, Ready::READABLE),
]);
expect_events(&mut poller, &mut events, vec![
Event::new(event_id, Ready::READABLE),
]);
handle1.join().unwrap();
handle2.join().unwrap();
}
#[test]
fn awakener_drain() {
let (mut poller, mut events) = init_with_poller();
let event_id = EventedId(10);
let awakener = Awakener::new(&mut poller, event_id)
.expect("unable to create awakener");
for _ in 0..3 {
awakener.wake().expect("unable to wake");
}
// Depending on the platform we can get 3 or 1 event.
expect_events(&mut poller, &mut events, vec![
Event::new(event_id, Ready::READABLE),
]);
awakener.drain().expect("unable to drain awakener");
}
#[test]
fn awakener_drain_empty() {
let (mut poller, _) = init_with_poller();
let event_id = EventedId(10);
let awakener = Awakener::new(&mut poller, event_id)
.expect("unable to create awakener");
awakener.drain().expect("unable to drain awakener");
}