third_party/WebKit/Source/core/dom/ExecutionContextTaskTest.cpp

// 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 "core/dom/ExecutionContextTask.h"

#include "platform/heap/Handle.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace blink {

class GCObject : public GarbageCollectedFinalized<GCObject> {
 public:
  static int s_counter;
  GCObject() { ++s_counter; }
  ~GCObject() { --s_counter; }
  DEFINE_INLINE_TRACE() {}
  void run(GCObject*) {}
};

int GCObject::s_counter = 0;

static void functionWithGarbageCollected(GCObject*) {}

static void functionWithExecutionContext(GCObject*, ExecutionContext*) {}

class CrossThreadTaskTest : public testing::Test {
 protected:
  void SetUp() override { GCObject::s_counter = 0; }
  void TearDown() override {
    ThreadState::current()->collectGarbage(BlinkGC::NoHeapPointersOnStack,
                                           BlinkGC::GCWithSweep,
                                           BlinkGC::ForcedGC);
    ASSERT_EQ(0, GCObject::s_counter);
  }
};

TEST_F(CrossThreadTaskTest, CreateForGarbageCollectedMethod) {
  std::unique_ptr<ExecutionContextTask> task1 = createCrossThreadTask(
      &GCObject::run, wrapCrossThreadPersistent(new GCObject),
      wrapCrossThreadPersistent(new GCObject));
  std::unique_ptr<ExecutionContextTask> task2 = createCrossThreadTask(
      &GCObject::run, wrapCrossThreadPersistent(new GCObject),
      wrapCrossThreadPersistent(new GCObject));
  ThreadState::current()->collectGarbage(
      BlinkGC::NoHeapPointersOnStack, BlinkGC::GCWithSweep, BlinkGC::ForcedGC);
  EXPECT_EQ(4, GCObject::s_counter);
}

TEST_F(CrossThreadTaskTest, CreateForFunctionWithGarbageCollected) {
  std::unique_ptr<ExecutionContextTask> task1 = createCrossThreadTask(
      &functionWithGarbageCollected, wrapCrossThreadPersistent(new GCObject));
  std::unique_ptr<ExecutionContextTask> task2 = createCrossThreadTask(
      &functionWithGarbageCollected, wrapCrossThreadPersistent(new GCObject));
  ThreadState::current()->collectGarbage(
      BlinkGC::NoHeapPointersOnStack, BlinkGC::GCWithSweep, BlinkGC::ForcedGC);
  EXPECT_EQ(2, GCObject::s_counter);
}

TEST_F(CrossThreadTaskTest, CreateForFunctionWithExecutionContext) {
  std::unique_ptr<ExecutionContextTask> task1 = createCrossThreadTask(
      &functionWithExecutionContext, wrapCrossThreadPersistent(new GCObject));
  std::unique_ptr<ExecutionContextTask> task2 = createCrossThreadTask(
      &functionWithExecutionContext, wrapCrossThreadPersistent(new GCObject));
  ThreadState::current()->collectGarbage(
      BlinkGC::NoHeapPointersOnStack, BlinkGC::GCWithSweep, BlinkGC::ForcedGC);
  EXPECT_EQ(2, GCObject::s_counter);
}

}  // namespace blink