Add a mechanism to extend a libevent based message pump.

message_loop/message_pump_libevent_unittest.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "base/message_loop/message_pump_libevent.h"
 
 #include <unistd.h>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
@@ skipping 253 matching lines @@
   loop.PostTask(FROM_HERE, Bind(&WaitableEvent::Signal, Unretained(&event)));
 
   // Now run the MessageLoop.
   run_loop.Run();
 
   // StartWatching can move |watcher| to IO thread. Release on IO thread.
   io_loop()->PostTask(FROM_HERE, Bind(&WaitableEventWatcher::StopWatching,
                                       Owned(watcher.release())));
 }
 
+class FakeEventSource : public MessagePumpLibevent::EventSource {
+ public:
+  FakeEventSource() : work_units_(0u) {}
+  ~FakeEventSource() override {}
+
+  void set_work_units(uint32_t work_units) { work_units_ = work_units; }
+  uint32_t work_units() const { return work_units_; }
+
+  bool Poll() override {
+    if (!work_units_)
+      return false;
+    work_units_--;
+    return true;
+  }
+
+ private:
+  uint32_t work_units_;
+};
+
+TEST_F(MessagePumpLibeventTest, PollEventSource) {
+  FakeEventSource event_source;
+
+  // Verify event source gets polled repeatedly until no work remains.
+  MessageLoopForUI::current()->SetEventSource(&event_source);
+  event_source.set_work_units(5u);
+  ui_loop_->RunUntilIdle();
+  EXPECT_EQ(0u, event_source.work_units());
+
+  // After removing the event source, it should no longer be polled.
+  MessageLoopForUI::current()->ClearEventSource();
+  event_source.set_work_units(5u);
+  ui_loop_->RunUntilIdle();
+  EXPECT_EQ(5u, event_source.work_units());
+}
+
 }  // namespace
 
 }  // namespace base