| 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/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 | |
| OLD | NEW |