OLD | NEW |
| (Empty) |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 // NOTE(vtl): Some of these tests are inherently flaky (e.g., if run on a | |
6 // heavily-loaded system). Sorry. |test::EpsilonTimeout()| may be increased to | |
7 // increase tolerance and reduce observed flakiness (though doing so reduces the | |
8 // meaningfulness of the test). | |
9 | |
10 #include "mojo/edk/system/waiter_list.h" | |
11 | |
12 #include "base/threading/platform_thread.h" // For |Sleep()|. | |
13 #include "base/time/time.h" | |
14 #include "mojo/edk/system/handle_signals_state.h" | |
15 #include "mojo/edk/system/test_utils.h" | |
16 #include "mojo/edk/system/waiter.h" | |
17 #include "mojo/edk/system/waiter_test_utils.h" | |
18 #include "testing/gtest/include/gtest/gtest.h" | |
19 | |
20 namespace mojo { | |
21 namespace system { | |
22 namespace { | |
23 | |
24 TEST(WaiterListTest, BasicCancel) { | |
25 MojoResult result; | |
26 uint32_t context; | |
27 | |
28 // Cancel immediately after thread start. | |
29 { | |
30 WaiterList waiter_list; | |
31 test::SimpleWaiterThread thread(&result, &context); | |
32 waiter_list.AddWaiter(thread.waiter(), MOJO_HANDLE_SIGNAL_READABLE, 1); | |
33 thread.Start(); | |
34 waiter_list.CancelAllWaiters(); | |
35 // Double-remove okay: | |
36 waiter_list.RemoveWaiter(thread.waiter()); | |
37 } // Join |thread|. | |
38 EXPECT_EQ(MOJO_RESULT_CANCELLED, result); | |
39 EXPECT_EQ(1u, context); | |
40 | |
41 // Cancel before after thread start. | |
42 { | |
43 WaiterList waiter_list; | |
44 test::SimpleWaiterThread thread(&result, &context); | |
45 waiter_list.AddWaiter(thread.waiter(), MOJO_HANDLE_SIGNAL_WRITABLE, 2); | |
46 waiter_list.CancelAllWaiters(); | |
47 thread.Start(); | |
48 } // Join |thread|. | |
49 EXPECT_EQ(MOJO_RESULT_CANCELLED, result); | |
50 EXPECT_EQ(2u, context); | |
51 | |
52 // Cancel some time after thread start. | |
53 { | |
54 WaiterList waiter_list; | |
55 test::SimpleWaiterThread thread(&result, &context); | |
56 waiter_list.AddWaiter(thread.waiter(), MOJO_HANDLE_SIGNAL_READABLE, 3); | |
57 thread.Start(); | |
58 base::PlatformThread::Sleep(2 * test::EpsilonTimeout()); | |
59 waiter_list.CancelAllWaiters(); | |
60 } // Join |thread|. | |
61 EXPECT_EQ(MOJO_RESULT_CANCELLED, result); | |
62 EXPECT_EQ(3u, context); | |
63 } | |
64 | |
65 TEST(WaiterListTest, BasicAwakeSatisfied) { | |
66 MojoResult result; | |
67 uint32_t context; | |
68 | |
69 // Awake immediately after thread start. | |
70 { | |
71 WaiterList waiter_list; | |
72 test::SimpleWaiterThread thread(&result, &context); | |
73 waiter_list.AddWaiter(thread.waiter(), MOJO_HANDLE_SIGNAL_READABLE, 1); | |
74 thread.Start(); | |
75 waiter_list.AwakeWaitersForStateChange(HandleSignalsState( | |
76 MOJO_HANDLE_SIGNAL_READABLE, | |
77 MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE)); | |
78 waiter_list.RemoveWaiter(thread.waiter()); | |
79 } // Join |thread|. | |
80 EXPECT_EQ(MOJO_RESULT_OK, result); | |
81 EXPECT_EQ(1u, context); | |
82 | |
83 // Awake before after thread start. | |
84 { | |
85 WaiterList waiter_list; | |
86 test::SimpleWaiterThread thread(&result, &context); | |
87 waiter_list.AddWaiter(thread.waiter(), MOJO_HANDLE_SIGNAL_WRITABLE, 2); | |
88 waiter_list.AwakeWaitersForStateChange(HandleSignalsState( | |
89 MOJO_HANDLE_SIGNAL_WRITABLE, | |
90 MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE)); | |
91 waiter_list.RemoveWaiter(thread.waiter()); | |
92 // Double-remove okay: | |
93 waiter_list.RemoveWaiter(thread.waiter()); | |
94 thread.Start(); | |
95 } // Join |thread|. | |
96 EXPECT_EQ(MOJO_RESULT_OK, result); | |
97 EXPECT_EQ(2u, context); | |
98 | |
99 // Awake some time after thread start. | |
100 { | |
101 WaiterList waiter_list; | |
102 test::SimpleWaiterThread thread(&result, &context); | |
103 waiter_list.AddWaiter(thread.waiter(), MOJO_HANDLE_SIGNAL_READABLE, 3); | |
104 thread.Start(); | |
105 base::PlatformThread::Sleep(2 * test::EpsilonTimeout()); | |
106 waiter_list.AwakeWaitersForStateChange(HandleSignalsState( | |
107 MOJO_HANDLE_SIGNAL_READABLE, | |
108 MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE)); | |
109 waiter_list.RemoveWaiter(thread.waiter()); | |
110 } // Join |thread|. | |
111 EXPECT_EQ(MOJO_RESULT_OK, result); | |
112 EXPECT_EQ(3u, context); | |
113 } | |
114 | |
115 TEST(WaiterListTest, BasicAwakeUnsatisfiable) { | |
116 MojoResult result; | |
117 uint32_t context; | |
118 | |
119 // Awake (for unsatisfiability) immediately after thread start. | |
120 { | |
121 WaiterList waiter_list; | |
122 test::SimpleWaiterThread thread(&result, &context); | |
123 waiter_list.AddWaiter(thread.waiter(), MOJO_HANDLE_SIGNAL_READABLE, 1); | |
124 thread.Start(); | |
125 waiter_list.AwakeWaitersForStateChange(HandleSignalsState( | |
126 MOJO_HANDLE_SIGNAL_NONE, MOJO_HANDLE_SIGNAL_WRITABLE)); | |
127 waiter_list.RemoveWaiter(thread.waiter()); | |
128 } // Join |thread|. | |
129 EXPECT_EQ(MOJO_RESULT_FAILED_PRECONDITION, result); | |
130 EXPECT_EQ(1u, context); | |
131 | |
132 // Awake (for unsatisfiability) before after thread start. | |
133 { | |
134 WaiterList waiter_list; | |
135 test::SimpleWaiterThread thread(&result, &context); | |
136 waiter_list.AddWaiter(thread.waiter(), MOJO_HANDLE_SIGNAL_WRITABLE, 2); | |
137 waiter_list.AwakeWaitersForStateChange(HandleSignalsState( | |
138 MOJO_HANDLE_SIGNAL_READABLE, MOJO_HANDLE_SIGNAL_READABLE)); | |
139 waiter_list.RemoveWaiter(thread.waiter()); | |
140 thread.Start(); | |
141 } // Join |thread|. | |
142 EXPECT_EQ(MOJO_RESULT_FAILED_PRECONDITION, result); | |
143 EXPECT_EQ(2u, context); | |
144 | |
145 // Awake (for unsatisfiability) some time after thread start. | |
146 { | |
147 WaiterList waiter_list; | |
148 test::SimpleWaiterThread thread(&result, &context); | |
149 waiter_list.AddWaiter(thread.waiter(), MOJO_HANDLE_SIGNAL_READABLE, 3); | |
150 thread.Start(); | |
151 base::PlatformThread::Sleep(2 * test::EpsilonTimeout()); | |
152 waiter_list.AwakeWaitersForStateChange(HandleSignalsState( | |
153 MOJO_HANDLE_SIGNAL_NONE, MOJO_HANDLE_SIGNAL_WRITABLE)); | |
154 waiter_list.RemoveWaiter(thread.waiter()); | |
155 // Double-remove okay: | |
156 waiter_list.RemoveWaiter(thread.waiter()); | |
157 } // Join |thread|. | |
158 EXPECT_EQ(MOJO_RESULT_FAILED_PRECONDITION, result); | |
159 EXPECT_EQ(3u, context); | |
160 } | |
161 | |
162 TEST(WaiterListTest, MultipleWaiters) { | |
163 MojoResult result1; | |
164 MojoResult result2; | |
165 MojoResult result3; | |
166 MojoResult result4; | |
167 uint32_t context1; | |
168 uint32_t context2; | |
169 uint32_t context3; | |
170 uint32_t context4; | |
171 | |
172 // Cancel two waiters. | |
173 { | |
174 WaiterList waiter_list; | |
175 test::SimpleWaiterThread thread1(&result1, &context1); | |
176 waiter_list.AddWaiter(thread1.waiter(), MOJO_HANDLE_SIGNAL_READABLE, 1); | |
177 thread1.Start(); | |
178 test::SimpleWaiterThread thread2(&result2, &context2); | |
179 waiter_list.AddWaiter(thread2.waiter(), MOJO_HANDLE_SIGNAL_WRITABLE, 2); | |
180 thread2.Start(); | |
181 base::PlatformThread::Sleep(2 * test::EpsilonTimeout()); | |
182 waiter_list.CancelAllWaiters(); | |
183 } // Join threads. | |
184 EXPECT_EQ(MOJO_RESULT_CANCELLED, result1); | |
185 EXPECT_EQ(1u, context1); | |
186 EXPECT_EQ(MOJO_RESULT_CANCELLED, result2); | |
187 EXPECT_EQ(2u, context2); | |
188 | |
189 // Awake one waiter, cancel other. | |
190 { | |
191 WaiterList waiter_list; | |
192 test::SimpleWaiterThread thread1(&result1, &context1); | |
193 waiter_list.AddWaiter(thread1.waiter(), MOJO_HANDLE_SIGNAL_READABLE, 3); | |
194 thread1.Start(); | |
195 test::SimpleWaiterThread thread2(&result2, &context2); | |
196 waiter_list.AddWaiter(thread2.waiter(), MOJO_HANDLE_SIGNAL_WRITABLE, 4); | |
197 thread2.Start(); | |
198 base::PlatformThread::Sleep(2 * test::EpsilonTimeout()); | |
199 waiter_list.AwakeWaitersForStateChange(HandleSignalsState( | |
200 MOJO_HANDLE_SIGNAL_READABLE, | |
201 MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE)); | |
202 waiter_list.RemoveWaiter(thread1.waiter()); | |
203 waiter_list.CancelAllWaiters(); | |
204 } // Join threads. | |
205 EXPECT_EQ(MOJO_RESULT_OK, result1); | |
206 EXPECT_EQ(3u, context1); | |
207 EXPECT_EQ(MOJO_RESULT_CANCELLED, result2); | |
208 EXPECT_EQ(4u, context2); | |
209 | |
210 // Cancel one waiter, awake other for unsatisfiability. | |
211 { | |
212 WaiterList waiter_list; | |
213 test::SimpleWaiterThread thread1(&result1, &context1); | |
214 waiter_list.AddWaiter(thread1.waiter(), MOJO_HANDLE_SIGNAL_READABLE, 5); | |
215 thread1.Start(); | |
216 test::SimpleWaiterThread thread2(&result2, &context2); | |
217 waiter_list.AddWaiter(thread2.waiter(), MOJO_HANDLE_SIGNAL_WRITABLE, 6); | |
218 thread2.Start(); | |
219 base::PlatformThread::Sleep(2 * test::EpsilonTimeout()); | |
220 waiter_list.AwakeWaitersForStateChange(HandleSignalsState( | |
221 MOJO_HANDLE_SIGNAL_NONE, MOJO_HANDLE_SIGNAL_READABLE)); | |
222 waiter_list.RemoveWaiter(thread2.waiter()); | |
223 waiter_list.CancelAllWaiters(); | |
224 } // Join threads. | |
225 EXPECT_EQ(MOJO_RESULT_CANCELLED, result1); | |
226 EXPECT_EQ(5u, context1); | |
227 EXPECT_EQ(MOJO_RESULT_FAILED_PRECONDITION, result2); | |
228 EXPECT_EQ(6u, context2); | |
229 | |
230 // Cancel one waiter, awake other for unsatisfiability. | |
231 { | |
232 WaiterList waiter_list; | |
233 test::SimpleWaiterThread thread1(&result1, &context1); | |
234 waiter_list.AddWaiter(thread1.waiter(), MOJO_HANDLE_SIGNAL_READABLE, 7); | |
235 thread1.Start(); | |
236 | |
237 base::PlatformThread::Sleep(1 * test::EpsilonTimeout()); | |
238 | |
239 // Should do nothing. | |
240 waiter_list.AwakeWaitersForStateChange(HandleSignalsState( | |
241 MOJO_HANDLE_SIGNAL_NONE, | |
242 MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE)); | |
243 | |
244 test::SimpleWaiterThread thread2(&result2, &context2); | |
245 waiter_list.AddWaiter(thread2.waiter(), MOJO_HANDLE_SIGNAL_WRITABLE, 8); | |
246 thread2.Start(); | |
247 | |
248 base::PlatformThread::Sleep(1 * test::EpsilonTimeout()); | |
249 | |
250 // Awake #1. | |
251 waiter_list.AwakeWaitersForStateChange(HandleSignalsState( | |
252 MOJO_HANDLE_SIGNAL_READABLE, | |
253 MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE)); | |
254 waiter_list.RemoveWaiter(thread1.waiter()); | |
255 | |
256 base::PlatformThread::Sleep(1 * test::EpsilonTimeout()); | |
257 | |
258 test::SimpleWaiterThread thread3(&result3, &context3); | |
259 waiter_list.AddWaiter(thread3.waiter(), MOJO_HANDLE_SIGNAL_WRITABLE, 9); | |
260 thread3.Start(); | |
261 | |
262 test::SimpleWaiterThread thread4(&result4, &context4); | |
263 waiter_list.AddWaiter(thread4.waiter(), MOJO_HANDLE_SIGNAL_READABLE, 10); | |
264 thread4.Start(); | |
265 | |
266 base::PlatformThread::Sleep(1 * test::EpsilonTimeout()); | |
267 | |
268 // Awake #2 and #3 for unsatisfiability. | |
269 waiter_list.AwakeWaitersForStateChange(HandleSignalsState( | |
270 MOJO_HANDLE_SIGNAL_NONE, MOJO_HANDLE_SIGNAL_READABLE)); | |
271 waiter_list.RemoveWaiter(thread2.waiter()); | |
272 waiter_list.RemoveWaiter(thread3.waiter()); | |
273 | |
274 // Cancel #4. | |
275 waiter_list.CancelAllWaiters(); | |
276 } // Join threads. | |
277 EXPECT_EQ(MOJO_RESULT_OK, result1); | |
278 EXPECT_EQ(7u, context1); | |
279 EXPECT_EQ(MOJO_RESULT_FAILED_PRECONDITION, result2); | |
280 EXPECT_EQ(8u, context2); | |
281 EXPECT_EQ(MOJO_RESULT_FAILED_PRECONDITION, result3); | |
282 EXPECT_EQ(9u, context3); | |
283 EXPECT_EQ(MOJO_RESULT_CANCELLED, result4); | |
284 EXPECT_EQ(10u, context4); | |
285 } | |
286 | |
287 } // namespace | |
288 } // namespace system | |
289 } // namespace mojo | |
OLD | NEW |