Specialize base::IsWeakReceiver for Blink weak pointers to support base::Bind (1/5)

third_party/WebKit/Source/platform/heap/PersistentTest.cpp

+// 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 "platform/heap/Persistent.h"
+
+#include "platform/ThreadSafeFunctional.h"
+#include "platform/heap/Heap.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include <memory>
+
+namespace blink {
+namespace {
+
+void preciselyCollectGarbage()
+{
+    ThreadHeap::collectGarbage(BlinkGC::NoHeapPointersOnStack, BlinkGC::GCWithSweep, BlinkGC::ForcedGC);
+}
+
+class Receiver : public GarbageCollected<Receiver> {
+public:
+    void increment(int* counter)
+    {
+        ++*counter;
+    }
+
+    DEFINE_INLINE_TRACE() {}
+};
+
+TEST(PersistentTest, BindCancellation)
+{
+    Receiver* receiver = new Receiver;
+    int counter = 0;
+    std::unique_ptr<SameThreadClosure> function = WTF::bind(&Receiver::increment, wrapWeakPersistent(receiver), WTF::unretained(&counter));
+
+    (*function)();
+    EXPECT_EQ(1, counter);
+
+    receiver = nullptr;
+    preciselyCollectGarbage();
+    (*function)();
+    EXPECT_EQ(1, counter);
+}
+
+TEST(PersistentTest, CrossThreadBindCancellation)
+{
+    Receiver* receiver = new Receiver;
+    int counter = 0;
+    std::unique_ptr<CrossThreadClosure> function = blink::threadSafeBind(&Receiver::increment, wrapCrossThreadWeakPersistent(receiver), WTF::crossThreadUnretained(&counter));
+
+    (*function)();
+    EXPECT_EQ(1, counter);
+
+    receiver = nullptr;
+    preciselyCollectGarbage();
+    (*function)();
+    EXPECT_EQ(1, counter);
+}
+
+} // namespace
+} // namespace blink