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

Side by Side Diff: base/message_loop/message_pump_glib_unittest.cc

Issue 16897006: Move message_pump to base/message_loop. (Closed) Base URL: svn://chrome-svn/chrome/trunk/src/
Patch Set: Created 7 years, 6 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 | Annotate | Revision Log
« no previous file with comments | « base/message_loop/message_pump_glib.cc ('k') | base/message_loop/message_pump_gtk.h » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be 2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file. 3 // found in the LICENSE file.
4 4
5 #include "base/message_pump_glib.h" 5 #include "base/message_loop/message_pump_glib.h"
6 6
7 #include <glib.h> 7 #include <glib.h>
8 #include <math.h> 8 #include <math.h>
9 9
10 #include <algorithm> 10 #include <algorithm>
11 #include <vector> 11 #include <vector>
12 12
13 #include "base/bind.h" 13 #include "base/bind.h"
14 #include "base/bind_helpers.h" 14 #include "base/bind_helpers.h"
15 #include "base/callback.h" 15 #include "base/callback.h"
(...skipping 24 matching lines...) Expand all
40 40
41 ~EventInjector() { 41 ~EventInjector() {
42 g_source_destroy(source_); 42 g_source_destroy(source_);
43 g_source_unref(source_); 43 g_source_unref(source_);
44 } 44 }
45 45
46 int HandlePrepare() { 46 int HandlePrepare() {
47 // If the queue is empty, block. 47 // If the queue is empty, block.
48 if (events_.empty()) 48 if (events_.empty())
49 return -1; 49 return -1;
50 base::TimeDelta delta = events_[0].time - base::Time::NowFromSystemTime(); 50 TimeDelta delta = events_[0].time - Time::NowFromSystemTime();
51 return std::max(0, static_cast<int>(ceil(delta.InMillisecondsF()))); 51 return std::max(0, static_cast<int>(ceil(delta.InMillisecondsF())));
52 } 52 }
53 53
54 bool HandleCheck() { 54 bool HandleCheck() {
55 if (events_.empty()) 55 if (events_.empty())
56 return false; 56 return false;
57 return events_[0].time <= base::Time::NowFromSystemTime(); 57 return events_[0].time <= Time::NowFromSystemTime();
58 } 58 }
59 59
60 void HandleDispatch() { 60 void HandleDispatch() {
61 if (events_.empty()) 61 if (events_.empty())
62 return; 62 return;
63 Event event = events_[0]; 63 Event event = events_[0];
64 events_.erase(events_.begin()); 64 events_.erase(events_.begin());
65 ++processed_events_; 65 ++processed_events_;
66 if (!event.callback.is_null()) 66 if (!event.callback.is_null())
67 event.callback.Run(); 67 event.callback.Run();
68 else if (!event.task.is_null()) 68 else if (!event.task.is_null())
69 event.task.Run(); 69 event.task.Run();
70 } 70 }
71 71
72 // Adds an event to the queue. When "handled", executes |callback|. 72 // Adds an event to the queue. When "handled", executes |callback|.
73 // delay_ms is relative to the last event if any, or to Now() otherwise. 73 // delay_ms is relative to the last event if any, or to Now() otherwise.
74 void AddEvent(int delay_ms, const base::Closure& callback) { 74 void AddEvent(int delay_ms, const Closure& callback) {
75 AddEventHelper(delay_ms, callback, base::Closure()); 75 AddEventHelper(delay_ms, callback, Closure());
76 } 76 }
77 77
78 void AddDummyEvent(int delay_ms) { 78 void AddDummyEvent(int delay_ms) {
79 AddEventHelper(delay_ms, base::Closure(), base::Closure()); 79 AddEventHelper(delay_ms, Closure(), Closure());
80 } 80 }
81 81
82 void AddEventAsTask(int delay_ms, const base::Closure& task) { 82 void AddEventAsTask(int delay_ms, const Closure& task) {
83 AddEventHelper(delay_ms, base::Closure(), task); 83 AddEventHelper(delay_ms, Closure(), task);
84 } 84 }
85 85
86 void Reset() { 86 void Reset() {
87 processed_events_ = 0; 87 processed_events_ = 0;
88 events_.clear(); 88 events_.clear();
89 } 89 }
90 90
91 int processed_events() const { return processed_events_; } 91 int processed_events() const { return processed_events_; }
92 92
93 private: 93 private:
94 struct Event { 94 struct Event {
95 base::Time time; 95 Time time;
96 base::Closure callback; 96 Closure callback;
97 base::Closure task; 97 Closure task;
98 }; 98 };
99 99
100 struct Source : public GSource { 100 struct Source : public GSource {
101 EventInjector* injector; 101 EventInjector* injector;
102 }; 102 };
103 103
104 void AddEventHelper( 104 void AddEventHelper(
105 int delay_ms, const base::Closure& callback, const base::Closure& task) { 105 int delay_ms, const Closure& callback, const Closure& task) {
106 base::Time last_time; 106 Time last_time;
107 if (!events_.empty()) 107 if (!events_.empty())
108 last_time = (events_.end()-1)->time; 108 last_time = (events_.end()-1)->time;
109 else 109 else
110 last_time = base::Time::NowFromSystemTime(); 110 last_time = Time::NowFromSystemTime();
111 111
112 base::Time future = last_time + base::TimeDelta::FromMilliseconds(delay_ms); 112 Time future = last_time + TimeDelta::FromMilliseconds(delay_ms);
113 EventInjector::Event event = {future, callback, task}; 113 EventInjector::Event event = {future, callback, task};
114 events_.push_back(event); 114 events_.push_back(event);
115 } 115 }
116 116
117 static gboolean Prepare(GSource* source, gint* timeout_ms) { 117 static gboolean Prepare(GSource* source, gint* timeout_ms) {
118 *timeout_ms = static_cast<Source*>(source)->injector->HandlePrepare(); 118 *timeout_ms = static_cast<Source*>(source)->injector->HandlePrepare();
119 return FALSE; 119 return FALSE;
120 } 120 }
121 121
122 static gboolean Check(GSource* source) { 122 static gboolean Check(GSource* source) {
(...skipping 25 matching lines...) Expand all
148 ++*value; 148 ++*value;
149 } 149 }
150 150
151 // Checks how many events have been processed by the injector. 151 // Checks how many events have been processed by the injector.
152 void ExpectProcessedEvents(EventInjector* injector, int count) { 152 void ExpectProcessedEvents(EventInjector* injector, int count) {
153 EXPECT_EQ(injector->processed_events(), count); 153 EXPECT_EQ(injector->processed_events(), count);
154 } 154 }
155 155
156 // Posts a task on the current message loop. 156 // Posts a task on the current message loop.
157 void PostMessageLoopTask(const tracked_objects::Location& from_here, 157 void PostMessageLoopTask(const tracked_objects::Location& from_here,
158 const base::Closure& task) { 158 const Closure& task) {
159 MessageLoop::current()->PostTask(from_here, task); 159 MessageLoop::current()->PostTask(from_here, task);
160 } 160 }
161 161
162 // Test fixture. 162 // Test fixture.
163 class MessagePumpGLibTest : public testing::Test { 163 class MessagePumpGLibTest : public testing::Test {
164 public: 164 public:
165 MessagePumpGLibTest() : loop_(NULL), injector_(NULL) { } 165 MessagePumpGLibTest() : loop_(NULL), injector_(NULL) { }
166 166
167 // Overridden from testing::Test: 167 // Overridden from testing::Test:
168 virtual void SetUp() OVERRIDE { 168 virtual void SetUp() OVERRIDE {
(...skipping 33 matching lines...) Expand 10 before | Expand all | Expand 10 after
202 } 202 }
203 203
204 TEST_F(MessagePumpGLibTest, TestEventTaskInterleave) { 204 TEST_F(MessagePumpGLibTest, TestEventTaskInterleave) {
205 // Checks that tasks posted by events are executed before the next event if 205 // Checks that tasks posted by events are executed before the next event if
206 // the posted task queue is empty. 206 // the posted task queue is empty.
207 // MessageLoop doesn't make strong guarantees that it is the case, but the 207 // MessageLoop doesn't make strong guarantees that it is the case, but the
208 // current implementation ensures it and the tests below rely on it. 208 // current implementation ensures it and the tests below rely on it.
209 // If changes cause this test to fail, it is reasonable to change it, but 209 // If changes cause this test to fail, it is reasonable to change it, but
210 // TestWorkWhileWaitingForEvents and TestEventsWhileWaitingForWork have to be 210 // TestWorkWhileWaitingForEvents and TestEventsWhileWaitingForWork have to be
211 // changed accordingly, otherwise they can become flaky. 211 // changed accordingly, otherwise they can become flaky.
212 injector()->AddEventAsTask(0, base::Bind(&base::DoNothing)); 212 injector()->AddEventAsTask(0, Bind(&DoNothing));
213 base::Closure check_task = 213 Closure check_task =
214 base::Bind(&ExpectProcessedEvents, base::Unretained(injector()), 2); 214 Bind(&ExpectProcessedEvents, Unretained(injector()), 2);
215 base::Closure posted_task = 215 Closure posted_task =
216 base::Bind(&PostMessageLoopTask, FROM_HERE, check_task); 216 Bind(&PostMessageLoopTask, FROM_HERE, check_task);
217 injector()->AddEventAsTask(0, posted_task); 217 injector()->AddEventAsTask(0, posted_task);
218 injector()->AddEventAsTask(0, base::Bind(&base::DoNothing)); 218 injector()->AddEventAsTask(0, Bind(&DoNothing));
219 injector()->AddEvent(0, MessageLoop::QuitWhenIdleClosure()); 219 injector()->AddEvent(0, MessageLoop::QuitWhenIdleClosure());
220 loop()->Run(); 220 loop()->Run();
221 EXPECT_EQ(4, injector()->processed_events()); 221 EXPECT_EQ(4, injector()->processed_events());
222 222
223 injector()->Reset(); 223 injector()->Reset();
224 injector()->AddEventAsTask(0, base::Bind(&base::DoNothing)); 224 injector()->AddEventAsTask(0, Bind(&DoNothing));
225 check_task = 225 check_task =
226 base::Bind(&ExpectProcessedEvents, base::Unretained(injector()), 2); 226 Bind(&ExpectProcessedEvents, Unretained(injector()), 2);
227 posted_task = base::Bind(&PostMessageLoopTask, FROM_HERE, check_task); 227 posted_task = Bind(&PostMessageLoopTask, FROM_HERE, check_task);
228 injector()->AddEventAsTask(0, posted_task); 228 injector()->AddEventAsTask(0, posted_task);
229 injector()->AddEventAsTask(10, base::Bind(&base::DoNothing)); 229 injector()->AddEventAsTask(10, Bind(&DoNothing));
230 injector()->AddEvent(0, MessageLoop::QuitWhenIdleClosure()); 230 injector()->AddEvent(0, MessageLoop::QuitWhenIdleClosure());
231 loop()->Run(); 231 loop()->Run();
232 EXPECT_EQ(4, injector()->processed_events()); 232 EXPECT_EQ(4, injector()->processed_events());
233 } 233 }
234 234
235 TEST_F(MessagePumpGLibTest, TestWorkWhileWaitingForEvents) { 235 TEST_F(MessagePumpGLibTest, TestWorkWhileWaitingForEvents) {
236 int task_count = 0; 236 int task_count = 0;
237 // Tests that we process tasks while waiting for new events. 237 // Tests that we process tasks while waiting for new events.
238 // The event queue is empty at first. 238 // The event queue is empty at first.
239 for (int i = 0; i < 10; ++i) { 239 for (int i = 0; i < 10; ++i) {
240 loop()->PostTask(FROM_HERE, base::Bind(&IncrementInt, &task_count)); 240 loop()->PostTask(FROM_HERE, Bind(&IncrementInt, &task_count));
241 } 241 }
242 // After all the previous tasks have executed, enqueue an event that will 242 // After all the previous tasks have executed, enqueue an event that will
243 // quit. 243 // quit.
244 loop()->PostTask( 244 loop()->PostTask(
245 FROM_HERE, 245 FROM_HERE,
246 base::Bind(&EventInjector::AddEvent, base::Unretained(injector()), 0, 246 Bind(&EventInjector::AddEvent, Unretained(injector()), 0,
247 MessageLoop::QuitWhenIdleClosure())); 247 MessageLoop::QuitWhenIdleClosure()));
248 loop()->Run(); 248 loop()->Run();
249 ASSERT_EQ(10, task_count); 249 ASSERT_EQ(10, task_count);
250 EXPECT_EQ(1, injector()->processed_events()); 250 EXPECT_EQ(1, injector()->processed_events());
251 251
252 // Tests that we process delayed tasks while waiting for new events. 252 // Tests that we process delayed tasks while waiting for new events.
253 injector()->Reset(); 253 injector()->Reset();
254 task_count = 0; 254 task_count = 0;
255 for (int i = 0; i < 10; ++i) { 255 for (int i = 0; i < 10; ++i) {
256 loop()->PostDelayedTask( 256 loop()->PostDelayedTask(
257 FROM_HERE, 257 FROM_HERE,
258 base::Bind(&IncrementInt, &task_count), 258 Bind(&IncrementInt, &task_count),
259 base::TimeDelta::FromMilliseconds(10*i)); 259 TimeDelta::FromMilliseconds(10*i));
260 } 260 }
261 // After all the previous tasks have executed, enqueue an event that will 261 // After all the previous tasks have executed, enqueue an event that will
262 // quit. 262 // quit.
263 // This relies on the fact that delayed tasks are executed in delay order. 263 // This relies on the fact that delayed tasks are executed in delay order.
264 // That is verified in message_loop_unittest.cc. 264 // That is verified in message_loop_unittest.cc.
265 loop()->PostDelayedTask( 265 loop()->PostDelayedTask(
266 FROM_HERE, 266 FROM_HERE,
267 base::Bind(&EventInjector::AddEvent, base::Unretained(injector()), 10, 267 Bind(&EventInjector::AddEvent, Unretained(injector()), 10,
268 MessageLoop::QuitWhenIdleClosure()), 268 MessageLoop::QuitWhenIdleClosure()),
269 base::TimeDelta::FromMilliseconds(150)); 269 TimeDelta::FromMilliseconds(150));
270 loop()->Run(); 270 loop()->Run();
271 ASSERT_EQ(10, task_count); 271 ASSERT_EQ(10, task_count);
272 EXPECT_EQ(1, injector()->processed_events()); 272 EXPECT_EQ(1, injector()->processed_events());
273 } 273 }
274 274
275 TEST_F(MessagePumpGLibTest, TestEventsWhileWaitingForWork) { 275 TEST_F(MessagePumpGLibTest, TestEventsWhileWaitingForWork) {
276 // Tests that we process events while waiting for work. 276 // Tests that we process events while waiting for work.
277 // The event queue is empty at first. 277 // The event queue is empty at first.
278 for (int i = 0; i < 10; ++i) { 278 for (int i = 0; i < 10; ++i) {
279 injector()->AddDummyEvent(0); 279 injector()->AddDummyEvent(0);
280 } 280 }
281 // After all the events have been processed, post a task that will check that 281 // After all the events have been processed, post a task that will check that
282 // the events have been processed (note: the task executes after the event 282 // the events have been processed (note: the task executes after the event
283 // that posted it has been handled, so we expect 11 at that point). 283 // that posted it has been handled, so we expect 11 at that point).
284 base::Closure check_task = 284 Closure check_task =
285 base::Bind(&ExpectProcessedEvents, base::Unretained(injector()), 11); 285 Bind(&ExpectProcessedEvents, Unretained(injector()), 11);
286 base::Closure posted_task = 286 Closure posted_task =
287 base::Bind(&PostMessageLoopTask, FROM_HERE, check_task); 287 Bind(&PostMessageLoopTask, FROM_HERE, check_task);
288 injector()->AddEventAsTask(10, posted_task); 288 injector()->AddEventAsTask(10, posted_task);
289 289
290 // And then quit (relies on the condition tested by TestEventTaskInterleave). 290 // And then quit (relies on the condition tested by TestEventTaskInterleave).
291 injector()->AddEvent(10, MessageLoop::QuitWhenIdleClosure()); 291 injector()->AddEvent(10, MessageLoop::QuitWhenIdleClosure());
292 loop()->Run(); 292 loop()->Run();
293 293
294 EXPECT_EQ(12, injector()->processed_events()); 294 EXPECT_EQ(12, injector()->processed_events());
295 } 295 }
296 296
297 namespace { 297 namespace {
298 298
299 // This class is a helper for the concurrent events / posted tasks test below. 299 // This class is a helper for the concurrent events / posted tasks test below.
300 // It will quit the main loop once enough tasks and events have been processed, 300 // It will quit the main loop once enough tasks and events have been processed,
301 // while making sure there is always work to do and events in the queue. 301 // while making sure there is always work to do and events in the queue.
302 class ConcurrentHelper : public base::RefCounted<ConcurrentHelper> { 302 class ConcurrentHelper : public RefCounted<ConcurrentHelper> {
303 public: 303 public:
304 explicit ConcurrentHelper(EventInjector* injector) 304 explicit ConcurrentHelper(EventInjector* injector)
305 : injector_(injector), 305 : injector_(injector),
306 event_count_(kStartingEventCount), 306 event_count_(kStartingEventCount),
307 task_count_(kStartingTaskCount) { 307 task_count_(kStartingTaskCount) {
308 } 308 }
309 309
310 void FromTask() { 310 void FromTask() {
311 if (task_count_ > 0) { 311 if (task_count_ > 0) {
312 --task_count_; 312 --task_count_;
313 } 313 }
314 if (task_count_ == 0 && event_count_ == 0) { 314 if (task_count_ == 0 && event_count_ == 0) {
315 MessageLoop::current()->QuitWhenIdle(); 315 MessageLoop::current()->QuitWhenIdle();
316 } else { 316 } else {
317 MessageLoop::current()->PostTask( 317 MessageLoop::current()->PostTask(
318 FROM_HERE, base::Bind(&ConcurrentHelper::FromTask, this)); 318 FROM_HERE, Bind(&ConcurrentHelper::FromTask, this));
319 } 319 }
320 } 320 }
321 321
322 void FromEvent() { 322 void FromEvent() {
323 if (event_count_ > 0) { 323 if (event_count_ > 0) {
324 --event_count_; 324 --event_count_;
325 } 325 }
326 if (task_count_ == 0 && event_count_ == 0) { 326 if (task_count_ == 0 && event_count_ == 0) {
327 MessageLoop::current()->QuitWhenIdle(); 327 MessageLoop::current()->QuitWhenIdle();
328 } else { 328 } else {
329 injector_->AddEventAsTask( 329 injector_->AddEventAsTask(
330 0, base::Bind(&ConcurrentHelper::FromEvent, this)); 330 0, Bind(&ConcurrentHelper::FromEvent, this));
331 } 331 }
332 } 332 }
333 333
334 int event_count() const { return event_count_; } 334 int event_count() const { return event_count_; }
335 int task_count() const { return task_count_; } 335 int task_count() const { return task_count_; }
336 336
337 private: 337 private:
338 friend class base::RefCounted<ConcurrentHelper>; 338 friend class RefCounted<ConcurrentHelper>;
339 339
340 ~ConcurrentHelper() {} 340 ~ConcurrentHelper() {}
341 341
342 static const int kStartingEventCount = 20; 342 static const int kStartingEventCount = 20;
343 static const int kStartingTaskCount = 20; 343 static const int kStartingTaskCount = 20;
344 344
345 EventInjector* injector_; 345 EventInjector* injector_;
346 int event_count_; 346 int event_count_;
347 int task_count_; 347 int task_count_;
348 }; 348 };
349 349
350 } // namespace 350 } // namespace
351 351
352 TEST_F(MessagePumpGLibTest, TestConcurrentEventPostedTask) { 352 TEST_F(MessagePumpGLibTest, TestConcurrentEventPostedTask) {
353 // Tests that posted tasks don't starve events, nor the opposite. 353 // Tests that posted tasks don't starve events, nor the opposite.
354 // We use the helper class above. We keep both event and posted task queues 354 // We use the helper class above. We keep both event and posted task queues
355 // full, the helper verifies that both tasks and events get processed. 355 // full, the helper verifies that both tasks and events get processed.
356 // If that is not the case, either event_count_ or task_count_ will not get 356 // If that is not the case, either event_count_ or task_count_ will not get
357 // to 0, and MessageLoop::QuitWhenIdle() will never be called. 357 // to 0, and MessageLoop::QuitWhenIdle() will never be called.
358 scoped_refptr<ConcurrentHelper> helper = new ConcurrentHelper(injector()); 358 scoped_refptr<ConcurrentHelper> helper = new ConcurrentHelper(injector());
359 359
360 // Add 2 events to the queue to make sure it is always full (when we remove 360 // Add 2 events to the queue to make sure it is always full (when we remove
361 // the event before processing it). 361 // the event before processing it).
362 injector()->AddEventAsTask( 362 injector()->AddEventAsTask(
363 0, base::Bind(&ConcurrentHelper::FromEvent, helper.get())); 363 0, Bind(&ConcurrentHelper::FromEvent, helper.get()));
364 injector()->AddEventAsTask( 364 injector()->AddEventAsTask(
365 0, base::Bind(&ConcurrentHelper::FromEvent, helper.get())); 365 0, Bind(&ConcurrentHelper::FromEvent, helper.get()));
366 366
367 // Similarly post 2 tasks. 367 // Similarly post 2 tasks.
368 loop()->PostTask( 368 loop()->PostTask(
369 FROM_HERE, base::Bind(&ConcurrentHelper::FromTask, helper.get())); 369 FROM_HERE, Bind(&ConcurrentHelper::FromTask, helper.get()));
370 loop()->PostTask( 370 loop()->PostTask(
371 FROM_HERE, base::Bind(&ConcurrentHelper::FromTask, helper.get())); 371 FROM_HERE, Bind(&ConcurrentHelper::FromTask, helper.get()));
372 372
373 loop()->Run(); 373 loop()->Run();
374 EXPECT_EQ(0, helper->event_count()); 374 EXPECT_EQ(0, helper->event_count());
375 EXPECT_EQ(0, helper->task_count()); 375 EXPECT_EQ(0, helper->task_count());
376 } 376 }
377 377
378 namespace { 378 namespace {
379 379
380 void AddEventsAndDrainGLib(EventInjector* injector) { 380 void AddEventsAndDrainGLib(EventInjector* injector) {
381 // Add a couple of dummy events 381 // Add a couple of dummy events
382 injector->AddDummyEvent(0); 382 injector->AddDummyEvent(0);
383 injector->AddDummyEvent(0); 383 injector->AddDummyEvent(0);
384 // Then add an event that will quit the main loop. 384 // Then add an event that will quit the main loop.
385 injector->AddEvent(0, MessageLoop::QuitWhenIdleClosure()); 385 injector->AddEvent(0, MessageLoop::QuitWhenIdleClosure());
386 386
387 // Post a couple of dummy tasks 387 // Post a couple of dummy tasks
388 MessageLoop::current()->PostTask(FROM_HERE, base::Bind(&base::DoNothing)); 388 MessageLoop::current()->PostTask(FROM_HERE, Bind(&DoNothing));
389 MessageLoop::current()->PostTask(FROM_HERE, base::Bind(&base::DoNothing)); 389 MessageLoop::current()->PostTask(FROM_HERE, Bind(&DoNothing));
390 390
391 // Drain the events 391 // Drain the events
392 while (g_main_context_pending(NULL)) { 392 while (g_main_context_pending(NULL)) {
393 g_main_context_iteration(NULL, FALSE); 393 g_main_context_iteration(NULL, FALSE);
394 } 394 }
395 } 395 }
396 396
397 } // namespace 397 } // namespace
398 398
399 TEST_F(MessagePumpGLibTest, TestDrainingGLib) { 399 TEST_F(MessagePumpGLibTest, TestDrainingGLib) {
400 // Tests that draining events using GLib works. 400 // Tests that draining events using GLib works.
401 loop()->PostTask( 401 loop()->PostTask(
402 FROM_HERE, 402 FROM_HERE,
403 base::Bind(&AddEventsAndDrainGLib, base::Unretained(injector()))); 403 Bind(&AddEventsAndDrainGLib, Unretained(injector())));
404 loop()->Run(); 404 loop()->Run();
405 405
406 EXPECT_EQ(3, injector()->processed_events()); 406 EXPECT_EQ(3, injector()->processed_events());
407 } 407 }
408 408
409 409
410 namespace { 410 namespace {
411 411
412 #if defined(TOOLKIT_GTK) 412 #if defined(TOOLKIT_GTK)
413 void AddEventsAndDrainGtk(EventInjector* injector) { 413 void AddEventsAndDrainGtk(EventInjector* injector) {
414 // Add a couple of dummy events 414 // Add a couple of dummy events
415 injector->AddDummyEvent(0); 415 injector->AddDummyEvent(0);
416 injector->AddDummyEvent(0); 416 injector->AddDummyEvent(0);
417 // Then add an event that will quit the main loop. 417 // Then add an event that will quit the main loop.
418 injector->AddEvent(0, MessageLoop::QuitWhenIdleClosure()); 418 injector->AddEvent(0, MessageLoop::QuitWhenIdleClosure());
419 419
420 // Post a couple of dummy tasks 420 // Post a couple of dummy tasks
421 MessageLoop::current()->PostTask(FROM_HERE, base::Bind(&base::DoNothing)); 421 MessageLoop::current()->PostTask(FROM_HERE, Bind(&DoNothing));
422 MessageLoop::current()->PostTask(FROM_HERE, base::Bind(&base::DoNothing)); 422 MessageLoop::current()->PostTask(FROM_HERE, Bind(&DoNothing));
423 423
424 // Drain the events 424 // Drain the events
425 while (gtk_events_pending()) { 425 while (gtk_events_pending()) {
426 gtk_main_iteration(); 426 gtk_main_iteration();
427 } 427 }
428 } 428 }
429 #endif 429 #endif
430 430
431 } // namespace 431 } // namespace
432 432
433 #if defined(TOOLKIT_GTK) 433 #if defined(TOOLKIT_GTK)
434 TEST_F(MessagePumpGLibTest, TestDrainingGtk) { 434 TEST_F(MessagePumpGLibTest, TestDrainingGtk) {
435 // Tests that draining events using Gtk works. 435 // Tests that draining events using Gtk works.
436 loop()->PostTask( 436 loop()->PostTask(
437 FROM_HERE, 437 FROM_HERE,
438 base::Bind(&AddEventsAndDrainGtk, base::Unretained(injector()))); 438 Bind(&AddEventsAndDrainGtk, Unretained(injector())));
439 loop()->Run(); 439 loop()->Run();
440 440
441 EXPECT_EQ(3, injector()->processed_events()); 441 EXPECT_EQ(3, injector()->processed_events());
442 } 442 }
443 #endif 443 #endif
444 444
445 namespace { 445 namespace {
446 446
447 // Helper class that lets us run the GLib message loop. 447 // Helper class that lets us run the GLib message loop.
448 class GLibLoopRunner : public base::RefCounted<GLibLoopRunner> { 448 class GLibLoopRunner : public RefCounted<GLibLoopRunner> {
449 public: 449 public:
450 GLibLoopRunner() : quit_(false) { } 450 GLibLoopRunner() : quit_(false) { }
451 451
452 void RunGLib() { 452 void RunGLib() {
453 while (!quit_) { 453 while (!quit_) {
454 g_main_context_iteration(NULL, TRUE); 454 g_main_context_iteration(NULL, TRUE);
455 } 455 }
456 } 456 }
457 457
458 void RunLoop() { 458 void RunLoop() {
(...skipping 10 matching lines...) Expand all
469 469
470 void Quit() { 470 void Quit() {
471 quit_ = true; 471 quit_ = true;
472 } 472 }
473 473
474 void Reset() { 474 void Reset() {
475 quit_ = false; 475 quit_ = false;
476 } 476 }
477 477
478 private: 478 private:
479 friend class base::RefCounted<GLibLoopRunner>; 479 friend class RefCounted<GLibLoopRunner>;
480 480
481 ~GLibLoopRunner() {} 481 ~GLibLoopRunner() {}
482 482
483 bool quit_; 483 bool quit_;
484 }; 484 };
485 485
486 void TestGLibLoopInternal(EventInjector* injector) { 486 void TestGLibLoopInternal(EventInjector* injector) {
487 // Allow tasks to be processed from 'native' event loops. 487 // Allow tasks to be processed from 'native' event loops.
488 MessageLoop::current()->SetNestableTasksAllowed(true); 488 MessageLoop::current()->SetNestableTasksAllowed(true);
489 scoped_refptr<GLibLoopRunner> runner = new GLibLoopRunner(); 489 scoped_refptr<GLibLoopRunner> runner = new GLibLoopRunner();
490 490
491 int task_count = 0; 491 int task_count = 0;
492 // Add a couple of dummy events 492 // Add a couple of dummy events
493 injector->AddDummyEvent(0); 493 injector->AddDummyEvent(0);
494 injector->AddDummyEvent(0); 494 injector->AddDummyEvent(0);
495 // Post a couple of dummy tasks 495 // Post a couple of dummy tasks
496 MessageLoop::current()->PostTask( 496 MessageLoop::current()->PostTask(
497 FROM_HERE, base::Bind(&IncrementInt, &task_count)); 497 FROM_HERE, Bind(&IncrementInt, &task_count));
498 MessageLoop::current()->PostTask( 498 MessageLoop::current()->PostTask(
499 FROM_HERE, base::Bind(&IncrementInt, &task_count)); 499 FROM_HERE, Bind(&IncrementInt, &task_count));
500 // Delayed events 500 // Delayed events
501 injector->AddDummyEvent(10); 501 injector->AddDummyEvent(10);
502 injector->AddDummyEvent(10); 502 injector->AddDummyEvent(10);
503 // Delayed work 503 // Delayed work
504 MessageLoop::current()->PostDelayedTask( 504 MessageLoop::current()->PostDelayedTask(
505 FROM_HERE, 505 FROM_HERE,
506 base::Bind(&IncrementInt, &task_count), 506 Bind(&IncrementInt, &task_count),
507 base::TimeDelta::FromMilliseconds(30)); 507 TimeDelta::FromMilliseconds(30));
508 MessageLoop::current()->PostDelayedTask( 508 MessageLoop::current()->PostDelayedTask(
509 FROM_HERE, 509 FROM_HERE,
510 base::Bind(&GLibLoopRunner::Quit, runner.get()), 510 Bind(&GLibLoopRunner::Quit, runner.get()),
511 base::TimeDelta::FromMilliseconds(40)); 511 TimeDelta::FromMilliseconds(40));
512 512
513 // Run a nested, straight GLib message loop. 513 // Run a nested, straight GLib message loop.
514 runner->RunGLib(); 514 runner->RunGLib();
515 515
516 ASSERT_EQ(3, task_count); 516 ASSERT_EQ(3, task_count);
517 EXPECT_EQ(4, injector->processed_events()); 517 EXPECT_EQ(4, injector->processed_events());
518 MessageLoop::current()->QuitWhenIdle(); 518 MessageLoop::current()->QuitWhenIdle();
519 } 519 }
520 520
521 void TestGtkLoopInternal(EventInjector* injector) { 521 void TestGtkLoopInternal(EventInjector* injector) {
522 // Allow tasks to be processed from 'native' event loops. 522 // Allow tasks to be processed from 'native' event loops.
523 MessageLoop::current()->SetNestableTasksAllowed(true); 523 MessageLoop::current()->SetNestableTasksAllowed(true);
524 scoped_refptr<GLibLoopRunner> runner = new GLibLoopRunner(); 524 scoped_refptr<GLibLoopRunner> runner = new GLibLoopRunner();
525 525
526 int task_count = 0; 526 int task_count = 0;
527 // Add a couple of dummy events 527 // Add a couple of dummy events
528 injector->AddDummyEvent(0); 528 injector->AddDummyEvent(0);
529 injector->AddDummyEvent(0); 529 injector->AddDummyEvent(0);
530 // Post a couple of dummy tasks 530 // Post a couple of dummy tasks
531 MessageLoop::current()->PostTask( 531 MessageLoop::current()->PostTask(
532 FROM_HERE, base::Bind(&IncrementInt, &task_count)); 532 FROM_HERE, Bind(&IncrementInt, &task_count));
533 MessageLoop::current()->PostTask( 533 MessageLoop::current()->PostTask(
534 FROM_HERE, base::Bind(&IncrementInt, &task_count)); 534 FROM_HERE, Bind(&IncrementInt, &task_count));
535 // Delayed events 535 // Delayed events
536 injector->AddDummyEvent(10); 536 injector->AddDummyEvent(10);
537 injector->AddDummyEvent(10); 537 injector->AddDummyEvent(10);
538 // Delayed work 538 // Delayed work
539 MessageLoop::current()->PostDelayedTask( 539 MessageLoop::current()->PostDelayedTask(
540 FROM_HERE, 540 FROM_HERE,
541 base::Bind(&IncrementInt, &task_count), 541 Bind(&IncrementInt, &task_count),
542 base::TimeDelta::FromMilliseconds(30)); 542 TimeDelta::FromMilliseconds(30));
543 MessageLoop::current()->PostDelayedTask( 543 MessageLoop::current()->PostDelayedTask(
544 FROM_HERE, 544 FROM_HERE,
545 base::Bind(&GLibLoopRunner::Quit, runner.get()), 545 Bind(&GLibLoopRunner::Quit, runner.get()),
546 base::TimeDelta::FromMilliseconds(40)); 546 TimeDelta::FromMilliseconds(40));
547 547
548 // Run a nested, straight Gtk message loop. 548 // Run a nested, straight Gtk message loop.
549 runner->RunLoop(); 549 runner->RunLoop();
550 550
551 ASSERT_EQ(3, task_count); 551 ASSERT_EQ(3, task_count);
552 EXPECT_EQ(4, injector->processed_events()); 552 EXPECT_EQ(4, injector->processed_events());
553 MessageLoop::current()->QuitWhenIdle(); 553 MessageLoop::current()->QuitWhenIdle();
554 } 554 }
555 555
556 } // namespace 556 } // namespace
557 557
558 TEST_F(MessagePumpGLibTest, TestGLibLoop) { 558 TEST_F(MessagePumpGLibTest, TestGLibLoop) {
559 // Tests that events and posted tasks are correctly executed if the message 559 // Tests that events and posted tasks are correctly executed if the message
560 // loop is not run by MessageLoop::Run() but by a straight GLib loop. 560 // loop is not run by MessageLoop::Run() but by a straight GLib loop.
561 // Note that in this case we don't make strong guarantees about niceness 561 // Note that in this case we don't make strong guarantees about niceness
562 // between events and posted tasks. 562 // between events and posted tasks.
563 loop()->PostTask( 563 loop()->PostTask(
564 FROM_HERE, 564 FROM_HERE,
565 base::Bind(&TestGLibLoopInternal, base::Unretained(injector()))); 565 Bind(&TestGLibLoopInternal, Unretained(injector())));
566 loop()->Run(); 566 loop()->Run();
567 } 567 }
568 568
569 TEST_F(MessagePumpGLibTest, TestGtkLoop) { 569 TEST_F(MessagePumpGLibTest, TestGtkLoop) {
570 // Tests that events and posted tasks are correctly executed if the message 570 // Tests that events and posted tasks are correctly executed if the message
571 // loop is not run by MessageLoop::Run() but by a straight Gtk loop. 571 // loop is not run by MessageLoop::Run() but by a straight Gtk loop.
572 // Note that in this case we don't make strong guarantees about niceness 572 // Note that in this case we don't make strong guarantees about niceness
573 // between events and posted tasks. 573 // between events and posted tasks.
574 loop()->PostTask( 574 loop()->PostTask(
575 FROM_HERE, 575 FROM_HERE,
576 base::Bind(&TestGtkLoopInternal, base::Unretained(injector()))); 576 Bind(&TestGtkLoopInternal, Unretained(injector())));
577 loop()->Run(); 577 loop()->Run();
578 } 578 }
579 579
580 } // namespace base 580 } // namespace base
OLDNEW
« no previous file with comments | « base/message_loop/message_pump_glib.cc ('k') | base/message_loop/message_pump_gtk.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698