Add Callback::IsCancelled

base/bind_unittest.cc

 // Copyright (c) 2012 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/bind.h"
 
 #include <memory>
 #include <utility>
 #include <vector>
 
@@ skipping 1016 matching lines @@
 
   int x = 1;
   base::Callback<void(int)> cb =
       Bind([](int* x, int i) { *x *= i; }, Unretained(&x));
   cb.Run(6);
   EXPECT_EQ(6, x);
   cb.Run(7);
   EXPECT_EQ(42, x);
 }
 
+TEST_F(BindTest, Cancellation) {
+  EXPECT_CALL(no_ref_, VoidMethod0()).Times(2);
+
+  WeakPtrFactory<NoRef> weak_factory(&no_ref_);
+  Closure cb = Bind(&NoRef::VoidMethod0, weak_factory.GetWeakPtr());
+  Closure cb2 = Bind(cb);
+
+  EXPECT_FALSE(cb.IsCancelled());
+  EXPECT_FALSE(cb2.IsCancelled());
+
+  cb.Run();
+  cb2.Run();
+
+  weak_factory.InvalidateWeakPtrs();
+
+  EXPECT_TRUE(cb.IsCancelled());
+  EXPECT_TRUE(cb2.IsCancelled());
+
+  cb.Run();
+  cb2.Run();
+}
+
 // Callback construction and assignment tests.
 //   - Construction from an InvokerStorageHolder should not cause ref/deref.
 //   - Assignment from other callback should only cause one ref
 //
 // TODO(ajwong): Is there actually a way to test this?
 
 #if defined(OS_WIN)
 int __fastcall FastCallFunc(int n) {
   return n;
 }
@@ skipping 16 matching lines @@
 
 // Test null callbacks cause a DCHECK.
 TEST(BindDeathTest, NullCallback) {
   base::Callback<void(int)> null_cb;
   ASSERT_TRUE(null_cb.is_null());
   EXPECT_DCHECK_DEATH(base::Bind(null_cb, 42));
 }
 
 }  // namespace
 }  // namespace base