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

chrome/browser/printing/print_view_manager.h

 // 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.
 
 #ifndef CHROME_BROWSER_PRINTING_PRINT_VIEW_MANAGER_H_
 #define CHROME_BROWSER_PRINTING_PRINT_VIEW_MANAGER_H_
 
 #include "base/macros.h"
 #include "chrome/browser/printing/print_view_manager_base.h"
 #include "content/public/browser/web_contents_user_data.h"
 #include "printing/features/features.h"
+#include "url/gurl.h"
 
 namespace content {
 class RenderFrameHost;
 class RenderProcessHost;
 }
 
 namespace printing {
 
 // Manages the print commands for a WebContents.
 class PrintViewManager : public PrintViewManagerBase,
                          public content::WebContentsUserData<PrintViewManager> {
  public:
+  class TestDelegate {
+   public:
+    virtual ~TestDelegate() {}
+
+    // Called when the given RFH performs a JS window.print() call. Returns true
+    // to continue processing window.print() normally, or false to immediately
+    // issue an IPC reply.
+    virtual bool ScriptedPrint(
+        const content::RenderFrameHost* render_frame_host);
+  };
+
   ~PrintViewManager() override;
 
 #if BUILDFLAG(ENABLE_BASIC_PRINTING)
   // Same as PrintNow(), but for the case where a user prints with the system
   // dialog from print preview.
   // |dialog_shown_callback| is called when the print dialog is shown.
   bool PrintForSystemDialogNow(const base::Closure& dialog_shown_callback);
 
   // Same as PrintNow(), but for the case where a user press "ctrl+shift+p" to
   // show the native system dialog. This can happen from both initiator and
@@ skipping 16 matching lines @@
   void PrintPreviewDone();
 
   // content::WebContentsObserver implementation.
   void RenderFrameCreated(content::RenderFrameHost* render_frame_host) override;
   void RenderFrameDeleted(content::RenderFrameHost* render_frame_host) override;
   bool OnMessageReceived(const IPC::Message& message,
                          content::RenderFrameHost* render_frame_host) override;
 
   content::RenderFrameHost* print_preview_rfh() { return print_preview_rfh_; }
 
+  // Used in tests to observe and/or override behavior.
+  void SetTestDelegate(TestDelegate* delegate);
+
  private:
   explicit PrintViewManager(content::WebContents* web_contents);
   friend class content::WebContentsUserData<PrintViewManager>;
 
   enum PrintPreviewState {
     NOT_PREVIEWING,
     USER_INITIATED_PREVIEW,
     SCRIPTED_PREVIEW,
   };
 
   // IPC Message handlers.
   void OnDidShowPrintDialog(content::RenderFrameHost* rfh);
   void OnSetupScriptedPrintPreview(content::RenderFrameHost* rfh,
                                    IPC::Message* reply_msg);
   void OnShowScriptedPrintPreview(content::RenderFrameHost* rfh,
                                   bool source_is_modifiable);
   void OnScriptedPrintPreviewReply(IPC::Message* reply_msg);
 
   base::Closure on_print_dialog_shown_callback_;
 
   // Current state of print preview for this view.
-  PrintPreviewState print_preview_state_;
+  PrintPreviewState print_preview_state_ = NOT_PREVIEWING;
 
   // The current RFH that is print previewing. It should be a nullptr when
   // |print_preview_state_| is NOT_PREVIEWING.
-  content::RenderFrameHost* print_preview_rfh_;
+  content::RenderFrameHost* print_preview_rfh_ = nullptr;
 
   // Keeps track of the pending callback during scripted print preview.
-  content::RenderProcessHost* scripted_print_preview_rph_;
+  content::RenderProcessHost* scripted_print_preview_rph_ = nullptr;
+
+  TestDelegate* test_delegate_ = nullptr;
 
   DISALLOW_COPY_AND_ASSIGN(PrintViewManager);
 };
 
 }  // namespace printing
 
 #endif  // CHROME_BROWSER_PRINTING_PRINT_VIEW_MANAGER_H_