Update mojo sdk to rev cc531b32182099a5a034a99daff35ed5d38a61c8

mojo/edk/system/core_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/edk/system/core.h"
 
 #include <stdint.h>
 
 #include <limits>
 
+#include "base/bind.h"
 #include "base/threading/platform_thread.h"
 #include "base/time/time.h"
+#include "mojo/edk/system/awakable.h"
 #include "mojo/edk/system/core_test_base.h"
 
 namespace mojo {
 namespace system {
 namespace {
 
 const MojoHandleSignalsState kEmptyMojoHandleSignalsState = {0u, 0u};
 const MojoHandleSignalsState kFullMojoHandleSignalsState = {~0u, ~0u};
 const MojoHandleSignals kAllSignals = MOJO_HANDLE_SIGNAL_READABLE |
                                       MOJO_HANDLE_SIGNAL_WRITABLE |
@@ skipping 1246 matching lines @@
   // Complete the two-phase read.
   EXPECT_EQ('x', static_cast<const char*>(read_ptr)[0]);
   EXPECT_EQ(MOJO_RESULT_OK, core()->EndReadData(ch, 1));
 
   EXPECT_EQ(MOJO_RESULT_OK, core()->Close(h_passing[0]));
   EXPECT_EQ(MOJO_RESULT_OK, core()->Close(h_passing[1]));
   EXPECT_EQ(MOJO_RESULT_OK, core()->Close(ph));
   EXPECT_EQ(MOJO_RESULT_OK, core()->Close(ch));
 }
 
+struct TestAsyncWaiter {
+  TestAsyncWaiter() : result(MOJO_RESULT_UNKNOWN) {}
+
+  void Awake(MojoResult r) { result = r; }
+
+  MojoResult result;
+};
+
+TEST_F(CoreTest, AsyncWait) {
+  TestAsyncWaiter waiter;
+  MockHandleInfo info;
+  MojoHandle h = CreateMockHandle(&info);
+
+  EXPECT_EQ(MOJO_RESULT_FAILED_PRECONDITION,
+            core()->AsyncWait(h, MOJO_HANDLE_SIGNAL_READABLE,
+                              base::Bind(&TestAsyncWaiter::Awake,
+                                         base::Unretained(&waiter))));
+  EXPECT_EQ(0u, info.GetAddedAwakableSize());
+
+  info.AllowAddAwakable(true);
+  EXPECT_EQ(MOJO_RESULT_OK,
+            core()->AsyncWait(h, MOJO_HANDLE_SIGNAL_READABLE,
+                              base::Bind(&TestAsyncWaiter::Awake,
+                                         base::Unretained(&waiter))));
+  EXPECT_EQ(1u, info.GetAddedAwakableSize());
+
+  EXPECT_FALSE(info.GetAddedAwakableAt(0)->Awake(MOJO_RESULT_BUSY, 0));
+  EXPECT_EQ(MOJO_RESULT_BUSY, waiter.result);
+
+  EXPECT_EQ(MOJO_RESULT_OK, core()->Close(h));
+}
+
 // TODO(vtl): Test |DuplicateBufferHandle()| and |MapBuffer()|.
 
 }  // namespace
 }  // namespace system
 }  // namespace mojo