Update monet.

mojo/common/strong_binding_set_unittest.cc

+// Copyright 2015 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/common/strong_binding_set.h"
+
+#include "base/message_loop/message_loop.h"
+#include "mojo/common/test_interfaces.mojom.h"
+#include "mojo/message_pump/message_pump_mojo.h"
+#include "mojo/public/cpp/bindings/binding.h"
+#include "mojo/public/cpp/bindings/interface_request.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace mojo {
+namespace common {
+namespace {
+
+class DummyImpl : public tests::Dummy {
+ public:
+  explicit DummyImpl(bool* deleted_flag) : deleted_flag_(deleted_flag) {}
+  ~DummyImpl() override { *deleted_flag_ = true; }
+
+  void Foo() override { call_count_++; }
+
+  int call_count() const { return call_count_; }
+
+ private:
+  bool* deleted_flag_;
+  int call_count_ = 0;
+
+  DISALLOW_COPY_AND_ASSIGN(DummyImpl);
+};
+
+// Tests all of the functionality of StrongBindingSet.
+TEST(StrongBindingSetTest, FullLifeCycle) {
+  base::MessageLoop loop(MessagePumpMojo::Create());
+
+  // Create 10 InterfacePtrs and DummyImpls.
+  const size_t kNumObjects = 10;
+  InterfacePtr<tests::Dummy> intrfc_ptrs[kNumObjects];
+  DummyImpl* impls[kNumObjects];
+  bool deleted_flags[kNumObjects] = {};
+
+  // Create 10 message pipes, bind everything together, and add the
+  // bindings to binding_set.
+  StrongBindingSet<tests::Dummy> binding_set;
+  EXPECT_EQ(0u, binding_set.size());
+  for (size_t i = 0; i < kNumObjects; i++) {
+    impls[i] = new DummyImpl(&deleted_flags[i]);
+    binding_set.AddBinding(impls[i], GetProxy(&intrfc_ptrs[i]));
+  }
+  EXPECT_EQ(kNumObjects, binding_set.size());
+
+  // Check that initially all call counts are zero.
+  for (const auto& impl : impls) {
+    EXPECT_EQ(0, impl->call_count());
+  }
+
+  // Invoke method foo() on all 10 InterfacePointers.
+  for (InterfacePtr<tests::Dummy>& ptr : intrfc_ptrs) {
+    ptr->Foo();
+  }
+
+  // Check that now all call counts are one.
+  loop.RunUntilIdle();
+  for (const auto& impl : impls) {
+    EXPECT_EQ(1, impl->call_count());
+  }
+
+  // Close the first 5 message pipes and destroy the first five
+  // InterfacePtrs.
+  for (size_t i = 0; i < kNumObjects / 2; i++) {
+    intrfc_ptrs[i].reset();
+  }
+
+  // Check that the set contains only five elements now.
+  loop.RunUntilIdle();
+  EXPECT_EQ(kNumObjects / 2, binding_set.size());
+
+  // Check that the first 5 interfaces have all been deleted.
+  for (size_t i = 0; i < kNumObjects; i++) {
+    bool expected = (i < kNumObjects / 2);
+    EXPECT_EQ(expected, deleted_flags[i]);
+  }
+
+  // Invoke method foo() on the second five InterfacePointers.
+  for (size_t i = kNumObjects / 2; i < kNumObjects; i++) {
+    intrfc_ptrs[i]->Foo();
+  }
+  loop.RunUntilIdle();
+
+  // Check that now the second five counts are two.
+  for (size_t i = kNumObjects / 2; i < kNumObjects; i++) {
+    EXPECT_EQ(2, impls[i]->call_count());
+  }
+
+  // Invoke CloseAllBindings
+  binding_set.CloseAllBindings();
+  EXPECT_EQ(0u, binding_set.size());
+
+  // Invoke method foo() on the second five InterfacePointers.
+  for (size_t i = kNumObjects / 2; i < kNumObjects; i++) {
+    intrfc_ptrs[i]->Foo();
+  }
+  loop.RunUntilIdle();
+
+  // Check that all interfaces have all been deleted.
+  for (size_t i = 0; i < kNumObjects; i++) {
+    EXPECT_TRUE(deleted_flags[i]);
+  }
+}
+
+}  // namespace
+}  // namespace common
+}  // namespace mojo