Move MainWorldScriptContext accessor/method from WebFrame to WebLocalFrame.

third_party/WebKit/Source/web/tests/WebFrameTest.cpp

 /*
  * Copyright (C) 2010 Google Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions are
  * met:
  *
  *     * Redistributions of source code must retain the above copyright
  * notice, this list of conditions and the following disclaimer.
  *     * Redistributions in binary form must reproduce the above
@@ skipping 384 matching lines @@
 TEST_P(ParameterizedWebFrameTest, FrameForEnteredContext) {
   RegisterMockedHttpURLLoad("iframes_test.html");
   RegisterMockedHttpURLLoad("visible_iframe.html");
   RegisterMockedHttpURLLoad("invisible_iframe.html");
   RegisterMockedHttpURLLoad("zero_sized_iframe.html");
 
   FrameTestHelpers::WebViewHelper web_view_helper;
   web_view_helper.InitializeAndLoad(base_url_ + "iframes_test.html", true);
 
   v8::HandleScope scope(v8::Isolate::GetCurrent());
-  EXPECT_EQ(
-      web_view_helper.WebView()->MainFrame(),
-      WebLocalFrame::FrameForContext(
-          web_view_helper.WebView()->MainFrame()->MainWorldScriptContext()));
+  EXPECT_EQ(web_view_helper.WebView()->MainFrame(),
+            WebLocalFrame::FrameForContext(web_view_helper.WebView()
+                                               ->MainFrameImpl()
+                                               ->MainWorldScriptContext()));
   EXPECT_EQ(web_view_helper.WebView()->MainFrame()->FirstChild(),
             WebLocalFrame::FrameForContext(web_view_helper.WebView()
                                                ->MainFrame()
                                                ->FirstChild()
+                                               ->ToWebLocalFrame()
                                                ->MainWorldScriptContext()));
 }
 
 class ScriptExecutionCallbackHelper : public WebScriptExecutionCallback {
  public:
   explicit ScriptExecutionCallbackHelper(v8::Local<v8::Context> context)
       : did_complete_(false), bool_value_(false), context_(context) {}
   ~ScriptExecutionCallbackHelper() {}
 
   bool DidComplete() const { return did_complete_; }
@@ skipping 20 matching lines @@
 };
 
 TEST_P(ParameterizedWebFrameTest, RequestExecuteScript) {
   RegisterMockedHttpURLLoad("foo.html");
 
   FrameTestHelpers::WebViewHelper web_view_helper;
   web_view_helper.InitializeAndLoad(base_url_ + "foo.html", true);
 
   v8::HandleScope scope(v8::Isolate::GetCurrent());
   ScriptExecutionCallbackHelper callback_helper(
-      web_view_helper.WebView()->MainFrame()->MainWorldScriptContext());
+      web_view_helper.WebView()->MainFrameImpl()->MainWorldScriptContext());
   web_view_helper.WebView()
       ->MainFrameImpl()
       ->RequestExecuteScriptAndReturnValue(
           WebScriptSource(WebString("'hello';")), false, &callback_helper);
   RunPendingTasks();
   EXPECT_TRUE(callback_helper.DidComplete());
   EXPECT_EQ("hello", callback_helper.StringValue());
 }
 
 TEST_P(ParameterizedWebFrameTest, SuspendedRequestExecuteScript) {
   RegisterMockedHttpURLLoad("foo.html");
   RegisterMockedHttpURLLoad("bar.html");
 
   FrameTestHelpers::WebViewHelper web_view_helper;
   web_view_helper.InitializeAndLoad(base_url_ + "foo.html", true);
 
   v8::HandleScope scope(v8::Isolate::GetCurrent());
   ScriptExecutionCallbackHelper callback_helper(
-      web_view_helper.WebView()->MainFrame()->MainWorldScriptContext());
+      web_view_helper.WebView()->MainFrameImpl()->MainWorldScriptContext());
 
   // Suspend scheduled tasks so the script doesn't run.
   web_view_helper.WebView()
       ->MainFrameImpl()
       ->GetFrame()
       ->GetDocument()
       ->SuspendScheduledTasks();
   web_view_helper.WebView()
       ->MainFrameImpl()
       ->RequestExecuteScriptAndReturnValue(
@@ skipping 14 matching lines @@
 
   FrameTestHelpers::WebViewHelper web_view_helper;
   web_view_helper.InitializeAndLoad(base_url_ + "foo.html", true);
 
   auto callback = [](const v8::FunctionCallbackInfo<v8::Value>& info) {
     info.GetReturnValue().Set(V8String(info.GetIsolate(), "hello"));
   };
 
   v8::HandleScope scope(v8::Isolate::GetCurrent());
   v8::Local<v8::Context> context =
-      web_view_helper.WebView()->MainFrame()->MainWorldScriptContext();
+      web_view_helper.WebView()->MainFrameImpl()->MainWorldScriptContext();
   ScriptExecutionCallbackHelper callback_helper(context);
   v8::Local<v8::Function> function =
       v8::Function::New(context, callback).ToLocalChecked();
   web_view_helper.WebView()
       ->MainFrame()
       ->ToWebLocalFrame()
       ->RequestExecuteV8Function(context, function,
                                  v8::Undefined(context->GetIsolate()), 0,
                                  nullptr, &callback_helper);
   RunPendingTasks();
   EXPECT_TRUE(callback_helper.DidComplete());
   EXPECT_EQ("hello", callback_helper.StringValue());
 }
 
 TEST_P(ParameterizedWebFrameTest, RequestExecuteV8FunctionWhileSuspended) {
   RegisterMockedHttpURLLoad("foo.html");
 
   FrameTestHelpers::WebViewHelper web_view_helper;
   web_view_helper.InitializeAndLoad(base_url_ + "foo.html", true);
 
   auto callback = [](const v8::FunctionCallbackInfo<v8::Value>& info) {
     info.GetReturnValue().Set(V8String(info.GetIsolate(), "hello"));
   };
 
   v8::HandleScope scope(v8::Isolate::GetCurrent());
   v8::Local<v8::Context> context =
-      web_view_helper.WebView()->MainFrame()->MainWorldScriptContext();
+      web_view_helper.WebView()->MainFrameImpl()->MainWorldScriptContext();
 
   // Suspend scheduled tasks so the script doesn't run.
   WebLocalFrameBase* main_frame = web_view_helper.WebView()->MainFrameImpl();
   main_frame->GetFrame()->GetDocument()->SuspendScheduledTasks();
 
   ScriptExecutionCallbackHelper callback_helper(context);
   v8::Local<v8::Function> function =
       v8::Function::New(context, callback).ToLocalChecked();
   main_frame->RequestExecuteV8Function(context, function,
                                        v8::Undefined(context->GetIsolate()), 0,
@@ skipping 19 matching lines @@
         info.GetIsolate(), UserGestureIndicator::ProcessingUserGesture()));
   };
 
   // Suspend scheduled tasks so the script doesn't run.
   WebLocalFrameBase* main_frame = web_view_helper.WebView()->MainFrameImpl();
   Document* document = main_frame->GetFrame()->GetDocument();
   document->SuspendScheduledTasks();
 
   v8::HandleScope scope(v8::Isolate::GetCurrent());
   v8::Local<v8::Context> context =
-      web_view_helper.WebView()->MainFrame()->MainWorldScriptContext();
+      web_view_helper.WebView()->MainFrameImpl()->MainWorldScriptContext();
 
   std::unique_ptr<UserGestureIndicator> indicator =
       WTF::WrapUnique(new UserGestureIndicator(
           UserGestureToken::Create(document, UserGestureToken::kNewGesture)));
   ScriptExecutionCallbackHelper callback_helper(context);
   v8::Local<v8::Function> function =
       v8::Function::New(context, callback).ToLocalChecked();
   main_frame->RequestExecuteV8Function(
       main_frame->MainWorldScriptContext(), function,
       v8::Undefined(context->GetIsolate()), 0, nullptr, &callback_helper);
 
   RunPendingTasks();
   EXPECT_FALSE(callback_helper.DidComplete());
 
   main_frame->GetFrame()->GetDocument()->ResumeScheduledTasks();
   RunPendingTasks();
   EXPECT_TRUE(callback_helper.DidComplete());
   EXPECT_EQ(true, callback_helper.BoolValue());
 }
 
 TEST_P(ParameterizedWebFrameTest, IframeScriptRemovesSelf) {
   RegisterMockedHttpURLLoad("single_iframe.html");
   RegisterMockedHttpURLLoad("visible_iframe.html");
 
   FrameTestHelpers::WebViewHelper web_view_helper;
   web_view_helper.InitializeAndLoad(base_url_ + "single_iframe.html", true);
 
   v8::HandleScope scope(v8::Isolate::GetCurrent());
   ScriptExecutionCallbackHelper callback_helper(
-      web_view_helper.WebView()->MainFrame()->MainWorldScriptContext());
+      web_view_helper.WebView()->MainFrameImpl()->MainWorldScriptContext());
   web_view_helper.WebView()
       ->MainFrame()
       ->FirstChild()
       ->ToWebLocalFrame()
       ->RequestExecuteScriptAndReturnValue(
           WebScriptSource(WebString(
               "var iframe = "
               "window.top.document.getElementsByTagName('iframe')[0]; "
               "window.top.document.body.removeChild(iframe); 'hello';")),
           false, &callback_helper);
@@ skipping 3949 matching lines @@
   Vector<std::unique_ptr<ContextLifetimeTestWebFrameClient::Notification>>
       create_notifications;
   Vector<std::unique_ptr<ContextLifetimeTestWebFrameClient::Notification>>
       release_notifications;
   ContextLifetimeTestMainFrameClient web_frame_client(create_notifications,
                                                       release_notifications);
   FrameTestHelpers::WebViewHelper web_view_helper;
   web_view_helper.InitializeAndLoad(
       base_url_ + "context_notifications_test.html", true, &web_frame_client);
 
-  WebFrame* main_frame = web_view_helper.WebView()->MainFrame();
+  WebLocalFrameBase* main_frame = web_view_helper.WebView()->MainFrameImpl();
   WebFrame* child_frame = main_frame->FirstChild();
 
   ASSERT_EQ(2u, create_notifications.size());
   EXPECT_EQ(0u, release_notifications.size());
 
   auto& first_create_notification = create_notifications[0];
   auto& second_create_notification = create_notifications[1];
 
   EXPECT_EQ(main_frame, first_create_notification->frame);
   EXPECT_EQ(main_frame->MainWorldScriptContext(),
             first_create_notification->context);
   EXPECT_EQ(0, first_create_notification->world_id);
 
   EXPECT_EQ(child_frame, second_create_notification->frame);
-  EXPECT_EQ(child_frame->MainWorldScriptContext(),
+  EXPECT_EQ(child_frame->ToWebLocalFrame()->MainWorldScriptContext(),
             second_create_notification->context);
   EXPECT_EQ(0, second_create_notification->world_id);
 
   // Close the view. We should get two release notifications that are exactly
   // the same as the create ones, in reverse order.
   web_view_helper.Reset();
 
   ASSERT_EQ(2u, release_notifications.size());
   auto& first_release_notification = release_notifications[0];
   auto& second_release_notification = release_notifications[1];
@@ skipping 28 matching lines @@
 
   // The two release notifications we got should be exactly the same as the
   // first two create notifications.
   for (size_t i = 0; i < release_notifications.size(); ++i) {
     EXPECT_TRUE(release_notifications[i]->Equals(
         create_notifications[create_notifications.size() - 3 - i].get()));
   }
 
   // The last two create notifications should be for the current frames and
   // context.
-  WebFrame* main_frame = web_view_helper.WebView()->MainFrame();
+  WebLocalFrameBase* main_frame = web_view_helper.WebView()->MainFrameImpl();
   WebFrame* child_frame = main_frame->FirstChild();
   auto& first_refresh_notification = create_notifications[2];
   auto& second_refresh_notification = create_notifications[3];
 
   EXPECT_EQ(main_frame, first_refresh_notification->frame);
   EXPECT_EQ(main_frame->MainWorldScriptContext(),
             first_refresh_notification->context);
   EXPECT_EQ(0, first_refresh_notification->world_id);
 
   EXPECT_EQ(child_frame, second_refresh_notification->frame);
-  EXPECT_EQ(child_frame->MainWorldScriptContext(),
+  EXPECT_EQ(child_frame->ToWebLocalFrame()->MainWorldScriptContext(),
             second_refresh_notification->context);
   EXPECT_EQ(0, second_refresh_notification->world_id);
 }
 
 TEST_P(ParameterizedWebFrameTest, ContextNotificationsIsolatedWorlds) {
   v8::Isolate* isolate = v8::Isolate::GetCurrent();
   v8::HandleScope handle_scope(isolate);
 
   RegisterMockedHttpURLLoad("context_notifications_test.html");
   RegisterMockedHttpURLLoad("context_notifications_test_frame.html");
@@ skipping 18 matching lines @@
       isolated_world_id, &script_source, num_sources);
 
   // We should now have a new create notification.
   ASSERT_EQ(1u, create_notifications.size());
   auto& notification = create_notifications[0];
   ASSERT_EQ(isolated_world_id, notification->world_id);
   ASSERT_EQ(web_view_helper.WebView()->MainFrame(), notification->frame);
 
   // We don't have an API to enumarate isolated worlds for a frame, but we can
   // at least assert that the context we got is *not* the main world's context.
-  ASSERT_NE(web_view_helper.WebView()->MainFrame()->MainWorldScriptContext(),
-            v8::Local<v8::Context>::New(isolate, notification->context));
+  ASSERT_NE(
+      web_view_helper.WebView()->MainFrameImpl()->MainWorldScriptContext(),
+      v8::Local<v8::Context>::New(isolate, notification->context));
 
   web_view_helper.Reset();
 
   // We should have gotten three release notifications (one for each of the
   // frames, plus one for the isolated context).
   ASSERT_EQ(3u, release_notifications.size());
 
   // And one of them should be exactly the same as the create notification for
   // the isolated context.
   int match_count = 0;
@@ skipping 5232 matching lines @@
     client_.screen_info_.rect = WebRect(0, 0, size.width, size.height);
     client_.screen_info_.available_rect = client_.screen_info_.rect;
     web_view_helper_.Resize(size);
     EXPECT_EQ(expected_size, DumpSize("test"));
   }
 
   String DumpSize(const String& id) {
     String code = "dumpSize('" + id + "')";
     v8::HandleScope scope(v8::Isolate::GetCurrent());
     ScriptExecutionCallbackHelper callback_helper(
-        web_view_helper_.WebView()->MainFrame()->MainWorldScriptContext());
+        web_view_helper_.WebView()->MainFrameImpl()->MainWorldScriptContext());
     web_view_helper_.WebView()
         ->MainFrameImpl()
         ->RequestExecuteScriptAndReturnValue(WebScriptSource(WebString(code)),
                                              false, &callback_helper);
     RunPendingTasks();
     EXPECT_TRUE(callback_helper.DidComplete());
     return callback_helper.StringValue();
   }
 
   FixedLayoutTestWebViewClient client_;
@@ skipping 2228 matching lines @@
     if (obj->IsText()) {
       LayoutText* layout_text = ToLayoutText(obj);
       text = layout_text->GetText();
       break;
     }
   }
   EXPECT_EQ("foo alt", text.Utf8());
 }
 
 }  // namespace blink