chrome/renderer/chrome_content_renderer_client_browsertest.cc

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/renderer/chrome_content_renderer_client.h"
   6
   7 #include <string>
   8 #include <vector>
   9
  10 #include "base/command_line.h"
  11 #include "base/strings/utf_string_conversions.h"
  12 #include "base/thread_task_runner_handle.h"
  13 #include "chrome/common/chrome_switches.h"
  14 #include "chrome/common/render_messages.h"
  15 #include "chrome/grit/generated_resources.h"
  16 #include "chrome/renderer/chrome_content_renderer_client.h"
  17 #include "chrome/renderer/plugins/shadow_dom_plugin_placeholder.h"
  18 #include "chrome/test/base/chrome_render_view_test.h"
  19 #include "content/public/common/content_constants.h"
  20 #include "content/public/renderer/render_frame.h"
  21 #include "content/public/renderer/render_view.h"
  22 #include "content/public/test/mock_render_thread.h"
  23 #include "ipc/ipc_listener.h"
  24 #include "ipc/ipc_sender.h"
  25 #include "ipc/ipc_test_sink.h"
  26 #include "testing/gmock/include/gmock/gmock.h"
  27 #include "third_party/WebKit/public/web/WebLocalFrame.h"
  28 #include "third_party/WebKit/public/web/WebPluginParams.h"
  29 #include "ui/base/l10n/l10n_util.h"
  30 #include "url/gurl.h"
  31
  32 using testing::_;
  33 using testing::SetArgPointee;
  34
  35 typedef ChromeRenderViewTest InstantProcessNavigationTest;
  36
  37 // Tests that renderer-initiated navigations from an Instant render process get
  38 // bounced back to the browser to be rebucketed into a non-Instant renderer if
  39 // necessary.
  40 TEST_F(InstantProcessNavigationTest, ForkForNavigationsFromInstantProcess) {
  41   base::CommandLine::ForCurrentProcess()->AppendSwitch(
  42       switches::kInstantProcess);
  43   bool unused;
  44   ChromeContentRendererClient* client =
  45       static_cast<ChromeContentRendererClient*>(content_renderer_client_.get());
  46   EXPECT_TRUE(client->ShouldFork(
  47       GetMainFrame(), GURL("http://foo"), "GET", false, false, &unused));
  48 }
  49
  50 // Tests that renderer-initiated navigations from a non-Instant render process
  51 // to potentially Instant URLs get bounced back to the browser to be rebucketed
  52 // into an Instant renderer if necessary.
  53 TEST_F(InstantProcessNavigationTest, ForkForNavigationsToSearchURLs) {
  54   ChromeContentRendererClient* client =
  55       static_cast<ChromeContentRendererClient*>(content_renderer_client_.get());
  56   chrome_render_thread_->set_io_message_loop_proxy(
  57       base::ThreadTaskRunnerHandle::Get());
  58   client->RenderThreadStarted();
  59   std::vector<GURL> search_urls;
  60   search_urls.push_back(GURL("http://example.com/search"));
  61   chrome_render_thread_->Send(new ChromeViewMsg_SetSearchURLs(
  62       search_urls, GURL("http://example.com/newtab")));
  63   bool unused;
  64   EXPECT_TRUE(client->ShouldFork(
  65       GetMainFrame(), GURL("http://example.com/newtab"), "GET", false, false,
  66       &unused));
  67   EXPECT_TRUE(client->ShouldFork(
  68       GetMainFrame(), GURL("http://example.com/search?q=foo"), "GET", false,
  69       false, &unused));
  70   EXPECT_FALSE(client->ShouldFork(
  71       GetMainFrame(), GURL("http://example.com/"), "GET", false, false,
  72       &unused));
  73 }
  74
  75 namespace {
  76
  77 // Intercepts plugin info IPCs for a mock render thread within its scope,
  78 // and allows tests to mock the response to each request.
  79 class ScopedMockPluginInfoFilter : public IPC::Listener, public IPC::Sender {
  80  public:
  81   explicit ScopedMockPluginInfoFilter(
  82       content::MockRenderThread* mock_render_thread)
  83       : sink_(mock_render_thread->sink()), sender_(mock_render_thread) {
  84     sink_.AddFilter(this);
  85   }
  86   ~ScopedMockPluginInfoFilter() override { sink_.RemoveFilter(this); }
  87
  88   bool OnMessageReceived(const IPC::Message& message) override {
  89     IPC_BEGIN_MESSAGE_MAP(ScopedMockPluginInfoFilter, message)
  90       IPC_MESSAGE_HANDLER(ChromeViewHostMsg_GetPluginInfo, OnGetPluginInfo)
  91       IPC_MESSAGE_UNHANDLED(return false)
  92     IPC_END_MESSAGE_MAP()
  93     return true;
  94   }
  95
  96   bool Send(IPC::Message* message) override { return sender_->Send(message); }
  97
  98   MOCK_METHOD5(OnGetPluginInfo,
  99                void(int render_frame_id,
 100                     const GURL& url,
 101                     const GURL& top_origin_url,
 102                     const std::string& mime_type,
 103                     ChromeViewHostMsg_GetPluginInfo_Output* output));
 104
 105  private:
 106   IPC::TestSink& sink_;
 107   IPC::Sender* sender_;
 108   DISALLOW_COPY_AND_ASSIGN(ScopedMockPluginInfoFilter);
 109 };
 110
 111 }  // namespace
 112
 113 class CreatePluginPlaceholderTest : public ChromeRenderViewTest {
 114  protected:
 115   void SetUp() override {
 116     ChromeRenderViewTest::SetUp();
 117     base::CommandLine::ForCurrentProcess()->AppendSwitch(
 118         switches::kEnablePluginPlaceholderShadowDom);
 119   }
 120
 121   content::RenderFrame* GetMainRenderFrame() const {
 122     return view_->GetMainRenderFrame();
 123   }
 124
 125   int GetRoutingID() const { return GetMainRenderFrame()->GetRoutingID(); }
 126 };
 127
 128 TEST_F(CreatePluginPlaceholderTest, MissingPlugin) {
 129   GURL url("http://www.example.com/example.swf");
 130   std::string mime_type("application/x-shockwave-flash");
 131
 132   blink::WebPluginParams params;
 133   params.url = url;
 134   params.mimeType = base::ASCIIToUTF16(mime_type);
 135
 136   ChromeViewHostMsg_GetPluginInfo_Output output;
 137   output.status = ChromeViewHostMsg_GetPluginInfo_Status::kNotFound;
 138
 139   ScopedMockPluginInfoFilter filter(render_thread_.get());
 140 #if defined(ENABLE_PLUGINS)
 141   EXPECT_CALL(filter, OnGetPluginInfo(GetRoutingID(), url, _, mime_type, _))
 142       .WillOnce(SetArgPointee<4>(output));
 143 #endif
 144
 145   scoped_ptr<blink::WebPluginPlaceholder> placeholder =
 146       content_renderer_client_->CreatePluginPlaceholder(
 147           GetMainRenderFrame(), GetMainFrame(), params);
 148   ASSERT_NE(nullptr, placeholder);
 149   EXPECT_EQ(l10n_util::GetStringUTF16(IDS_PLUGIN_NOT_SUPPORTED),
 150             placeholder->message());
 151 }
 152
 153 TEST_F(CreatePluginPlaceholderTest, PluginFound) {
 154   GURL url("http://www.example.com/example.swf");
 155   std::string mime_type(content::kFlashPluginSwfMimeType);
 156
 157   blink::WebPluginParams params;
 158   params.url = url;
 159   params.mimeType = base::ASCIIToUTF16(mime_type);
 160
 161   ChromeViewHostMsg_GetPluginInfo_Output output;
 162   output.status = ChromeViewHostMsg_GetPluginInfo_Status::kAllowed;
 163
 164   ScopedMockPluginInfoFilter filter(render_thread_.get());
 165 #if defined(ENABLE_PLUGINS)
 166   EXPECT_CALL(filter, OnGetPluginInfo(GetRoutingID(), url, _, mime_type, _))
 167       .WillOnce(SetArgPointee<4>(output));
 168 #endif
 169
 170   scoped_ptr<blink::WebPluginPlaceholder> placeholder =
 171       content_renderer_client_->CreatePluginPlaceholder(
 172           GetMainRenderFrame(), GetMainFrame(), params);
 173   EXPECT_EQ(nullptr, placeholder);
 174 }