Implement MessagePumpMojo using WaitSet.

mojo/message_pump/message_pump_mojo_unittest.cc

 // Copyright 2013 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 "mojo/message_pump/message_pump_mojo.h"
 
 #include "base/message_loop/message_loop_test.h"
 #include "base/run_loop.h"
 #include "mojo/message_pump/message_pump_mojo_handler.h"
 #include "mojo/public/cpp/system/core.h"
@@ skipping 101 matching lines @@
       handles.handle0.get(),
       MOJO_HANDLE_SIGNAL_READABLE,
       base::TimeTicks::Now() - base::TimeDelta::FromSeconds(1));
   for (int i = 0; i < 2; ++i) {
     base::RunLoop run_loop;
     run_loop.RunUntilIdle();
   }
   EXPECT_EQ(1, handler.error_count());
 }
 
+TEST(MessagePumpMojo, AddClosedHandle) {
+  base::MessageLoop message_loop(MessagePumpMojo::Create());
+  CountingMojoHandler handler;
+  MessagePipe handles;
+  Handle closed_handle = handles.handle0.get();
+  handles.handle0.reset();
+  MessagePumpMojo::current()->AddHandler(
+      &handler,
+      closed_handle,
+      MOJO_HANDLE_SIGNAL_READABLE,
+      base::TimeTicks());
+  base::RunLoop run_loop;
+  run_loop.RunUntilIdle();
+  MessagePumpMojo::current()->RemoveHandler(closed_handle);
+  EXPECT_EQ(0, handler.error_count());
+  EXPECT_EQ(0, handler.success_count());
+}
+
+TEST(MessagePumpMojo, CloseAfterAdding) {
+  base::MessageLoop message_loop(MessagePumpMojo::Create());
+  CountingMojoHandler handler;
+  MessagePipe handles;
+  MessagePumpMojo::current()->AddHandler(
+      &handler,
+      handles.handle0.get(),
+      MOJO_HANDLE_SIGNAL_READABLE,
+      base::TimeTicks());
+  handles.handle0.reset();
+  base::RunLoop run_loop;
+  run_loop.RunUntilIdle();
+  EXPECT_EQ(1, handler.error_count());
+  EXPECT_EQ(0, handler.success_count());
+}
+
 }  // namespace test
 }  // namespace common
 }  // namespace mojo