Move [Provisional]DataSource accessors/methods 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 4329 matching lines @@
 
   FrameTestHelpers::WebViewHelper web_view_helper;
   web_view_helper.Initialize();
   WebURLRequest request(ToKURL(base_url_ + "fixed_layout.html"));
   web_view_helper.WebView()->MainFrame()->LoadRequest(request);
   // start reload before first request is delivered.
   FrameTestHelpers::ReloadFrameBypassingCache(
       web_view_helper.WebView()->MainFrame());
 
   WebDataSource* data_source =
-      web_view_helper.WebView()->MainFrame()->DataSource();
+      web_view_helper.WebView()->MainFrameImpl()->DataSource();
   ASSERT_TRUE(data_source);
   EXPECT_EQ(ToKURL(base_url_ + "fixed_layout.html"),
             KURL(data_source->GetRequest().Url()));
 }
 
 TEST_P(ParameterizedWebFrameTest, AppendRedirects) {
   const std::string first_url = "about:blank";
   const std::string second_url = "http://internal.test";
 
   FrameTestHelpers::WebViewHelper web_view_helper;
   web_view_helper.InitializeAndLoad(first_url, true);
 
   WebDataSource* data_source =
-      web_view_helper.WebView()->MainFrame()->DataSource();
+      web_view_helper.WebView()->MainFrameImpl()->DataSource();
   ASSERT_TRUE(data_source);
   data_source->AppendRedirect(ToKURL(second_url));
 
   WebVector<WebURL> redirects;
   data_source->RedirectChain(redirects);
   ASSERT_EQ(2U, redirects.size());
   EXPECT_EQ(ToKURL(first_url), KURL(redirects[0]));
   EXPECT_EQ(ToKURL(second_url), KURL(redirects[1]));
 }
 
 TEST_P(ParameterizedWebFrameTest, IframeRedirect) {
   RegisterMockedHttpURLLoad("iframe_redirect.html");
   RegisterMockedHttpURLLoad("visible_iframe.html");
 
   FrameTestHelpers::WebViewHelper web_view_helper;
   web_view_helper.InitializeAndLoad(base_url_ + "iframe_redirect.html", true);
   // Pump pending requests one more time. The test page loads script that
   // navigates.
   FrameTestHelpers::PumpPendingRequestsForFrameToLoad(
       web_view_helper.WebView()->MainFrame());
 
   WebFrame* iframe =
       web_view_helper.WebView()->MainFrameImpl()->FindFrameByName(
           WebString::FromUTF8("ifr"));
-  ASSERT_TRUE(iframe);
-  WebDataSource* iframe_data_source = iframe->DataSource();
+  ASSERT_TRUE(iframe && iframe->IsWebLocalFrame());
+  WebDataSource* iframe_data_source = iframe->ToWebLocalFrame()->DataSource();
   ASSERT_TRUE(iframe_data_source);
   WebVector<WebURL> redirects;
   iframe_data_source->RedirectChain(redirects);
   ASSERT_EQ(2U, redirects.size());
   EXPECT_EQ(ToKURL("about:blank"), KURL(redirects[0]));
   EXPECT_EQ(ToKURL("http://internal.test/visible_iframe.html"),
             KURL(redirects[1]));
 }
 
 TEST_P(ParameterizedWebFrameTest, ClearFocusedNodeTest) {
@@ skipping 2933 matching lines @@
   // decidePolicyForNavigation should be called both for the original request
   // and the ctrl+click.
   EXPECT_EQ(2, web_frame_client.DecidePolicyCallCount());
 }
 
 TEST_P(ParameterizedWebFrameTest, BackToReload) {
   RegisterMockedHttpURLLoad("fragment_middle_click.html");
   FrameTestHelpers::WebViewHelper web_view_helper;
   web_view_helper.InitializeAndLoad(base_url_ + "fragment_middle_click.html",
                                     true);
-  WebFrame* frame = web_view_helper.WebView()->MainFrame();
+  WebLocalFrame* frame = web_view_helper.WebView()->MainFrameImpl();
   const FrameLoader& main_frame_loader =
       web_view_helper.WebView()->MainFrameImpl()->GetFrame()->Loader();
   Persistent<HistoryItem> first_item =
       main_frame_loader.GetDocumentLoader()->GetHistoryItem();
   EXPECT_TRUE(first_item);
 
   RegisterMockedHttpURLLoad("white-1x1.png");
   FrameTestHelpers::LoadFrame(frame, base_url_ + "white-1x1.png");
   EXPECT_NE(first_item.Get(),
             main_frame_loader.GetDocumentLoader()->GetHistoryItem());
@@ skipping 38 matching lines @@
   EXPECT_EQ(item.UrlString(), main_frame->GetDocument().Url().GetString());
   EXPECT_EQ(item.UrlString(), WebString(main_frame_loader.GetDocumentLoader()
                                             ->GetHistoryItem()
                                             ->UrlString()));
 }
 
 TEST_P(ParameterizedWebFrameTest, ReloadPost) {
   RegisterMockedHttpURLLoad("reload_post.html");
   FrameTestHelpers::WebViewHelper web_view_helper;
   web_view_helper.InitializeAndLoad(base_url_ + "reload_post.html", true);
-  WebFrame* frame = web_view_helper.WebView()->MainFrame();
+  WebLocalFrame* frame = web_view_helper.WebView()->MainFrameImpl();
 
   FrameTestHelpers::LoadFrame(web_view_helper.WebView()->MainFrame(),
                               "javascript:document.forms[0].submit()");
   // Pump requests one more time after the javascript URL has executed to
   // trigger the actual POST load request.
   FrameTestHelpers::PumpPendingRequestsForFrameToLoad(
       web_view_helper.WebView()->MainFrame());
   EXPECT_EQ(WebString::FromUTF8("POST"),
             frame->DataSource()->GetRequest().HttpMethod());
 
   FrameTestHelpers::ReloadFrame(frame);
   EXPECT_EQ(WebCachePolicy::kValidatingCacheData,
             frame->DataSource()->GetRequest().GetCachePolicy());
   EXPECT_EQ(kWebNavigationTypeFormResubmitted,
             frame->DataSource()->GetNavigationType());
 }
 
 TEST_P(ParameterizedWebFrameTest, LoadHistoryItemReload) {
   RegisterMockedHttpURLLoad("fragment_middle_click.html");
   FrameTestHelpers::WebViewHelper web_view_helper;
   web_view_helper.InitializeAndLoad(base_url_ + "fragment_middle_click.html",
                                     true);
-  WebFrame* frame = web_view_helper.WebView()->MainFrame();
+  WebLocalFrame* frame = web_view_helper.WebView()->MainFrameImpl();
   const FrameLoader& main_frame_loader =
       web_view_helper.WebView()->MainFrameImpl()->GetFrame()->Loader();
   Persistent<HistoryItem> first_item =
       main_frame_loader.GetDocumentLoader()->GetHistoryItem();
   EXPECT_TRUE(first_item);
 
   RegisterMockedHttpURLLoad("white-1x1.png");
   FrameTestHelpers::LoadFrame(frame, base_url_ + "white-1x1.png");
   EXPECT_NE(first_item.Get(),
             main_frame_loader.GetDocumentLoader()->GetHistoryItem());
@@ skipping 1285 matching lines @@
 
   EXPECT_TRUE(resource->IsLoaded());
 }
 
 TEST_P(ParameterizedWebFrameTest, ReloadBypassingCache) {
   // Check that a reload bypassing cache on a frame will result in the cache
   // policy of the request being set to ReloadBypassingCache.
   RegisterMockedHttpURLLoad("foo.html");
   FrameTestHelpers::WebViewHelper web_view_helper;
   web_view_helper.InitializeAndLoad(base_url_ + "foo.html", true);
-  WebFrame* frame = web_view_helper.WebView()->MainFrame();
+  WebLocalFrame* frame = web_view_helper.WebView()->MainFrameImpl();
   FrameTestHelpers::ReloadFrameBypassingCache(frame);
   EXPECT_EQ(WebCachePolicy::kBypassingCache,
             frame->DataSource()->GetRequest().GetCachePolicy());
 }
 
 static void NodeImageTestValidation(const IntSize& reference_bitmap_size,
                                     DragImage* drag_image) {
   // Prepare the reference bitmap.
   SkBitmap bitmap;
   bitmap.allocN32Pixels(reference_bitmap_size.Width(),
@@ skipping 3438 matching lines @@
     if (obj->IsText()) {
       LayoutText* layout_text = ToLayoutText(obj);
       text = layout_text->GetText();
       break;
     }
   }
   EXPECT_EQ("foo alt", text.Utf8());
 }
 
 }  // namespace blink