Retrieve NTP thumbnail URLs from the right frame

chrome/renderer/chrome_content_renderer_client_browsertest.cc

 // Copyright 2013 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 "chrome/renderer/chrome_content_renderer_client.h"
 
 #include <string>
 #include <vector>
 
 #include "base/command_line.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/threading/thread_task_runner_handle.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/render_messages.h"
+#include "chrome/renderer/searchbox/searchbox.h"
 #include "chrome/test/base/chrome_render_view_test.h"
 #include "chrome/test/base/in_process_browser_test.h"
 #include "chrome/test/base/ui_test_utils.h"
 #include "content/public/common/content_constants.h"
 #include "content/public/renderer/render_frame.h"
 #include "content/public/renderer/render_view.h"
 #include "content/public/test/browser_test_utils.h"
 #include "content/public/test/mock_render_thread.h"
 #include "content/public/test/test_utils.h"
 #include "ipc/ipc_listener.h"
 #include "ipc/ipc_sender.h"
 #include "ipc/ipc_test_sink.h"
 #include "net/dns/mock_host_resolver.h"
 #include "net/test/embedded_test_server/http_request.h"
 #include "net/test/embedded_test_server/http_response.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "third_party/WebKit/public/web/WebLocalFrame.h"
 #include "third_party/WebKit/public/web/WebPluginParams.h"
 #include "url/gurl.h"
 
-using testing::_;
-using testing::SetArgPointee;
-
-typedef ChromeRenderViewTest InstantProcessNavigationTest;
+using InstantProcessNavigationTest = ChromeRenderViewTest;
+using ChromeContentRendererClientSearchBoxTest = ChromeRenderViewTest;
 
 const base::FilePath::CharType kDocRoot[] =
     FILE_PATH_LITERAL("chrome/test/data");
 
+const char kHtmlWithIframe[] ="<iframe srcdoc=\"Nothing here\"></iframe>";
+
 // Tests that renderer-initiated navigations from an Instant render process get
 // bounced back to the browser to be rebucketed into a non-Instant renderer if
 // necessary.
 TEST_F(InstantProcessNavigationTest, ForkForNavigationsFromInstantProcess) {
   base::CommandLine::ForCurrentProcess()->AppendSwitch(
       switches::kInstantProcess);
   bool unused;
   ChromeContentRendererClient* client =
       static_cast<ChromeContentRendererClient*>(content_renderer_client_.get());
   EXPECT_TRUE(client->ShouldFork(
@@ skipping 18 matching lines @@
       GetMainFrame(), GURL("http://example.com/newtab"), "GET", false, false,
       &unused));
   EXPECT_TRUE(client->ShouldFork(
       GetMainFrame(), GURL("http://example.com/search?q=foo"), "GET", false,
       false, &unused));
   EXPECT_FALSE(client->ShouldFork(
       GetMainFrame(), GURL("http://example.com/"), "GET", false, false,
       &unused));
 }
 
+TEST_F(ChromeContentRendererClientSearchBoxTest, RewriteThumbnailURL) {
+  // Instantiate a SearchBox for the main render frame.
+  content::RenderFrame* render_frame =
+      content::RenderFrame::FromWebFrame(GetMainFrame());
+  new SearchBox(render_frame);
+
+  // Load a page that contains an iframe.
+  LoadHTML(kHtmlWithIframe);
+
+  ChromeContentRendererClient client;
+
+  // Create a thumbnail URL containing the correct render view ID and an
+  // arbitrary instant restricted ID.
+  GURL thumbnail_url(base::StringPrintf(
+      "chrome-search:/thumb/%i/1",
+      render_frame->GetRenderView()->GetRoutingID()));
+
+  GURL result;
+  // Make sure the SearchBox rewrites a thumbnail request from the main frame.
+  EXPECT_TRUE(client.WillSendRequest(GetMainFrame(), ui::PAGE_TRANSITION_LINK,
+                                     blink::WebURL(thumbnail_url), &result));
+
+  // Make sure the SearchBox rewrites a thumbnail request from the iframe.
+  blink::WebFrame* child_frame = GetMainFrame()->firstChild();
+  ASSERT_TRUE(child_frame);
+  ASSERT_TRUE(child_frame->isWebLocalFrame());
+  blink::WebLocalFrame* local_child =
+      static_cast<blink::WebLocalFrame*>(child_frame);
+  EXPECT_TRUE(client.WillSendRequest(local_child, ui::PAGE_TRANSITION_LINK,
+                                     blink::WebURL(thumbnail_url), &result));
+}
+
 // The tests below examine Youtube requests that use the Flash API and ensure
 // that the requests have been modified to instead use HTML5. The tests also
 // check the MIME type of the request to ensure that it is "text/html".
 namespace {
 
 struct FlashEmbedsTestData {
   std::string name;
   std::string host;
   std::string path;
   std::string type;
@@ skipping 110 matching lines @@
      "http://" + GetParam().host + ":" + port + GetParam().path;
   EXPECT_TRUE(ExecuteScript(web_contents,
      "appendDataEmbedToDOM('" + video_url + "','" + GetParam().type + "');"));
   WaitForYouTubeRequest();
 }
 
 INSTANTIATE_TEST_CASE_P(
     FlashEmbeds,
     ChromeContentRendererClientBrowserTest,
     ::testing::ValuesIn(kFlashEmbedsTestData));