[mojo-edk] Add MojoWatch and MojoCancelWatch APIs

mojo/edk/system/shared_buffer_dispatcher_unittest.cc

 // Copyright 2014 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/shared_buffer_dispatcher.h"
 
 #include <stddef.h>
 #include <stdint.h>
 
 #include <limits>
@@ skipping 118 matching lines @@
   static_cast<char*>(mapping1->GetBase())[50] = 'x';
 
   scoped_ptr<PlatformSharedBufferMapping> mapping2;
   EXPECT_EQ(MOJO_RESULT_OK, dispatcher->MapBuffer(
                                 50, 50, MOJO_MAP_BUFFER_FLAG_NONE, &mapping2));
   ASSERT_TRUE(mapping2);
   ASSERT_TRUE(mapping2->GetBase());
   EXPECT_EQ(50u, mapping2->GetLength());
   EXPECT_EQ('x', static_cast<char*>(mapping2->GetBase())[0]);
 
-  EXPECT_EQ(MOJO_RESULT_OK, dispatcher->Close());
+  Dispatcher::RequestContext request_context;
+  EXPECT_EQ(MOJO_RESULT_OK, dispatcher->Close(&request_context));
 
   // Check that we can still read/write to mappings after the dispatcher has
   // gone away.
   static_cast<char*>(mapping2->GetBase())[1] = 'y';
   EXPECT_EQ('y', static_cast<char*>(mapping1->GetBase())[51]);
 }
 
 TEST_F(SharedBufferDispatcherTest, CreateAndMapBufferFromPlatformBuffer) {
   scoped_refptr<PlatformSharedBuffer> platform_shared_buffer =
       PlatformSharedBuffer::Create(100);
@@ skipping 16 matching lines @@
   static_cast<char*>(mapping1->GetBase())[50] = 'x';
 
   scoped_ptr<PlatformSharedBufferMapping> mapping2;
   EXPECT_EQ(MOJO_RESULT_OK, dispatcher->MapBuffer(
                                 50, 50, MOJO_MAP_BUFFER_FLAG_NONE, &mapping2));
   ASSERT_TRUE(mapping2);
   ASSERT_TRUE(mapping2->GetBase());
   EXPECT_EQ(50u, mapping2->GetLength());
   EXPECT_EQ('x', static_cast<char*>(mapping2->GetBase())[0]);
 
-  EXPECT_EQ(MOJO_RESULT_OK, dispatcher->Close());
+  Dispatcher::RequestContext request_context;
+  EXPECT_EQ(MOJO_RESULT_OK, dispatcher->Close(&request_context));
 
   // Check that we can still read/write to mappings after the dispatcher has
   // gone away.
   static_cast<char*>(mapping2->GetBase())[1] = 'y';
   EXPECT_EQ('y', static_cast<char*>(mapping1->GetBase())[51]);
 }
 
 TEST_F(SharedBufferDispatcherTest, DuplicateBufferHandle) {
   scoped_refptr<SharedBufferDispatcher> dispatcher1;
   EXPECT_EQ(MOJO_RESULT_OK, SharedBufferDispatcher::Create(
                                 SharedBufferDispatcher::kDefaultCreateOptions,
                                 100, &dispatcher1));
 
   // Map and write something.
   scoped_ptr<PlatformSharedBufferMapping> mapping;
   EXPECT_EQ(MOJO_RESULT_OK, dispatcher1->MapBuffer(
                                 0, 100, MOJO_MAP_BUFFER_FLAG_NONE, &mapping));
   static_cast<char*>(mapping->GetBase())[0] = 'x';
   mapping.reset();
 
   // Duplicate |dispatcher1| and then close it.
   scoped_refptr<Dispatcher> dispatcher2;
   EXPECT_EQ(MOJO_RESULT_OK, dispatcher1->DuplicateBufferHandle(
                                 nullptr, &dispatcher2));
   ASSERT_TRUE(dispatcher2);
   EXPECT_EQ(Dispatcher::Type::SHARED_BUFFER, dispatcher2->GetType());
 
-  EXPECT_EQ(MOJO_RESULT_OK, dispatcher1->Close());
+  Dispatcher::RequestContext request_context;
+  EXPECT_EQ(MOJO_RESULT_OK, dispatcher1->Close(&request_context));
 
   // Map |dispatcher2| and read something.
   EXPECT_EQ(MOJO_RESULT_OK, dispatcher2->MapBuffer(
                                 0, 100, MOJO_MAP_BUFFER_FLAG_NONE, &mapping));
   EXPECT_EQ('x', static_cast<char*>(mapping->GetBase())[0]);
 
-  EXPECT_EQ(MOJO_RESULT_OK, dispatcher2->Close());
+  EXPECT_EQ(MOJO_RESULT_OK, dispatcher2->Close(&request_context));
 }
 
 TEST_F(SharedBufferDispatcherTest, DuplicateBufferHandleOptionsValid) {
   scoped_refptr<SharedBufferDispatcher> dispatcher1;
   EXPECT_EQ(MOJO_RESULT_OK, SharedBufferDispatcher::Create(
                                 SharedBufferDispatcher::kDefaultCreateOptions,
                                 100, &dispatcher1));
 
+  Dispatcher::RequestContext request_context;
   MojoDuplicateBufferHandleOptions options[] = {
       {sizeof(MojoDuplicateBufferHandleOptions),
        MOJO_DUPLICATE_BUFFER_HANDLE_OPTIONS_FLAG_NONE},
       {sizeof(MojoDuplicateBufferHandleOptionsFlags), ~0u}};
   for (size_t i = 0; i < MOJO_ARRAYSIZE(options); i++) {
     scoped_refptr<Dispatcher> dispatcher2;
     EXPECT_EQ(MOJO_RESULT_OK, dispatcher1->DuplicateBufferHandle(
                                   &options[i], &dispatcher2));
     ASSERT_TRUE(dispatcher2);
     EXPECT_EQ(Dispatcher::Type::SHARED_BUFFER, dispatcher2->GetType());
-    EXPECT_EQ(MOJO_RESULT_OK, dispatcher2->Close());
+    EXPECT_EQ(MOJO_RESULT_OK, dispatcher2->Close(&request_context));
   }
 
-  EXPECT_EQ(MOJO_RESULT_OK, dispatcher1->Close());
+  EXPECT_EQ(MOJO_RESULT_OK, dispatcher1->Close(&request_context));
 }
 
 TEST_F(SharedBufferDispatcherTest, DuplicateBufferHandleOptionsInvalid) {
   scoped_refptr<SharedBufferDispatcher> dispatcher1;
   EXPECT_EQ(MOJO_RESULT_OK, SharedBufferDispatcher::Create(
                                 SharedBufferDispatcher::kDefaultCreateOptions,
                                 100, &dispatcher1));
 
   // Invalid |struct_size|.
   {
     MojoDuplicateBufferHandleOptions options = {
         1u, MOJO_DUPLICATE_BUFFER_HANDLE_OPTIONS_FLAG_NONE};
     scoped_refptr<Dispatcher> dispatcher2;
     EXPECT_EQ(MOJO_RESULT_INVALID_ARGUMENT,
               dispatcher1->DuplicateBufferHandle(&options, &dispatcher2));
     EXPECT_FALSE(dispatcher2);
   }
 
   // Unknown |flags|.
   {
     MojoDuplicateBufferHandleOptions options = {
         sizeof(MojoDuplicateBufferHandleOptions), ~0u};
     scoped_refptr<Dispatcher> dispatcher2;
     EXPECT_EQ(MOJO_RESULT_UNIMPLEMENTED,
               dispatcher1->DuplicateBufferHandle(&options, &dispatcher2));
     EXPECT_FALSE(dispatcher2);
   }
 
-  EXPECT_EQ(MOJO_RESULT_OK, dispatcher1->Close());
+  Dispatcher::RequestContext request_context;
+  EXPECT_EQ(MOJO_RESULT_OK, dispatcher1->Close(&request_context));
 }
 
 TEST_F(SharedBufferDispatcherTest, CreateInvalidNumBytes) {
   // Size too big.
   scoped_refptr<SharedBufferDispatcher> dispatcher;
   EXPECT_EQ(MOJO_RESULT_RESOURCE_EXHAUSTED,
             SharedBufferDispatcher::Create(
                 SharedBufferDispatcher::kDefaultCreateOptions,
                 std::numeric_limits<uint64_t>::max(), &dispatcher));
   EXPECT_FALSE(dispatcher);
@@ skipping 17 matching lines @@
   EXPECT_FALSE(mapping);
 
   EXPECT_EQ(MOJO_RESULT_INVALID_ARGUMENT,
             dispatcher->MapBuffer(1, 100, MOJO_MAP_BUFFER_FLAG_NONE, &mapping));
   EXPECT_FALSE(mapping);
 
   EXPECT_EQ(MOJO_RESULT_INVALID_ARGUMENT,
             dispatcher->MapBuffer(0, 0, MOJO_MAP_BUFFER_FLAG_NONE, &mapping));
   EXPECT_FALSE(mapping);
 
-  EXPECT_EQ(MOJO_RESULT_OK, dispatcher->Close());
+  Dispatcher::RequestContext request_context;
+  EXPECT_EQ(MOJO_RESULT_OK, dispatcher->Close(&request_context));
 }
 
 }  // namespace
 }  // namespace edk
 }  // namespace mojo