Index: chrome/renderer/chrome_content_renderer_client_browsertest.cc |
diff --git a/chrome/renderer/chrome_content_renderer_client_browsertest.cc b/chrome/renderer/chrome_content_renderer_client_browsertest.cc |
index 3424e5c34e623a07ad0cebc1ddd35ce21995cb6f..f3735391789e65afca5e270d2f9a50f62745a7a4 100644 |
--- a/chrome/renderer/chrome_content_renderer_client_browsertest.cc |
+++ b/chrome/renderer/chrome_content_renderer_client_browsertest.cc |
@@ -14,6 +14,7 @@ |
#include "chrome/common/render_messages.h" |
#include "chrome/grit/generated_resources.h" |
#include "chrome/renderer/chrome_content_renderer_client.h" |
+#include "chrome/renderer/plugins/shadow_dom_plugin_placeholder.h" |
#include "chrome/test/base/chrome_render_view_test.h" |
#include "content/public/common/content_constants.h" |
#include "content/public/renderer/render_frame.h" |
@@ -70,3 +71,104 @@ |
GetMainFrame(), GURL("http://example.com/"), "GET", false, false, |
&unused)); |
} |
+ |
+namespace { |
+ |
+// Intercepts plugin info IPCs for a mock render thread within its scope, |
+// and allows tests to mock the response to each request. |
+class ScopedMockPluginInfoFilter : public IPC::Listener, public IPC::Sender { |
+ public: |
+ explicit ScopedMockPluginInfoFilter( |
+ content::MockRenderThread* mock_render_thread) |
+ : sink_(mock_render_thread->sink()), sender_(mock_render_thread) { |
+ sink_.AddFilter(this); |
+ } |
+ ~ScopedMockPluginInfoFilter() override { sink_.RemoveFilter(this); } |
+ |
+ bool OnMessageReceived(const IPC::Message& message) override { |
+ IPC_BEGIN_MESSAGE_MAP(ScopedMockPluginInfoFilter, message) |
+ IPC_MESSAGE_HANDLER(ChromeViewHostMsg_GetPluginInfo, OnGetPluginInfo) |
+ IPC_MESSAGE_UNHANDLED(return false) |
+ IPC_END_MESSAGE_MAP() |
+ return true; |
+ } |
+ |
+ bool Send(IPC::Message* message) override { return sender_->Send(message); } |
+ |
+ MOCK_METHOD5(OnGetPluginInfo, |
+ void(int render_frame_id, |
+ const GURL& url, |
+ const GURL& top_origin_url, |
+ const std::string& mime_type, |
+ ChromeViewHostMsg_GetPluginInfo_Output* output)); |
+ |
+ private: |
+ IPC::TestSink& sink_; |
+ IPC::Sender* sender_; |
+ DISALLOW_COPY_AND_ASSIGN(ScopedMockPluginInfoFilter); |
+}; |
+ |
+} // namespace |
+ |
+class CreatePluginPlaceholderTest : public ChromeRenderViewTest { |
+ protected: |
+ void SetUp() override { |
+ ChromeRenderViewTest::SetUp(); |
+ base::CommandLine::ForCurrentProcess()->AppendSwitch( |
+ switches::kEnablePluginPlaceholderShadowDom); |
+ } |
+ |
+ content::RenderFrame* GetMainRenderFrame() const { |
+ return view_->GetMainRenderFrame(); |
+ } |
+ |
+ int GetRoutingID() const { return GetMainRenderFrame()->GetRoutingID(); } |
+}; |
+ |
+TEST_F(CreatePluginPlaceholderTest, MissingPlugin) { |
+ GURL url("http://www.example.com/example.swf"); |
+ std::string mime_type("application/x-shockwave-flash"); |
+ |
+ blink::WebPluginParams params; |
+ params.url = url; |
+ params.mimeType = base::ASCIIToUTF16(mime_type); |
+ |
+ ChromeViewHostMsg_GetPluginInfo_Output output; |
+ output.status = ChromeViewHostMsg_GetPluginInfo_Status::kNotFound; |
+ |
+ ScopedMockPluginInfoFilter filter(render_thread_.get()); |
+#if defined(ENABLE_PLUGINS) |
+ EXPECT_CALL(filter, OnGetPluginInfo(GetRoutingID(), url, _, mime_type, _)) |
+ .WillOnce(SetArgPointee<4>(output)); |
+#endif |
+ |
+ scoped_ptr<blink::WebPluginPlaceholder> placeholder = |
+ content_renderer_client_->CreatePluginPlaceholder( |
+ GetMainRenderFrame(), GetMainFrame(), params); |
+ ASSERT_NE(nullptr, placeholder); |
+ EXPECT_EQ(l10n_util::GetStringUTF16(IDS_PLUGIN_NOT_SUPPORTED), |
+ placeholder->message()); |
+} |
+ |
+TEST_F(CreatePluginPlaceholderTest, PluginFound) { |
+ GURL url("http://www.example.com/example.swf"); |
+ std::string mime_type(content::kFlashPluginSwfMimeType); |
+ |
+ blink::WebPluginParams params; |
+ params.url = url; |
+ params.mimeType = base::ASCIIToUTF16(mime_type); |
+ |
+ ChromeViewHostMsg_GetPluginInfo_Output output; |
+ output.status = ChromeViewHostMsg_GetPluginInfo_Status::kAllowed; |
+ |
+ ScopedMockPluginInfoFilter filter(render_thread_.get()); |
+#if defined(ENABLE_PLUGINS) |
+ EXPECT_CALL(filter, OnGetPluginInfo(GetRoutingID(), url, _, mime_type, _)) |
+ .WillOnce(SetArgPointee<4>(output)); |
+#endif |
+ |
+ scoped_ptr<blink::WebPluginPlaceholder> placeholder = |
+ content_renderer_client_->CreatePluginPlaceholder( |
+ GetMainRenderFrame(), GetMainFrame(), params); |
+ EXPECT_EQ(nullptr, placeholder); |
+} |