chromium_src/content/renderer/gin_browsertest.cc · OpenHarmony-TPC/chromium_src - AtomGit

// Copyright 2013 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "base/command_line.h"
#include "base/memory/raw_ptr.h"
#include "content/public/test/render_view_test.h"
#include "gin/per_isolate_data.h"
#include "gin/public/wrappable_pointer_tags.h"
#include "gin/wrappable.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/blink/public/common/switches.h"
#include "third_party/blink/public/web/web_local_frame.h"
#include "third_party/blink/public/web/web_view.h"
#include "v8/include/cppgc/allocation.h"
#include "v8/include/v8-context.h"
#include "v8/include/v8-cppgc.h"

namespace content {

namespace {

class TestGinObject : public gin::Wrappable<TestGinObject> {
 public:
  static constexpr gin::WrapperInfo kWrapperInfo = {{gin::kEmbedderNativeGin},
                                                    gin::kTestObject};

  static void Create(v8::Isolate* isolate, bool* alive) {
    auto* obj = cppgc::MakeGarbageCollected<TestGinObject>(
        isolate->GetCppHeap()->GetAllocationHandle(), alive);
    // We need to create a wrapper to keep it alive until the next GC. The local
    // handle will be destroyed at the end of the scope, making the wrapper
    // object eligible for GC.
    obj->GetWrapper(isolate).ToLocalChecked();
  }

  TestGinObject(const TestGinObject&) = delete;
  TestGinObject& operator=(const TestGinObject&) = delete;

  // Make public for cppgc::MakeGarbageCollected.
  explicit TestGinObject(bool* alive) : alive_(alive) { *alive_ = true; }
  ~TestGinObject() override { *alive_ = false; }

 private:
  // gin::Wrappable
  const gin::WrapperInfo* wrapper_info() const override {
    return &kWrapperInfo;
  }

  raw_ptr<bool> alive_;
};

class GinBrowserTest : public RenderViewTest {
 public:
  GinBrowserTest() = default;

  GinBrowserTest(const GinBrowserTest&) = delete;
  GinBrowserTest& operator=(const GinBrowserTest&) = delete;

  ~GinBrowserTest() override {}

  void SetUp() override {
    base::CommandLine::ForCurrentProcess()->AppendSwitchASCII(
        blink::switches::kJavaScriptFlags, "--expose_gc");

    RenderViewTest::SetUp();
  }
};

// Test that garbage collection doesn't crash if a gin-wrapped object is
// present.
TEST_F(GinBrowserTest, GinAndGarbageCollection) {
  LoadHTML("<!doctype html>");

  bool alive = false;

  {
    v8::Isolate* isolate = Isolate();
    v8::HandleScope handle_scope(isolate);
    v8::Context::Scope context_scope(GetMainFrame()->MainWorldScriptContext());

    // We create the object inside a scope so it's not kept alive by a handle
    // on the stack.
    TestGinObject::Create(isolate, &alive);
  }

  CHECK(alive);

  // Should not crash.
  Isolate()->RequestGarbageCollectionForTesting(
      v8::Isolate::kFullGarbageCollection, v8::StackState::kNoHeapPointers);
  CHECK(!alive);
}

}  // namespace

}  // namespace content