Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(1832)

Side by Side Diff: mojo/edk/system/awakable_list_unittest.cc

Issue 814543006: Move //mojo/{public, edk} underneath //third_party (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Rebase Created 5 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « mojo/edk/system/awakable_list.cc ('k') | mojo/edk/system/channel.h » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
(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/awakable_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(AwakableListTest, BasicCancel) {
25 MojoResult result;
26 uint32_t context;
27
28 // Cancel immediately after thread start.
29 {
30 AwakableList awakable_list;
31 test::SimpleWaiterThread thread(&result, &context);
32 awakable_list.Add(thread.waiter(), MOJO_HANDLE_SIGNAL_READABLE, 1);
33 thread.Start();
34 awakable_list.CancelAll();
35 // Double-remove okay:
36 awakable_list.Remove(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 AwakableList awakable_list;
44 test::SimpleWaiterThread thread(&result, &context);
45 awakable_list.Add(thread.waiter(), MOJO_HANDLE_SIGNAL_WRITABLE, 2);
46 awakable_list.CancelAll();
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 AwakableList awakable_list;
55 test::SimpleWaiterThread thread(&result, &context);
56 awakable_list.Add(thread.waiter(), MOJO_HANDLE_SIGNAL_READABLE, 3);
57 thread.Start();
58 base::PlatformThread::Sleep(2 * test::EpsilonTimeout());
59 awakable_list.CancelAll();
60 } // Join |thread|.
61 EXPECT_EQ(MOJO_RESULT_CANCELLED, result);
62 EXPECT_EQ(3u, context);
63 }
64
65 TEST(AwakableListTest, BasicAwakeSatisfied) {
66 MojoResult result;
67 uint32_t context;
68
69 // Awake immediately after thread start.
70 {
71 AwakableList awakable_list;
72 test::SimpleWaiterThread thread(&result, &context);
73 awakable_list.Add(thread.waiter(), MOJO_HANDLE_SIGNAL_READABLE, 1);
74 thread.Start();
75 awakable_list.AwakeForStateChange(HandleSignalsState(
76 MOJO_HANDLE_SIGNAL_READABLE,
77 MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE));
78 awakable_list.Remove(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 AwakableList awakable_list;
86 test::SimpleWaiterThread thread(&result, &context);
87 awakable_list.Add(thread.waiter(), MOJO_HANDLE_SIGNAL_WRITABLE, 2);
88 awakable_list.AwakeForStateChange(HandleSignalsState(
89 MOJO_HANDLE_SIGNAL_WRITABLE,
90 MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE));
91 awakable_list.Remove(thread.waiter());
92 // Double-remove okay:
93 awakable_list.Remove(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 AwakableList awakable_list;
102 test::SimpleWaiterThread thread(&result, &context);
103 awakable_list.Add(thread.waiter(), MOJO_HANDLE_SIGNAL_READABLE, 3);
104 thread.Start();
105 base::PlatformThread::Sleep(2 * test::EpsilonTimeout());
106 awakable_list.AwakeForStateChange(HandleSignalsState(
107 MOJO_HANDLE_SIGNAL_READABLE,
108 MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE));
109 awakable_list.Remove(thread.waiter());
110 } // Join |thread|.
111 EXPECT_EQ(MOJO_RESULT_OK, result);
112 EXPECT_EQ(3u, context);
113 }
114
115 TEST(AwakableListTest, BasicAwakeUnsatisfiable) {
116 MojoResult result;
117 uint32_t context;
118
119 // Awake (for unsatisfiability) immediately after thread start.
120 {
121 AwakableList awakable_list;
122 test::SimpleWaiterThread thread(&result, &context);
123 awakable_list.Add(thread.waiter(), MOJO_HANDLE_SIGNAL_READABLE, 1);
124 thread.Start();
125 awakable_list.AwakeForStateChange(HandleSignalsState(
126 MOJO_HANDLE_SIGNAL_NONE, MOJO_HANDLE_SIGNAL_WRITABLE));
127 awakable_list.Remove(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 AwakableList awakable_list;
135 test::SimpleWaiterThread thread(&result, &context);
136 awakable_list.Add(thread.waiter(), MOJO_HANDLE_SIGNAL_WRITABLE, 2);
137 awakable_list.AwakeForStateChange(HandleSignalsState(
138 MOJO_HANDLE_SIGNAL_READABLE, MOJO_HANDLE_SIGNAL_READABLE));
139 awakable_list.Remove(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 AwakableList awakable_list;
148 test::SimpleWaiterThread thread(&result, &context);
149 awakable_list.Add(thread.waiter(), MOJO_HANDLE_SIGNAL_READABLE, 3);
150 thread.Start();
151 base::PlatformThread::Sleep(2 * test::EpsilonTimeout());
152 awakable_list.AwakeForStateChange(HandleSignalsState(
153 MOJO_HANDLE_SIGNAL_NONE, MOJO_HANDLE_SIGNAL_WRITABLE));
154 awakable_list.Remove(thread.waiter());
155 // Double-remove okay:
156 awakable_list.Remove(thread.waiter());
157 } // Join |thread|.
158 EXPECT_EQ(MOJO_RESULT_FAILED_PRECONDITION, result);
159 EXPECT_EQ(3u, context);
160 }
161
162 TEST(AwakableListTest, MultipleAwakables) {
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 awakables.
173 {
174 AwakableList awakable_list;
175 test::SimpleWaiterThread thread1(&result1, &context1);
176 awakable_list.Add(thread1.waiter(), MOJO_HANDLE_SIGNAL_READABLE, 1);
177 thread1.Start();
178 test::SimpleWaiterThread thread2(&result2, &context2);
179 awakable_list.Add(thread2.waiter(), MOJO_HANDLE_SIGNAL_WRITABLE, 2);
180 thread2.Start();
181 base::PlatformThread::Sleep(2 * test::EpsilonTimeout());
182 awakable_list.CancelAll();
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 awakable, cancel other.
190 {
191 AwakableList awakable_list;
192 test::SimpleWaiterThread thread1(&result1, &context1);
193 awakable_list.Add(thread1.waiter(), MOJO_HANDLE_SIGNAL_READABLE, 3);
194 thread1.Start();
195 test::SimpleWaiterThread thread2(&result2, &context2);
196 awakable_list.Add(thread2.waiter(), MOJO_HANDLE_SIGNAL_WRITABLE, 4);
197 thread2.Start();
198 base::PlatformThread::Sleep(2 * test::EpsilonTimeout());
199 awakable_list.AwakeForStateChange(HandleSignalsState(
200 MOJO_HANDLE_SIGNAL_READABLE,
201 MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE));
202 awakable_list.Remove(thread1.waiter());
203 awakable_list.CancelAll();
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 awakable, awake other for unsatisfiability.
211 {
212 AwakableList awakable_list;
213 test::SimpleWaiterThread thread1(&result1, &context1);
214 awakable_list.Add(thread1.waiter(), MOJO_HANDLE_SIGNAL_READABLE, 5);
215 thread1.Start();
216 test::SimpleWaiterThread thread2(&result2, &context2);
217 awakable_list.Add(thread2.waiter(), MOJO_HANDLE_SIGNAL_WRITABLE, 6);
218 thread2.Start();
219 base::PlatformThread::Sleep(2 * test::EpsilonTimeout());
220 awakable_list.AwakeForStateChange(HandleSignalsState(
221 MOJO_HANDLE_SIGNAL_NONE, MOJO_HANDLE_SIGNAL_READABLE));
222 awakable_list.Remove(thread2.waiter());
223 awakable_list.CancelAll();
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 awakable, awake other for unsatisfiability.
231 {
232 AwakableList awakable_list;
233 test::SimpleWaiterThread thread1(&result1, &context1);
234 awakable_list.Add(thread1.waiter(), MOJO_HANDLE_SIGNAL_READABLE, 7);
235 thread1.Start();
236
237 base::PlatformThread::Sleep(1 * test::EpsilonTimeout());
238
239 // Should do nothing.
240 awakable_list.AwakeForStateChange(HandleSignalsState(
241 MOJO_HANDLE_SIGNAL_NONE,
242 MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE));
243
244 test::SimpleWaiterThread thread2(&result2, &context2);
245 awakable_list.Add(thread2.waiter(), MOJO_HANDLE_SIGNAL_WRITABLE, 8);
246 thread2.Start();
247
248 base::PlatformThread::Sleep(1 * test::EpsilonTimeout());
249
250 // Awake #1.
251 awakable_list.AwakeForStateChange(HandleSignalsState(
252 MOJO_HANDLE_SIGNAL_READABLE,
253 MOJO_HANDLE_SIGNAL_READABLE | MOJO_HANDLE_SIGNAL_WRITABLE));
254 awakable_list.Remove(thread1.waiter());
255
256 base::PlatformThread::Sleep(1 * test::EpsilonTimeout());
257
258 test::SimpleWaiterThread thread3(&result3, &context3);
259 awakable_list.Add(thread3.waiter(), MOJO_HANDLE_SIGNAL_WRITABLE, 9);
260 thread3.Start();
261
262 test::SimpleWaiterThread thread4(&result4, &context4);
263 awakable_list.Add(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 awakable_list.AwakeForStateChange(HandleSignalsState(
270 MOJO_HANDLE_SIGNAL_NONE, MOJO_HANDLE_SIGNAL_READABLE));
271 awakable_list.Remove(thread2.waiter());
272 awakable_list.Remove(thread3.waiter());
273
274 // Cancel #4.
275 awakable_list.CancelAll();
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 class KeepAwakable : public Awakable {
288 public:
289 KeepAwakable() : awake_count(0) {}
290
291 bool Awake(MojoResult result, uintptr_t context) override {
292 awake_count++;
293 return true;
294 }
295
296 int awake_count;
297
298 DISALLOW_COPY_AND_ASSIGN(KeepAwakable);
299 };
300
301 class RemoveAwakable : public Awakable {
302 public:
303 RemoveAwakable() : awake_count(0) {}
304
305 bool Awake(MojoResult result, uintptr_t context) override {
306 awake_count++;
307 return false;
308 }
309
310 int awake_count;
311
312 DISALLOW_COPY_AND_ASSIGN(RemoveAwakable);
313 };
314
315 TEST(AwakableListTest, KeepAwakablesReturningTrue) {
316 KeepAwakable keep0;
317 KeepAwakable keep1;
318 RemoveAwakable remove0;
319 RemoveAwakable remove1;
320 RemoveAwakable remove2;
321
322 HandleSignalsState hss(MOJO_HANDLE_SIGNAL_WRITABLE,
323 MOJO_HANDLE_SIGNAL_WRITABLE);
324
325 AwakableList remove_all;
326 remove_all.Add(&remove0, MOJO_HANDLE_SIGNAL_WRITABLE, 0);
327 remove_all.Add(&remove1, MOJO_HANDLE_SIGNAL_WRITABLE, 0);
328
329 remove_all.AwakeForStateChange(hss);
330 EXPECT_EQ(remove0.awake_count, 1);
331 EXPECT_EQ(remove1.awake_count, 1);
332
333 remove_all.AwakeForStateChange(hss);
334 EXPECT_EQ(remove0.awake_count, 1);
335 EXPECT_EQ(remove1.awake_count, 1);
336
337 AwakableList remove_first;
338 remove_first.Add(&remove2, MOJO_HANDLE_SIGNAL_WRITABLE, 0);
339 remove_first.Add(&keep0, MOJO_HANDLE_SIGNAL_WRITABLE, 0);
340 remove_first.Add(&keep1, MOJO_HANDLE_SIGNAL_WRITABLE, 0);
341
342 remove_first.AwakeForStateChange(hss);
343 EXPECT_EQ(keep0.awake_count, 1);
344 EXPECT_EQ(keep1.awake_count, 1);
345 EXPECT_EQ(remove2.awake_count, 1);
346
347 remove_first.AwakeForStateChange(hss);
348 EXPECT_EQ(keep0.awake_count, 2);
349 EXPECT_EQ(keep1.awake_count, 2);
350 EXPECT_EQ(remove2.awake_count, 1);
351
352 remove_first.Remove(&keep0);
353 remove_first.Remove(&keep1);
354 }
355
356 } // namespace
357 } // namespace system
358 } // namespace mojo
OLDNEW
« no previous file with comments | « mojo/edk/system/awakable_list.cc ('k') | mojo/edk/system/channel.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698