Add a CallbackRegistry class to base/ to manage callbacks

base/callback_list_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 "base/callback_list.h"
+
+#include <vector>
+
+#include "base/compiler_specific.h"
+#include "base/memory/weak_ptr.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace base {
+namespace {
+
+typedef CallbackList<base::Closure> TestCallbackList;
+
+typedef base::Callback<void(int)> OneParamCallback;
+typedef CallbackList<OneParamCallback> TestCallbackListWithParam;
+
+class Listener {
+ public:
+  explicit Listener() : total(0), scaler_(1) {}
+  Listener(int scaler) : total(0), scaler_(scaler) {}
+  virtual ~Listener() {}
+  virtual void OnEvent() {
+    total++;
+  }
+  virtual void OnEventWithParam(int x) {
+    total += x * scaler_;
+  }
+  int total;
+ private:
+  int scaler_;
+};
+
+class Remover : public Listener {
+ public:
+  explicit Remover() : removal_cb_(base::Closure()) {}
+  virtual ~Remover() {}
+  virtual void OnEvent() OVERRIDE {
+    Listener::OnEvent();
+    removal_cb_.Run();
+  }
+  virtual void SetCallback(const base::Closure& cb) {
+    removal_cb_ = cb;
+  }
+
+ private:
+  base::Closure removal_cb_;
+};
+
+class Adder : public Listener {
+ public:
+  explicit Adder(TestCallbackList* cb_list)
+      : added(false),
+        cb_list_(cb_list) {}
+  virtual ~Adder() {}
+  virtual void OnAddedEvent() {
+    if (!added) {
+      added = true;
+      cb_list_->RegisterCallback(
+          base::Bind(&Adder::OnEvent, base::Unretained(this)));
+    }
+  }
+  bool added;
+
+ private:
+  TestCallbackList* cb_list_;
+};
+
+class Clearer : public Listener {
+ public:
+  explicit Clearer(TestCallbackList* cb_list)
+      : added(false),
+        cb_list_(cb_list) {}
+  virtual ~Clearer() {}
+  virtual void OnClearAndAddEvent() {
+    cb_list_->Clear();
+    cb_list_->RegisterCallback(
+        base::Bind(&Clearer::OnEvent, base::Unretained(this)));
+    added = true;
+  }
+  bool added;
+
+ private:
+  TestCallbackList* cb_list_;
+};
+
+class ListDestructor {
+ public:
+  explicit ListDestructor(TestCallbackList* cb_list)
+      : cb_list_(cb_list) {}
+  virtual ~ListDestructor() {}
+  virtual void OnEvent() {
+    delete cb_list_;
+  }
+
+ private:
+  TestCallbackList* cb_list_;
+};
+
+TEST(CallbackListTest, BasicTest) {
+  TestCallbackList cb_list;
+  Listener a, b, c;
+  base::Closure a_cb = base::Bind(&Listener::OnEvent, base::Unretained(&a));
+  base::Closure b_cb = base::Bind(&Listener::OnEvent, base::Unretained(&b));
+
+  base::Closure remove_a = cb_list.RegisterCallback(a_cb);
+
+  EXPECT_TRUE(cb_list.HasCallback(a_cb));
+  EXPECT_FALSE(cb_list.HasCallback(b_cb));
+
+  base::Closure remove_b = cb_list.RegisterCallback(b_cb);
+  EXPECT_TRUE(cb_list.HasCallback(b_cb));
+
+  EXPECT_FALSE(remove_a.is_null());
+  EXPECT_FALSE(remove_b.is_null());
+
+  FOR_EACH_CALLBACK(base::Closure, cb_list);
+
+  EXPECT_EQ(1, a.total);
+  EXPECT_EQ(1, b.total);
+
+  remove_b.Run();
+
+  base::Closure remove_c = cb_list.RegisterCallback(
+      base::Bind(&Listener::OnEvent, base::Unretained(&c)));
+
+  FOR_EACH_CALLBACK(base::Closure, cb_list);
+
+  EXPECT_EQ(2, a.total);
+  EXPECT_EQ(1, b.total);
+  EXPECT_EQ(1, c.total);
+
+  remove_a.Run();
+  remove_b.Run();
+  remove_c.Run();
+
+  EXPECT_FALSE(cb_list.might_have_callbacks());
+}
+
+TEST(CallbackListTest, BasicTestWithParams) {
+  TestCallbackListWithParam cb_list;
+  Listener a(1), b(-1), c(1);
+  OneParamCallback a_cb =
+      base::Bind(&Listener::OnEventWithParam, base::Unretained(&a));
+
+  OneParamCallback b_cb =
+      base::Bind(&Listener::OnEventWithParam, base::Unretained(&b));
+
+  base::Closure remove_a = cb_list.RegisterCallback(a_cb);
+
+  EXPECT_TRUE(cb_list.HasCallback(a_cb));
+  EXPECT_FALSE(cb_list.HasCallback(b_cb));
+
+  base::Closure remove_b = cb_list.RegisterCallback(b_cb);
+  EXPECT_TRUE(cb_list.HasCallback(b_cb));
+
+  EXPECT_FALSE(remove_a.is_null());
+  EXPECT_FALSE(remove_b.is_null());
+
+  FOR_EACH_CALLBACK(OneParamCallback, cb_list, 10);
+
+  EXPECT_EQ(10, a.total);
+  EXPECT_EQ(-10, b.total);
+
+  remove_b.Run();
+
+  base::Closure remove_c = cb_list.RegisterCallback(
+      base::Bind(&Listener::OnEventWithParam, base::Unretained(&c)));
+
+  FOR_EACH_CALLBACK(OneParamCallback, cb_list, 10);
+
+  EXPECT_EQ(20, a.total);
+  EXPECT_EQ(-10, b.total);
+  EXPECT_EQ(10, c.total);
+
+  remove_a.Run();
+  remove_b.Run();
+  remove_c.Run();
+
+  EXPECT_FALSE(cb_list.might_have_callbacks());
+}
+
+TEST(CallbackListTest, RemoveCallbacksDuringIteration) {
+  TestCallbackList cb_list;
+  Listener a, b;
+  Remover remover_1, remover_2;
+
+  base::Closure remover_1_cb = cb_list.RegisterCallback(
+      base::Bind(&Remover::OnEvent, base::Unretained(&remover_1)));
+  base::Closure remover_2_cb = cb_list.RegisterCallback(
+      base::Bind(&Remover::OnEvent, base::Unretained(&remover_2)));
+  base::Closure a_cb = cb_list.RegisterCallback(
+      base::Bind(&Listener::OnEvent, base::Unretained(&a)));
+  base::Closure b_cb = cb_list.RegisterCallback(
+      base::Bind(&Listener::OnEvent, base::Unretained(&b)));
+
+  // |remover_1| will remove itself.
+  remover_1.SetCallback(remover_1_cb);
+  // |remover_2| will remove a.
+  remover_2.SetCallback(a_cb);
+
+  FOR_EACH_CALLBACK(base::Closure, cb_list);
+
+  // |remover_1| runs once (and removes itself), |remover_2| runs once (and
+  // removes a), |a| never runs, and |b| runs once.
+  EXPECT_EQ(1, remover_1.total);
+  EXPECT_EQ(1, remover_2.total);
+  EXPECT_EQ(0, a.total);
+  EXPECT_EQ(1, b.total);
+
+  FOR_EACH_CALLBACK(base::Closure, cb_list);
+
+  // Only |remover_2| and |b| run this time.
+  EXPECT_EQ(1, remover_1.total);
+  EXPECT_EQ(2, remover_2.total);
+  EXPECT_EQ(0, a.total);
+  EXPECT_EQ(2, b.total);
+
+  EXPECT_TRUE(cb_list.might_have_callbacks());
+
+  remover_2_cb.Run();
+  b_cb.Run();
+
+  EXPECT_FALSE(cb_list.might_have_callbacks());
+}
+
+TEST(CallbackListTest, AddCallbacksDuringIteration) {
+  TestCallbackList cb_list;
+  Adder a(&cb_list);
+  Listener b;
+  cb_list.RegisterCallback(
+      base::Bind(&Adder::OnAddedEvent, base::Unretained(&a)));
+  cb_list.RegisterCallback(
+      base::Bind(&Listener::OnEvent, base::Unretained(&b)));
+
+  FOR_EACH_CALLBACK(base::Closure, cb_list);
+
+  EXPECT_EQ(1, a.total);
+  EXPECT_EQ(1, b.total);
+  EXPECT_TRUE(a.added);
+
+  FOR_EACH_CALLBACK(base::Closure, cb_list);
+
+  EXPECT_EQ(2, a.total);
+  EXPECT_EQ(2, b.total);
+}
+
+TEST(CallbackListTest, AddCallbacksDuringIteration_Existing) {
+  TestCallbackList cb_list(TestCallbackList::NOTIFY_EXISTING_ONLY);
+  Adder a(&cb_list);
+  Listener b;
+  cb_list.RegisterCallback(
+      base::Bind(&Adder::OnAddedEvent, base::Unretained(&a)));
+  cb_list.RegisterCallback(
+      base::Bind(&Listener::OnEvent, base::Unretained(&b)));
+
+  FOR_EACH_CALLBACK(base::Closure, cb_list);
+
+  EXPECT_EQ(0, a.total);
+  EXPECT_EQ(1, b.total);
+  EXPECT_TRUE(a.added);
+
+  FOR_EACH_CALLBACK(base::Closure, cb_list);
+
+  EXPECT_EQ(1, a.total);
+  EXPECT_EQ(2, b.total);
+}
+
+TEST(CallbackListTest, AddAndRemoveCallbacksDuringIteration_Existing) {
+  TestCallbackList cb_list(TestCallbackList::NOTIFY_EXISTING_ONLY);
+  Adder a(&cb_list);
+  Remover b;
+  Listener c;
+
+  base::Closure c_cb = cb_list.RegisterCallback(
+      base::Bind(&Listener::OnEvent, base::Unretained(&c)));
+
+  // |b| removes |c|.
+  b.SetCallback(c_cb);
+  cb_list.RegisterCallback(
+      base::Bind(&Remover::OnEvent, base::Unretained(&b)));
+
+  // |a| adds a new callback.
+  base::Closure a_cb = cb_list.RegisterCallback(
+      base::Bind(&Adder::OnAddedEvent, base::Unretained(&a)));
+
+  FOR_EACH_CALLBACK(base::Closure, cb_list);
+
+  // |c| ran once, new callback should have been added but not yet run.
+  EXPECT_EQ(1, c.total);
+  EXPECT_EQ(0, a.total);
+  EXPECT_TRUE(a.added);
+
+  FOR_EACH_CALLBACK(base::Closure, cb_list);
+
+  // Now it should have been run.
+  EXPECT_EQ(1, c.total);
+  EXPECT_EQ(1, a.total);
+}
+
+TEST(CallbackListTest, ClearCallbacksDuringIteration) {
+  TestCallbackList cb_list;
+  Clearer a(&cb_list);
+  Listener b;
+  cb_list.RegisterCallback(
+      base::Bind(&Clearer::OnClearAndAddEvent, base::Unretained(&a)));
+  cb_list.RegisterCallback(
+      base::Bind(&Listener::OnEvent, base::Unretained(&b)));
+
+  FOR_EACH_CALLBACK(base::Closure, cb_list);
+
+  EXPECT_EQ(1, a.total);
+  EXPECT_EQ(0, b.total);
+  EXPECT_TRUE(a.added);
+}
+
+TEST(CallbackListTest, IteratorOutlivesList) {
+  TestCallbackList* cb_list = new TestCallbackList();
+  ListDestructor destructor(cb_list);
+  cb_list->RegisterCallback(
+      base::Bind(&ListDestructor::OnEvent, base::Unretained(&destructor)));
+  Listener a;
+  cb_list->RegisterCallback(
+      base::Bind(&Listener::OnEvent, base::Unretained(&a)));
+
+  FOR_EACH_CALLBACK(base::Closure, *cb_list);
+
+  // |a| never gets called, as |cb_list| got deleted first.
+  EXPECT_EQ(0, a.total);
+}
+
+}  // namespace
+}  // namespace base