Move ScriptState::forWorld/ScriptState::forMainWorld (Part 2)

third_party/WebKit/Source/core/dom/custom/CustomElementRegistryTest.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 "core/dom/custom/CustomElementRegistry.h"
 
+#include <memory>
 #include "bindings/core/v8/ExceptionState.h"
+#include "bindings/core/v8/V8Binding.h"
 #include "core/dom/Document.h"
 #include "core/dom/Element.h"
 #include "core/dom/ElementDefinitionOptions.h"
 #include "core/dom/custom/CEReactionsScope.h"
 #include "core/dom/custom/CustomElementDefinition.h"
 #include "core/dom/custom/CustomElementDefinitionBuilder.h"
 #include "core/dom/custom/CustomElementDescriptor.h"
 #include "core/dom/custom/CustomElementTestHelpers.h"
 #include "core/dom/shadow/ShadowRoot.h"
 #include "core/dom/shadow/ShadowRootInit.h"
 #include "core/testing/DummyPageHolder.h"
 #include "platform/ScriptForbiddenScope.h"
 #include "platform/heap/Handle.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "wtf/text/AtomicString.h"
-#include <memory>
 
 namespace blink {
 
 class CustomElementRegistryTest : public ::testing::Test {
  protected:
   void SetUp() {
     m_page.reset(DummyPageHolder::create(IntSize(1, 1)).release());
   }
 
   void TearDown() { m_page = nullptr; }
 
   Document& document() { return m_page->document(); }
 
   CustomElementRegistry& registry() {
     return *m_page->frame().domWindow()->customElements();
   }
 
   ScriptState* scriptState() {
-    return ScriptState::forMainWorld(&m_page->frame());
+    return toScriptStateForMainWorld(&m_page->frame());
   }
 
   void collectCandidates(const CustomElementDescriptor& desc,
                          HeapVector<Member<Element>>* elements) {
     registry().collectCandidates(desc, elements);
   }
 
   ShadowRoot* attachShadowTo(Element* element) {
     NonThrowableExceptionState noExceptions;
     ShadowRootInit shadowRootInit;
@@ skipping 385 matching lines @@
   EXPECT_EQ(definitionB, definition);
   // look up undefined customized built-in element
   definition = registry().definitionFor(CustomElementDescriptor("a-a", "div"));
   EXPECT_EQ(nullptr, definition) << "a-a, div should not be registered";
 }
 
 // TODO(dominicc): Add tests which adjust the "is" attribute when type
 // extensions are implemented.
 
 }  // namespace blink