Add printing tests to ensure window.print() works for subframes.

chrome/browser/printing/print_view_manager.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 "chrome/browser/printing/print_view_manager.h"
 
 #include <map>
 #include <utility>
 
 #include "base/bind.h"
@@ skipping 36 matching lines @@
   }
 
   ChromePluginServiceFilter::GetInstance()->OverridePluginForFrame(
       render_process_id, render_frame_id, GURL(), pdf_plugin);
 }
 
 }  // namespace
 
 namespace printing {
 
+bool PrintViewManager::TestDelegate::ScriptedPrint(
+    const content::RenderFrameHost* render_frame_host) {
+  return true;
+}
+
 PrintViewManager::PrintViewManager(content::WebContents* web_contents)
-    : PrintViewManagerBase(web_contents),
-      print_preview_state_(NOT_PREVIEWING),
-      print_preview_rfh_(nullptr),
-      scripted_print_preview_rph_(nullptr) {
+    : PrintViewManagerBase(web_contents) {
   if (PrintPreviewDialogController::IsPrintPreviewDialog(web_contents)) {
     EnableInternalPDFPluginForContents(
         web_contents->GetRenderProcessHost()->GetID(),
         web_contents->GetMainFrame()->GetRoutingID());
   }
 }
 
 PrintViewManager::~PrintViewManager() {
   DCHECK_EQ(NOT_PREVIEWING, print_preview_state_);
 }
@@ skipping 121 matching lines @@
     return;
   }
 
   PrintPreviewDialogController* dialog_controller =
       PrintPreviewDialogController::GetInstance();
   if (!dialog_controller) {
     rfh->Send(reply_msg);
     return;
   }
 
+  if (test_delegate_ && !test_delegate_->ScriptedPrint(rfh)) {
+    rfh->Send(reply_msg);
+    return;
+  }
+
   DCHECK(!print_preview_rfh_);
   print_preview_rfh_ = rfh;
   print_preview_state_ = SCRIPTED_PREVIEW;
   map[rph] = base::Bind(&PrintViewManager::OnScriptedPrintPreviewReply,
                         base::Unretained(this), reply_msg);
   scripted_print_preview_rph_ = rph;
 }
 
 void PrintViewManager::OnShowScriptedPrintPreview(content::RenderFrameHost* rfh,
                                                   bool source_is_modifiable) {
-  DCHECK(print_preview_rfh_);
   if (rfh != print_preview_rfh_)
     return;
 
   PrintPreviewDialogController* dialog_controller =
       PrintPreviewDialogController::GetInstance();
   if (!dialog_controller) {
     PrintPreviewDone();
     return;
   }
 
@@ skipping 19 matching lines @@
         PrintHostMsg_SetupScriptedPrintPreview, OnSetupScriptedPrintPreview)
     IPC_MESSAGE_HANDLER(PrintHostMsg_ShowScriptedPrintPreview,
                         OnShowScriptedPrintPreview)
     IPC_MESSAGE_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP()
 
   return handled ||
          PrintViewManagerBase::OnMessageReceived(message, render_frame_host);
 }
 
+void PrintViewManager::SetTestDelegate(TestDelegate* delegate) {
+  test_delegate_ = delegate;
+}
+
 }  // namespace printing