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/CrossThreadFunctional.h"
#include "platform/heap/Handle.h"
#include "testing/gtest/include/gtest/gtest.h"
#include <memory>

namespace blink {
namespace {

void preciselyCollectGarbage() {
  ThreadState::current()->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<WTF::Closure> 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::crossThreadBind(
      &Receiver::increment, wrapCrossThreadWeakPersistent(receiver),
      WTF::crossThreadUnretained(&counter));

  (*function)();
  EXPECT_EQ(1, counter);

  receiver = nullptr;
  preciselyCollectGarbage();
  (*function)();
  EXPECT_EQ(1, counter);
}

}  // namespace
}  // namespace blink