1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70
|
// 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
|