Added a test that currently is able to print to pdf.

chrome/browser/printing/print_preview_pdf_generated_browsertest.cc

Index: chrome/browser/printing/print_preview_pdf_generated_browsertest.cc
diff --git a/chrome/browser/printing/print_preview_pdf_generated_browsertest.cc b/chrome/browser/printing/print_preview_pdf_generated_browsertest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..cfd76b229d9e21f17c8974c2b36b48e666a83487
--- /dev/null
+++ b/chrome/browser/printing/print_preview_pdf_generated_browsertest.cc
@@ -0,0 +1,355 @@
+// 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 <string>
+#include <vector>
+
+#include "base/files/file_path.h"
+#include "base/logging.h"
+#include "base/memory/scoped_ptr.h"
+#include "base/path_service.h"
+#include "base/run_loop.h"
+#include "base/strings/string_util.h"
+#include "chrome/app/chrome_command_ids.h"
+#include "chrome/browser/net/referrer.h"
+#include "chrome/browser/printing/print_preview_dialog_controller.h"
+#include "chrome/browser/profiles/profile.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/browser/ui/webui/print_preview/print_preview_handler.h"
+#include "chrome/browser/ui/webui/print_preview/print_preview_ui.h"
+#include "chrome/browser/ui/webui/print_preview/sticky_settings.h"
+#include "chrome/common/chrome_paths.h"
+#include "chrome/common/print_messages.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 "content/public/browser/web_contents.h"
+#include "content/public/browser/web_ui_message_handler.h"
+#include "content/public/common/page_transition_types.h"
+#include "content/public/test/browser_test_utils.h"
+#include "content/public/test/test_navigation_observer.h"
+#include "content/public/test/test_utils.h"
+#include "printing/print_job_constants.h"
+#include "ui/events/keycodes/keyboard_codes.h"
+#include "url/gurl.h"
+#include "ipc/ipc_message_macros.h"
+
+using content::WebContents;
+using content::WebContentsObserver;
+
+class PrintPreviewObserver;
+class UIDoneLoadingMessageHandler;
+
+// Message refers to the 'UILoaded' message from print_preview.js.
+// It gets sent either from onPreviewGenerationDone or from
+// onManipulateSettings_() in print_preview.js
+enum State {
+  // Waiting for the first message so the program can select Save as PDF
+  kWaitingToSendSaveAsPdf = 0,
+  // Waiting for the second message so the test can set the layout
+  kWaitingToSendLayoutSettings = 1,
+  // Waiting for the third message so the test can set the page numbers
+  kWaitingToSendPageNumbers = 2,
+  // Waiting for the forth message so the test can set the headers checkbox
+  kWaitingToSendHeadersAndFooters = 3,
+  // Waiting for the fifth message so the test can set the background checkbox
+  kWaitingToSendBackgroundColorsAndImages = 4,
+  // Waiting for the sixth message so the test can set the margins combobox
+  kWaitingToSendMargins = 5,
+  // Waiting for the final message so the program can save to PDF.
+  kWaitingForFinalMessage = 6
+};
+
+class PrintPreviewObserver : public WebContentsObserver {
+  public:
+    explicit PrintPreviewObserver(WebContents* dialog, Browser* browser);
+    virtual ~PrintPreviewObserver();
+
+    void set_quit_closure(const base::Closure &closure);
+    void EndLoop();
+    bool OnMessageReceived(const IPC::Message& message) OVERRIDE;
+    PrintPreviewUI* GetUI() {
+      return ui_;
+    }
+
+    WebContents* GetPreviewContents() {
+      return web_contents_;
+    }
+
+    // Calls a javascript function that will change the print preview settings,
+    // such as the layout, the margins, page numbers, etc.
+    void ManipulatePreviewSettings(State state);
+
+    // Function to set the print preview settings and save them so they
+    // can be sent later. Currently only used in the constructor. Will be
+    // used when creating a test and take command line arguments.
+    // |page_numbers| is a comma separated page range.

[Lei Zhang, 2014/06/18 00:47:58]

The lingering question is whether |page_numbers| can be empty or not.

[ivandavid, 2014/06/18 02:29:37]

I think it can be. It seems to just revert back to the 'All' option. If the
string is completely empty. However, if there is only whitespace, that will
cause it to not work.

[Lei Zhang, 2014/06/18 21:01:28]

You should mention this in the comment in the code, rather than in the code
review.

+    //    Example: "1-5,9" will print pages 1 through 5 and page 9.
+    //    The pages specified must be less than or equal to the maximum
+    //    page number.
+    void SetPrintPreviewSettings(bool is_portrait,
+                                 const std::string& page_numbers,
+                                 bool headers_and_footers,
+                                 bool background_colors_and_images,
+                                 printing::MarginType margin_type);
+
+  private:
+    void OnDidGetPreviewPageCount(
+        const PrintHostMsg_DidGetPreviewPageCount_Params &params);
+
+    void DidCloneToNewWebContents(WebContents* old_web_contents,
+                                  WebContents* new_web_contents)
+                                  OVERRIDE;
+
+    void WebContentsDestroyed();
+
+    Browser* browser_;
+    base::Closure closure_;
+    WebContents* web_contents_;
+
+    PrintPreviewUI* ui_;
+
+    int count_;
+
+    scoped_ptr<PrintPreviewObserver> print_preview_observer_;
+    scoped_ptr<base::StringValue> message_name_;
+    scoped_ptr<base::FundamentalValue> is_portrait_;
+    scoped_ptr<base::StringValue> page_numbers_;
+    scoped_ptr<base::FundamentalValue> headers_and_footers_;
+    scoped_ptr<base::FundamentalValue> background_colors_and_images_;
+    scoped_ptr<base::FundamentalValue> margins_;
+
+    DISALLOW_COPY_AND_ASSIGN(PrintPreviewObserver);
+};
+
+class UIDoneLoadingMessageHandler : public content::WebUIMessageHandler {
+  public:
+    explicit UIDoneLoadingMessageHandler(PrintPreviewObserver* observer);
+    virtual ~UIDoneLoadingMessageHandler();
+    void HandleDone(const base::ListValue* /* args */);
+    void RegisterMessages() OVERRIDE;
+  private:
+    PrintPreviewObserver* observer_;
+    State state_;
+
+  DISALLOW_COPY_AND_ASSIGN(UIDoneLoadingMessageHandler);
+};
+
+UIDoneLoadingMessageHandler::UIDoneLoadingMessageHandler(
+    PrintPreviewObserver* observer) :
+  observer_(observer), state_(kWaitingToSendSaveAsPdf) {}
+
+UIDoneLoadingMessageHandler::~UIDoneLoadingMessageHandler() {
+  observer_ = NULL;
+}
+
+void UIDoneLoadingMessageHandler::RegisterMessages() OVERRIDE {
+  web_ui()->RegisterMessageCallback(
+  "UILoadedForTest",
+  base::Bind(&UIDoneLoadingMessageHandler::HandleDone,
+             base::Unretained(this)));
+}
+
+void UIDoneLoadingMessageHandler::HandleDone(
+    const base::ListValue* /* args */) {
+  if (state_ == kWaitingForFinalMessage) {
+    observer_->EndLoop();
+  } else {
+    observer_->ManipulatePreviewSettings(state_);
+    state_ = static_cast<State>(static_cast<int>(state_) + 1);
+  }
+}
+
+PrintPreviewObserver::PrintPreviewObserver(
+    WebContents* dialog,
+    Browser* browser) :
+    WebContentsObserver(dialog), browser_(browser), count_(0) {
+  SetPrintPreviewSettings("PORTRAIT",
+                          "",
+                          false,
+                          false,
+                          printing::DEFAULT_MARGINS);
+}
+
+PrintPreviewObserver::~PrintPreviewObserver() {
+  browser_ = NULL;
+  message_name_.reset();
+  is_portrait_.reset();
+  page_numbers_.reset();
+  headers_and_footers_.reset();
+  background_colors_and_images_.reset();
+  margins_.reset();
+}
+
+void PrintPreviewObserver::set_quit_closure(const base::Closure &closure) {
+  closure_ = closure;
+}
+
+void PrintPreviewObserver::EndLoop() {
+  base::MessageLoop::current()->PostTask(FROM_HERE, closure_);
+}
+
+bool PrintPreviewObserver::OnMessageReceived(const IPC::Message& message)
+    OVERRIDE {
+  IPC_BEGIN_MESSAGE_MAP(PrintPreviewObserver, message)
+    IPC_MESSAGE_HANDLER(PrintHostMsg_DidGetPreviewPageCount,
+                        OnDidGetPreviewPageCount)
+    IPC_MESSAGE_UNHANDLED(break;)
+  IPC_END_MESSAGE_MAP();
+  return false;
+}
+
+void PrintPreviewObserver::OnDidGetPreviewPageCount(
+    const PrintHostMsg_DidGetPreviewPageCount_Params &params) {
+
+  WebContents* tab = browser_->tab_strip_model()->GetActiveWebContents();
+
+  ASSERT_TRUE(tab);
+
+  printing::PrintPreviewDialogController* dialog_controller =
+      printing::PrintPreviewDialogController::GetInstance();
+
+  ASSERT_TRUE(dialog_controller);
+
+  web_contents_ = dialog_controller->GetPrintPreviewForContents(tab);
+
+  ASSERT_TRUE(web_contents_ && printing::PrintPreviewDialogController::
+      IsPrintPreviewDialog(web_contents_));
+
+  ui_ = static_cast<PrintPreviewUI*>(web_contents_->GetWebUI()->
+      GetController());
+
+  ASSERT_TRUE(ui_);
+
+  ASSERT_TRUE(ui_->web_ui());
+
+  ui_->web_ui()->CallJavascriptFunction("onEnableManipulateSettingsForTest");
+
+  this->Observe(web_contents_);
+
+  UIDoneLoadingMessageHandler* handler = new UIDoneLoadingMessageHandler(this);
+
+  ui_->web_ui()->AddMessageHandler(handler);
+}
+
+void PrintPreviewObserver::SetPrintPreviewSettings(
+    bool is_portrait,
+    const std::string& page_numbers,
+    bool headers_and_footers,
+    bool background_colors_and_images,
+    printing::MarginType margins) {
+  is_portrait_.reset(new base::FundamentalValue(is_portrait));
+  page_numbers_.reset(new base::StringValue(page_numbers));
+  headers_and_footers_.reset(
+      new base::FundamentalValue(headers_and_footers));
+  background_colors_and_images_.reset(
+      new base::FundamentalValue(background_colors_and_images));
+  margins_.reset(new base::FundamentalValue(margins));
+}
+
+
+void PrintPreviewObserver::ManipulatePreviewSettings(State state) {
+  std::vector<const base::Value*> args;
+  if (state == kWaitingToSendSaveAsPdf) {
+    message_name_.reset(new base::StringValue("SAVE_AS_PDF"));
+    args.push_back(message_name_.get());
+  } else if (state == kWaitingToSendLayoutSettings) {
+    message_name_.reset(new base::StringValue("LAYOUT_SETTINGS"));
+    args.push_back(message_name_.get());
+    args.push_back(is_portrait_.get());
+  } else if (state == kWaitingToSendPageNumbers) {
+    message_name_.reset(new base::StringValue("PAGE_NUMBERS"));
+    args.push_back(message_name_.get());
+    args.push_back(page_numbers_.get());
+  } else if (state == kWaitingToSendHeadersAndFooters) {
+    message_name_.reset(new base::StringValue("HEADERS_AND_FOOTERS"));
+    args.push_back(message_name_.get());
+    args.push_back(headers_and_footers_.get());
+  } else if (state == kWaitingToSendBackgroundColorsAndImages) {
+    message_name_.reset(new base::StringValue("BACKGROUND_COLORS_AND_IMAGES"));
+    args.push_back(message_name_.get());
+    args.push_back(background_colors_and_images_.get());
+  } else if (state == kWaitingToSendMargins) {
+    message_name_.reset(new base::StringValue("MARGINS"));
+    args.push_back(message_name_.get());
+    args.push_back(margins_.get());
+  }
+  DCHECK(!args.empty());
+  ui_->web_ui()->CallJavascriptFunction("onManipulateSettingsForTest", args);
+}
+
+void PrintPreviewObserver::DidCloneToNewWebContents(
+    WebContents* old_web_contents,
+    WebContents* new_web_contents)
+    OVERRIDE {
+  this->Observe(new_web_contents);
+}
+
+void PrintPreviewObserver::WebContentsDestroyed() {
+  this->EndLoop();
+}
+
+class PrintPreviewPdfGeneratedBrowserTest : public InProcessBrowserTest {
+  public:
+    PrintPreviewPdfGeneratedBrowserTest() {}
+    virtual ~PrintPreviewPdfGeneratedBrowserTest() {}
+
+    void NavigateAndPreview(std::string file_name) {
+      base::FilePath directory("printing");
+      base::FilePath file(file_name);
+
+      ui_test_utils::NavigateToURL(browser(),
+                                   ui_test_utils::GetTestUrl(directory, file));
+
+      base::RunLoop loop;
+      print_preview_observer_->set_quit_closure(loop.QuitClosure());
+      chrome::Print(browser());
+      loop.Run();
+    }
+
+    void Print() {
+      base::FilePath pdf_file_save_path;
+      ASSERT_TRUE(PathService::Get(chrome::DIR_TEST_DATA, &pdf_file_save_path));
+      pdf_file_save_path = pdf_file_save_path.AppendASCII("printing");
+      pdf_file_save_path = pdf_file_save_path.AppendASCII("dummy.pdf");
+
+      base::RunLoop loop;
+      print_preview_observer_->set_quit_closure(loop.QuitClosure());
+
+      print_preview_observer_->GetUI()->handler_->
+          FileSelected(pdf_file_save_path, 0, NULL);
+
+      loop.Run();
+    }
+
+  private:
+    virtual void SetUpOnMainThread() OVERRIDE {
+      WebContents* tab =
+          browser()->tab_strip_model()->GetActiveWebContents();
+      ASSERT_TRUE(tab);
+
+      print_preview_observer_.reset(new PrintPreviewObserver(tab, browser()));
+      chrome::DuplicateTab(browser());
+
+      WebContents* initiator_ =
+          browser()->tab_strip_model()->GetActiveWebContents();
+      ASSERT_TRUE(initiator_);
+      ASSERT_NE(tab, initiator_);
+    }
+
+    virtual void CleanUpOnMainThread() OVERRIDE {
+      print_preview_observer_.reset();
+    }
+
+    scoped_ptr<PrintPreviewObserver> print_preview_observer_;
+
+    DISALLOW_COPY_AND_ASSIGN(PrintPreviewPdfGeneratedBrowserTest);
+};
+
+IN_PROC_BROWSER_TEST_F(PrintPreviewPdfGeneratedBrowserTest, DummyTest) {
+    NavigateAndPreview("test2.html");
+    Print();
+}