Enable PDF plugin for iframes within the print preview dialog.

chrome/browser/printing/print_preview_dialog_controller_browsertest.cc

 // Copyright (c) 2012 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 "base/bind_helpers.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/run_loop.h"
+#include "base/strings/utf_string_conversions.h"
+#include "chrome/browser/plugins/chrome_plugin_service_filter.h"
+#include "chrome/browser/plugins/plugin_prefs.h"
 #include "chrome/browser/printing/print_preview_dialog_controller.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/browser_commands.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
+#include "chrome/common/chrome_content_client.h"
 #include "chrome/common/url_constants.h"
 #include "chrome/test/base/in_process_browser_test.h"
 #include "chrome/test/base/ui_test_utils.h"
 #include "components/printing/common/print_messages.h"
+#include "content/public/browser/plugin_service.h"
+#include "content/public/browser/render_frame_host.h"
+#include "content/public/browser/render_process_host.h"
 #include "content/public/browser/web_contents_observer.h"
 #include "content/public/test/browser_test_utils.h"
 #include "ipc/ipc_message_macros.h"
 #include "url/gurl.h"
 
 using content::WebContents;
 using content::WebContentsObserver;
 
+namespace {
+
 class RequestPrintPreviewObserver : public WebContentsObserver {
  public:
   explicit RequestPrintPreviewObserver(WebContents* dialog)
       : WebContentsObserver(dialog) {
   }
   ~RequestPrintPreviewObserver() override {}
 
   void set_quit_closure(const base::Closure& quit_closure) {
     quit_closure_ = quit_closure;
   }
@@ skipping 55 matching lines @@
 
  private:
   // content::WebContentsObserver implementation.
   void WebContentsDestroyed() override { dialog_destroyed_ = true; }
 
   bool dialog_destroyed_;
 
   DISALLOW_COPY_AND_ASSIGN(PrintPreviewDialogDestroyedObserver);
 };
 
+bool GetPdfPluginInfo(content::WebPluginInfo* info) {
+  base::FilePath pdf_plugin_path = base::FilePath::FromUTF8Unsafe(
+      ChromeContentClient::kPDFPluginPath);
+  return content::PluginService::GetInstance()->GetPluginInfoByPath(
+      pdf_plugin_path, info);
+}
+
+const char kDummyPrintUrl[] = "chrome://print/dummy.pdf";
+
+void CountFrames(int* frame_count,
+                 content::RenderFrameHost* frame) {
+  ++(*frame_count);
+}
+
+void CheckPdfPluginForRenderFrame(content::RenderFrameHost* frame) {
+  content::WebPluginInfo pdf_plugin_info;
+  ASSERT_TRUE(GetPdfPluginInfo(&pdf_plugin_info));
+
+  ChromePluginServiceFilter* filter = ChromePluginServiceFilter::GetInstance();
+  EXPECT_TRUE(filter->IsPluginAvailable(
+      frame->GetProcess()->GetID(),
+      frame->GetRoutingID(),
+      nullptr,
+      GURL(kDummyPrintUrl),
+      GURL(),
+      &pdf_plugin_info));
+}
+
+}  // namespace
+
 class PrintPreviewDialogControllerBrowserTest : public InProcessBrowserTest {
  public:
   PrintPreviewDialogControllerBrowserTest() : initiator_(NULL) {}
   ~PrintPreviewDialogControllerBrowserTest() override {}
 
   WebContents* initiator() {
     return initiator_;
   }
 
   void PrintPreview() {
@@ skipping 46 matching lines @@
 // Test to verify that when a initiator navigates, we can create a new preview
 // dialog for the new tab contents.
 // http://crbug.com/377337
 #if defined(OS_WIN)
 #define MAYBE_NavigateFromInitiatorTab DISABLED_NavigateFromInitiatorTab
 #else
 #define MAYBE_NavigateFromInitiatorTab NavigateFromInitiatorTab
 #endif
 IN_PROC_BROWSER_TEST_F(PrintPreviewDialogControllerBrowserTest,
                        MAYBE_NavigateFromInitiatorTab) {
-  // print for the first time.
+  // Print for the first time.
   PrintPreview();
 
   // Get the preview dialog for the initiator tab.
   WebContents* preview_dialog = GetPrintPreviewDialog();
 
   // Check a new print preview dialog got created.
   ASSERT_TRUE(preview_dialog);
   ASSERT_NE(initiator(), preview_dialog);
 
   // Navigate in the initiator tab. Make sure navigating destroys the print
@@ skipping 15 matching lines @@
 // Test to verify that after reloading the initiator, it creates a new print
 // preview dialog.
 // http://crbug.com/377337
 #if defined(OS_WIN)
 #define MAYBE_ReloadInitiatorTab DISABLED_ReloadInitiatorTab
 #else
 #define MAYBE_ReloadInitiatorTab ReloadInitiatorTab
 #endif
 IN_PROC_BROWSER_TEST_F(PrintPreviewDialogControllerBrowserTest,
                        MAYBE_ReloadInitiatorTab) {
-  // print for the first time.
+  // Print for the first time.
   PrintPreview();
 
   WebContents* preview_dialog = GetPrintPreviewDialog();
 
   // Check a new print preview dialog got created.
   ASSERT_TRUE(preview_dialog);
   ASSERT_NE(initiator(), preview_dialog);
 
   // Reload the initiator. Make sure reloading destroys the print preview
   // dialog.
   PrintPreviewDialogDestroyedObserver dialog_destroyed_observer(preview_dialog);
   chrome::Reload(browser(), CURRENT_TAB);
   content::WaitForLoadStop(
       browser()->tab_strip_model()->GetActiveWebContents());
   ASSERT_TRUE(dialog_destroyed_observer.dialog_destroyed());
 
   // Try printing again.
   PrintPreview();
 
   // Create a preview dialog for the initiator tab.
   WebContents* new_preview_dialog = GetPrintPreviewDialog();
   EXPECT_TRUE(new_preview_dialog);
 }
+
+// Test to verify that after print preview works even when the PDF plugin is
+// disabled for webpages.
+IN_PROC_BROWSER_TEST_F(PrintPreviewDialogControllerBrowserTest,
+                       PdfPluginDisabled) {
+  // Print for the first time.
+  PrintPreview();
+
+  // Navigate in the initiator tab.
+  WebContents* preview_dialog = GetPrintPreviewDialog();
+  PrintPreviewDialogDestroyedObserver dialog_destroyed_observer(preview_dialog);
+  ui_test_utils::NavigateToURL(browser(), GURL(chrome::kChromeUINewTabURL));
+  ASSERT_TRUE(dialog_destroyed_observer.dialog_destroyed());
+
+  // Get the PDF plugin info.
+  content::WebPluginInfo pdf_plugin_info;
+  ASSERT_TRUE(GetPdfPluginInfo(&pdf_plugin_info));

[Lei Zhang, 2015/03/19 02:55:34]

This failed on the CrOS bot. I'm pretty sure CrOS has the PDF plugin, so I'm
attributing the failure to the plugin service not being fully initialized. I
need to find a way to do that...

+
+  // Disable the PDF plugin.
+  PluginPrefs::GetForProfile(browser()->profile())->EnablePluginGroup(
+      false, base::ASCIIToUTF16(ChromeContentClient::kPDFPluginName));
+
+  // Make sure it is actually disabled for webpages.
+  ChromePluginServiceFilter* filter = ChromePluginServiceFilter::GetInstance();
+  content::WebPluginInfo dummy_pdf_plugin_info = pdf_plugin_info;
+  EXPECT_FALSE(filter->IsPluginAvailable(
+      initiator()->GetRenderProcessHost()->GetID(),
+      initiator()->GetMainFrame()->GetRoutingID(),
+      nullptr,
+      GURL("http://google.com"),
+      GURL(),
+      &dummy_pdf_plugin_info));
+
+  // Print preview again.
+  PrintPreview();
+
+  // Check a new print preview dialog got created.
+  WebContents* new_preview_dialog = GetPrintPreviewDialog();
+  ASSERT_TRUE(new_preview_dialog);
+  ASSERT_NE(initiator(), new_preview_dialog);
+
+  // Wait until the <iframe> in the print preview renderer has loaded.
+  int frame_count;
+  do {
+    base::RunLoop run_loop;
+    base::MessageLoop::current()->PostDelayedTask(
+        FROM_HERE, run_loop.QuitClosure(), base::TimeDelta::FromSeconds(1));
+    run_loop.Run();
+
+    frame_count = 0;
+    new_preview_dialog->ForEachFrame(
+        base::Bind(&CountFrames, base::Unretained(&frame_count)));
+  } while (frame_count < 2);
+  ASSERT_EQ(2, frame_count);
+
+  // Make sure all the frames in the dialog has access to the PDF plugin.
+  new_preview_dialog->ForEachFrame(base::Bind(&CheckPdfPluginForRenderFrame));
+}