OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chrome/renderer/chrome_content_renderer_client.h" | 5 #include "chrome/renderer/chrome_content_renderer_client.h" |
6 | 6 |
7 #include <string> | 7 #include <string> |
8 #include <vector> | 8 #include <vector> |
9 | 9 |
10 #include "base/command_line.h" | 10 #include "base/command_line.h" |
11 #include "base/strings/utf_string_conversions.h" | 11 #include "base/strings/utf_string_conversions.h" |
12 #include "base/thread_task_runner_handle.h" | 12 #include "base/thread_task_runner_handle.h" |
13 #include "chrome/common/chrome_switches.h" | 13 #include "chrome/common/chrome_switches.h" |
14 #include "chrome/common/render_messages.h" | 14 #include "chrome/common/render_messages.h" |
15 #include "chrome/grit/generated_resources.h" | 15 #include "chrome/grit/generated_resources.h" |
16 #include "chrome/renderer/chrome_content_renderer_client.h" | 16 #include "chrome/renderer/chrome_content_renderer_client.h" |
| 17 #include "chrome/renderer/plugins/shadow_dom_plugin_placeholder.h" |
17 #include "chrome/test/base/chrome_render_view_test.h" | 18 #include "chrome/test/base/chrome_render_view_test.h" |
18 #include "content/public/common/content_constants.h" | 19 #include "content/public/common/content_constants.h" |
19 #include "content/public/renderer/render_frame.h" | 20 #include "content/public/renderer/render_frame.h" |
20 #include "content/public/renderer/render_view.h" | 21 #include "content/public/renderer/render_view.h" |
21 #include "content/public/test/mock_render_thread.h" | 22 #include "content/public/test/mock_render_thread.h" |
22 #include "ipc/ipc_listener.h" | 23 #include "ipc/ipc_listener.h" |
23 #include "ipc/ipc_sender.h" | 24 #include "ipc/ipc_sender.h" |
24 #include "ipc/ipc_test_sink.h" | 25 #include "ipc/ipc_test_sink.h" |
25 #include "testing/gmock/include/gmock/gmock.h" | 26 #include "testing/gmock/include/gmock/gmock.h" |
26 #include "third_party/WebKit/public/web/WebLocalFrame.h" | 27 #include "third_party/WebKit/public/web/WebLocalFrame.h" |
(...skipping 36 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
63 EXPECT_TRUE(client->ShouldFork( | 64 EXPECT_TRUE(client->ShouldFork( |
64 GetMainFrame(), GURL("http://example.com/newtab"), "GET", false, false, | 65 GetMainFrame(), GURL("http://example.com/newtab"), "GET", false, false, |
65 &unused)); | 66 &unused)); |
66 EXPECT_TRUE(client->ShouldFork( | 67 EXPECT_TRUE(client->ShouldFork( |
67 GetMainFrame(), GURL("http://example.com/search?q=foo"), "GET", false, | 68 GetMainFrame(), GURL("http://example.com/search?q=foo"), "GET", false, |
68 false, &unused)); | 69 false, &unused)); |
69 EXPECT_FALSE(client->ShouldFork( | 70 EXPECT_FALSE(client->ShouldFork( |
70 GetMainFrame(), GURL("http://example.com/"), "GET", false, false, | 71 GetMainFrame(), GURL("http://example.com/"), "GET", false, false, |
71 &unused)); | 72 &unused)); |
72 } | 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 } |
OLD | NEW |