EDK: Add HandleTable::AddHandleVector().

mojo/edk/system/handle_table_unittest.cc

 // Copyright 2016 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/handle_table.h"
 
+#include <vector>
+
 #include "mojo/edk/system/dispatcher.h"
 #include "mojo/edk/system/handle.h"
 #include "mojo/edk/system/mock_simple_dispatcher.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using mojo::util::MakeRefCounted;
 using mojo::util::RefPtr;
 
 namespace mojo {
 namespace system {
@@ skipping 151 matching lines @@
   d = nullptr;
   ASSERT_EQ(MOJO_RESULT_OK, ht.GetAndRemoveDispatcher(h4, &d));
   ASSERT_EQ(d4, d);
 
   EXPECT_EQ(MOJO_RESULT_OK, d1->Close());
   EXPECT_EQ(MOJO_RESULT_OK, d2->Close());
   EXPECT_EQ(MOJO_RESULT_OK, d3->Close());
   EXPECT_EQ(MOJO_RESULT_OK, d4->Close());
 }
 
+TEST(HandleTableTest, AddHandleVector) {
+  static constexpr size_t kNumHandles = 10u;
+
+  HandleTable ht(1000u);
+
+  HandleVector handles;
+  std::vector<RefPtr<Dispatcher>> dispatchers;
+  for (size_t i = 0u; i < kNumHandles; i++) {
+    dispatchers.push_back(MakeRefCounted<test::MockSimpleDispatcher>());
+    handles.push_back(
+        Handle(dispatchers.back().Clone(), MOJO_HANDLE_RIGHT_NONE));
+    ASSERT_TRUE(handles[i]) << i;
+  }
+
+  std::vector<MojoHandle> handle_values(kNumHandles, MOJO_HANDLE_INVALID);
+
+  ASSERT_TRUE(ht.AddHandleVector(&handles, handle_values.data()));
+
+  for (size_t i = 0u; i < kNumHandles; i++) {
+    ASSERT_NE(handle_values[i], MOJO_HANDLE_INVALID) << i;
+
+    RefPtr<Dispatcher> d;
+    ASSERT_EQ(MOJO_RESULT_OK, ht.GetAndRemoveDispatcher(handle_values[i], &d))
+        << i;
+    ASSERT_EQ(dispatchers[i], d) << i;
+
+    EXPECT_EQ(MOJO_RESULT_OK, dispatchers[i]->Close()) << i;
+  }
+}
+
+TEST(HandleTableTest, AddHandleVectorTooMany) {
+  static constexpr size_t kHandleTableSize = 10u;
+  static constexpr size_t kNumHandles = kHandleTableSize + 1u;
+
+  HandleTable ht(kHandleTableSize);
+
+  HandleVector handles;
+  std::vector<RefPtr<Dispatcher>> dispatchers;
+  for (size_t i = 0u; i < kNumHandles; i++) {
+    dispatchers.push_back(MakeRefCounted<test::MockSimpleDispatcher>());
+    handles.push_back(
+        Handle(dispatchers.back().Clone(), MOJO_HANDLE_RIGHT_NONE));
+    ASSERT_TRUE(handles[i]) << i;
+  }
+
+  std::vector<MojoHandle> handle_values(kNumHandles, MOJO_HANDLE_INVALID);
+
+  EXPECT_FALSE(ht.AddHandleVector(&handles, handle_values.data()));
+
+  handles.pop_back();
+  handle_values.pop_back();
+
+  ASSERT_TRUE(ht.AddHandleVector(&handles, handle_values.data()));
+
+  for (size_t i = 0u; i < kNumHandles - 1u; i++) {
+    ASSERT_NE(handle_values[i], MOJO_HANDLE_INVALID) << i;
+
+    RefPtr<Dispatcher> d;
+    ASSERT_EQ(MOJO_RESULT_OK, ht.GetAndRemoveDispatcher(handle_values[i], &d))
+        << i;
+    ASSERT_EQ(dispatchers[i], d) << i;
+  }
+
+  for (size_t i = 0u; i < kNumHandles; i++)
+    EXPECT_EQ(MOJO_RESULT_OK, dispatchers[i]->Close()) << i;
+}
+
 }  // namespace
 }  // namespace system
 }  // namespace mojo