blob: 9e89eba11d7bf5c5b25328afb90136c076ddca2c [file] [log] [blame]
// Copyright 2020 PDFium 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 "fxjs/gc/heap.h"
#include <memory>
#include <set>
#include "testing/fxgc_unittest.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/base/containers/contains.h"
#include "v8/include/cppgc/allocation.h"
#include "v8/include/cppgc/persistent.h"
namespace {
class PseudoCollectible : public cppgc::GarbageCollected<PseudoCollectible> {
public:
static void ClearCounts() {
s_live_.clear();
s_dead_.clear();
}
static size_t LiveCount() { return s_live_.size(); }
static size_t DeadCount() { return s_dead_.size(); }
PseudoCollectible() { s_live_.insert(this); }
virtual ~PseudoCollectible() {
s_live_.erase(this);
s_dead_.insert(this);
}
bool IsLive() const { return pdfium::Contains(s_live_, this); }
virtual void Trace(cppgc::Visitor* visitor) const {}
private:
static std::set<const PseudoCollectible*> s_live_;
static std::set<const PseudoCollectible*> s_dead_;
};
std::set<const PseudoCollectible*> PseudoCollectible::s_live_;
std::set<const PseudoCollectible*> PseudoCollectible::s_dead_;
class CollectibleHolder {
public:
explicit CollectibleHolder(PseudoCollectible* holdee) : holdee_(holdee) {}
~CollectibleHolder() = default;
PseudoCollectible* holdee() const { return holdee_; }
private:
cppgc::Persistent<PseudoCollectible> holdee_;
};
} // namespace
class HeapUnitTest : public FXGCUnitTest {
public:
HeapUnitTest() = default;
~HeapUnitTest() override = default;
// FXGCUnitTest:
void TearDown() override {
PseudoCollectible::ClearCounts();
FXGCUnitTest::TearDown();
}
};
TEST_F(HeapUnitTest, SeveralHeaps) {
FXGCScopedHeap heap1 = FXGC_CreateHeap();
EXPECT_TRUE(heap1);
FXGCScopedHeap heap2 = FXGC_CreateHeap();
EXPECT_TRUE(heap2);
FXGCScopedHeap heap3 = FXGC_CreateHeap();
EXPECT_TRUE(heap3);
// Test manually destroying the heap.
heap3.reset();
EXPECT_FALSE(heap3);
heap3.reset();
EXPECT_FALSE(heap3);
}
TEST_F(HeapUnitTest, NoReferences) {
FXGCScopedHeap heap1 = FXGC_CreateHeap();
ASSERT_TRUE(heap1);
{
auto holder = std::make_unique<CollectibleHolder>(
cppgc::MakeGarbageCollected<PseudoCollectible>(
heap1->GetAllocationHandle()));
EXPECT_TRUE(holder->holdee()->IsLive());
EXPECT_EQ(1u, PseudoCollectible::LiveCount());
EXPECT_EQ(0u, PseudoCollectible::DeadCount());
}
FXGC_ForceGarbageCollection(heap1.get());
EXPECT_EQ(0u, PseudoCollectible::LiveCount());
EXPECT_EQ(1u, PseudoCollectible::DeadCount());
}
TEST_F(HeapUnitTest, HasReferences) {
FXGCScopedHeap heap1 = FXGC_CreateHeap();
ASSERT_TRUE(heap1);
{
auto holder = std::make_unique<CollectibleHolder>(
cppgc::MakeGarbageCollected<PseudoCollectible>(
heap1->GetAllocationHandle()));
EXPECT_TRUE(holder->holdee()->IsLive());
EXPECT_EQ(1u, PseudoCollectible::LiveCount());
EXPECT_EQ(0u, PseudoCollectible::DeadCount());
FXGC_ForceGarbageCollection(heap1.get());
EXPECT_TRUE(holder->holdee()->IsLive());
EXPECT_EQ(1u, PseudoCollectible::LiveCount());
EXPECT_EQ(0u, PseudoCollectible::DeadCount());
}
}
// TODO(tsepez): enable when CPPGC fixes this segv.
TEST_F(HeapUnitTest, DISABLED_DeleteHeapHasReferences) {
FXGCScopedHeap heap1 = FXGC_CreateHeap();
ASSERT_TRUE(heap1);
{
auto holder = std::make_unique<CollectibleHolder>(
cppgc::MakeGarbageCollected<PseudoCollectible>(
heap1->GetAllocationHandle()));
EXPECT_TRUE(holder->holdee()->IsLive());
EXPECT_EQ(1u, PseudoCollectible::LiveCount());
EXPECT_EQ(0u, PseudoCollectible::DeadCount());
heap1.reset();
// Maybe someday magically nulled by heap destruction.
EXPECT_FALSE(holder->holdee());
EXPECT_EQ(1u, PseudoCollectible::LiveCount());
EXPECT_EQ(0u, PseudoCollectible::DeadCount());
}
}
TEST_F(HeapUnitTest, DeleteHeapNoReferences) {
FXGCScopedHeap heap1 = FXGC_CreateHeap();
ASSERT_TRUE(heap1);
{
auto holder = std::make_unique<CollectibleHolder>(
cppgc::MakeGarbageCollected<PseudoCollectible>(
heap1->GetAllocationHandle()));
EXPECT_TRUE(holder->holdee()->IsLive());
EXPECT_EQ(1u, PseudoCollectible::LiveCount());
EXPECT_EQ(0u, PseudoCollectible::DeadCount());
}
heap1.reset();
EXPECT_EQ(0u, PseudoCollectible::LiveCount());
EXPECT_EQ(1u, PseudoCollectible::DeadCount());
}