Worklet: Lazily create PaintWorkletGlobalScopes

third_party/WebKit/Source/modules/csspaint/PaintWorkletTest.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 "modules/csspaint/PaintWorklet.h"
 
+#include <memory>
 #include "bindings/core/v8/ScriptSourceCode.h"
 #include "bindings/core/v8/V8GCController.h"
 #include "bindings/core/v8/WorkerOrWorkletScriptController.h"
 #include "core/frame/LocalFrame.h"
 #include "core/testing/DummyPageHolder.h"
 #include "modules/csspaint/CSSPaintDefinition.h"
 #include "modules/csspaint/PaintWorkletGlobalScope.h"
+#include "modules/csspaint/PaintWorkletGlobalScopeProxy.h"
 #include "modules/csspaint/WindowPaintWorklet.h"
 #include "testing/gtest/include/gtest/gtest.h"
-#include <memory>
 
 namespace blink {
 
+using GlobalScopeProxies =
+    WTF::HashSet<std::unique_ptr<WorkletGlobalScopeProxy>>;
+
 class PaintWorkletTest : public testing::Test {
  public:
   PaintWorkletTest() : page_(DummyPageHolder::Create()) {}
 
+  void SetUp() override { GetPaintWorklet()->AddWorkletGlobalScopesIfNeeded(); }
+
   PaintWorklet* GetPaintWorklet() {
     return WindowPaintWorklet::From(*page_->GetFrame().DomWindow())
         .paintWorklet();
   }
 
+  GlobalScopeProxies& GetGlobalScopeProxies() {
+    return GetPaintWorklet()->global_scope_proxies_;
+  }
+
  protected:
   std::unique_ptr<DummyPageHolder> page_;
 };
 
 TEST_F(PaintWorkletTest, GarbageCollectionOfCSSPaintDefinition) {
-  PaintWorkletGlobalScopeProxy* proxy = PaintWorkletGlobalScopeProxy::From(
-      GetPaintWorklet()->GetWorkletGlobalScopeProxy());
+  GlobalScopeProxies& proxies = GetGlobalScopeProxies();
+  EXPECT_EQ(1u, proxies.size());
+  PaintWorkletGlobalScopeProxy* proxy =
+      PaintWorkletGlobalScopeProxy::From(proxies.begin()->get());
   PaintWorkletGlobalScope* global_scope = proxy->global_scope();
   global_scope->ScriptController()->Evaluate(
       ScriptSourceCode("registerPaint('foo', class { paint() { } });"));
 
   CSSPaintDefinition* definition = global_scope->FindDefinition("foo");
   ASSERT(definition);
 
   v8::Isolate* isolate =
       global_scope->ScriptController()->GetScriptState()->GetIsolate();
   ASSERT(isolate);
@@ skipping 16 matching lines @@
   // Delete the page & associated objects.
   page_.reset();
 
   // Run a GC, the persistent should have been collected.
   ThreadState::Current()->CollectAllGarbage();
   V8GCController::CollectAllGarbageForTesting(isolate);
   ASSERT(handle.IsEmpty());
 }
 
 }  // namespace blink