Print Preview: Making margin selection sticky (part 2/2, again)

chrome/browser/ui/webui/print_preview_handler_unittest.cc

Index: chrome/browser/ui/webui/print_preview_handler_unittest.cc
diff --git a/chrome/browser/ui/webui/print_preview_handler_unittest.cc b/chrome/browser/ui/webui/print_preview_handler_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..d44ce53e867dfac17f8f7c767ad371a020747b35
--- /dev/null
+++ b/chrome/browser/ui/webui/print_preview_handler_unittest.cc
@@ -0,0 +1,261 @@
+// Copyright (c) 2011 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/command_line.h"
+#include "base/json/json_writer.h"
+#include "base/values.h"
+#include "chrome/browser/browser_process.h"
+#include "chrome/browser/printing/background_printing_manager.h"
+#include "chrome/browser/printing/print_preview_tab_controller.h"
+#include "chrome/browser/ui/browser_list.h"
+#include "chrome/browser/ui/tab_contents/tab_contents_wrapper.h"
+#include "chrome/browser/ui/webui/print_preview_handler.h"
+#include "chrome/browser/ui/webui/print_preview_ui.h"
+#include "chrome/common/chrome_switches.h"
+#include "chrome/test/base/browser_with_test_window_test.h"
+#include "printing/page_size_margins.h"
+#include "printing/print_job_constants.h"
+
+namespace {
+
+DictionaryValue* GetCustomMarginsDictionary(
+    const double margin_top, const double margin_right,
+    const double margin_bottom,const double margin_left) {
+  base::DictionaryValue* custom_settings = new base::DictionaryValue();
+  custom_settings->SetDouble(printing::kSettingMarginTop, margin_top);
+  custom_settings->SetDouble(printing::kSettingMarginRight, margin_right);
+  custom_settings->SetDouble(printing::kSettingMarginBottom, margin_bottom);
+  custom_settings->SetDouble(printing::kSettingMarginLeft, margin_left);
+  return custom_settings;
+}
+
+}
+
+class PrintPreviewHandlerTest : public BrowserWithTestWindowTest {
+ protected:
+  void SetUp() {
+    BrowserWithTestWindowTest::SetUp();
+#if !defined(GOOGLE_CHROME_BUILD) || defined(OS_CHROMEOS)
+    CommandLine::ForCurrentProcess()->AppendSwitch(
+        switches::kEnablePrintPreview);
+#endif
+    ASSERT_TRUE(browser());
+    BrowserList::SetLastActive(browser());
+    ASSERT_TRUE(BrowserList::GetLastActive());
+
+    browser()->NewTab();
+    EXPECT_EQ(1, browser()->tab_count());
+    OpenPrintPreviewTab();
+  }
+
+  virtual void TearDown() {
+    DeletePrintPreviewTab();
+    ClearStickySettings();
+  }
+
+  void OpenPrintPreviewTab() {
+    TabContentsWrapper* initiator_tab =
+        browser()->GetSelectedTabContentsWrapper();
+    ASSERT_TRUE(initiator_tab);
+
+    scoped_refptr<printing::PrintPreviewTabController>
+        controller(new printing::PrintPreviewTabController());
+    ASSERT_TRUE(controller.get());
+
+    preview_tab_ = controller->GetOrCreatePreviewTab(initiator_tab);
+    ASSERT_TRUE(preview_tab_);
+    EXPECT_EQ(2, browser()->tab_count());
+
+    preview_ui_ = static_cast<PrintPreviewUI*>(preview_tab_->web_ui());
+    ASSERT_TRUE(preview_ui_);
+  }
+
+  void DeletePrintPreviewTab() {
+    printing::BackgroundPrintingManager* bg_printing_manager =
+        g_browser_process->background_printing_manager();
+    ASSERT_TRUE(bg_printing_manager->HasPrintPreviewTab(preview_tab_));
+
+    // Deleting TabContentsWrapper* to avoid warings from pref_notifier_impl.cc
+    // after the test ends.
+    delete preview_tab_;
+  }
+
+  void CheckCustomMargins(const double margin_top,
+                          const double margin_right,
+                          const double margin_bottom,
+                          const double margin_left) {
+    EXPECT_EQ(PrintPreviewHandler::last_used_page_size_margins_->margin_top,
+              margin_top);
+    EXPECT_EQ(PrintPreviewHandler::last_used_page_size_margins_->margin_right,
+              margin_right);
+    EXPECT_EQ(PrintPreviewHandler::last_used_page_size_margins_->margin_bottom,
+              margin_bottom);
+    EXPECT_EQ(PrintPreviewHandler::last_used_page_size_margins_->margin_left,
+              margin_left);
+  }
+
+  void RequestPrintWithDefaultMargins() {
+    // Set the minimal dummy settings to make the HandlePrint() code happy.
+    DictionaryValue settings;
+    settings.SetBoolean(printing::kSettingPreviewModifiable, true);
+    settings.SetInteger(printing::kSettingColor, printing::COLOR);
+    settings.SetBoolean(printing::kSettingPrintToPDF, false);
+    settings.SetInteger(printing::kSettingMarginsType,
+                        printing::DEFAULT_MARGINS);
+
+    // Put |settings| in to |args| as a JSON string.
+    std::string json_string;
+    base::JSONWriter::Write(&settings, false, &json_string);
+    ListValue args;
+    args.Append(new base::StringValue(json_string));  // |args| takes ownership.
+    preview_ui_->handler_->HandlePrint(&args);
+  }
+
+  void RequestPrintWithCustomMargins(
+    const double margin_top, const double margin_right,
+    const double margin_bottom,const double margin_left) {
+    // Set the minimal dummy settings to make the HandlePrint() code happy.
+    DictionaryValue settings;
+    settings.SetBoolean(printing::kSettingPreviewModifiable, true);
+    settings.SetInteger(printing::kSettingColor, printing::COLOR);
+    settings.SetBoolean(printing::kSettingPrintToPDF, false);
+    settings.SetInteger(printing::kSettingMarginsType,
+                        printing::CUSTOM_MARGINS);
+
+    // Creating custom margins dictionary and nesting it in |settings|.
+    DictionaryValue* custom_settings = GetCustomMarginsDictionary(
+        margin_top, margin_right, margin_bottom, margin_left);
+    // |settings| takes ownership.
+    settings.Set(printing::kSettingMarginsCustom, custom_settings);
+
+    // Put |settings| in to |args| as a JSON string.
+    std::string json_string;
+    base::JSONWriter::Write(&settings, false, &json_string);
+    ListValue args;
+    args.Append(new base::StringValue(json_string));  // |args| takes ownership.
+    preview_ui_->handler_->HandlePrint(&args);
+  }
+
+  TabContentsWrapper* preview_tab_;
+  PrintPreviewUI* preview_ui_;
+
+ private:
+  void ClearStickySettings() {
+    PrintPreviewHandler::last_used_margins_type_ = printing::DEFAULT_MARGINS;
+    delete PrintPreviewHandler::last_used_page_size_margins_;
+    PrintPreviewHandler::last_used_page_size_margins_ = NULL;
+  }
+
+};
+
+// Tests that margin settings are saved correctly when printing with custom
+// margins selected.
+TEST_F(PrintPreviewHandlerTest, StickyMarginsCustom) {
+  const double kMarginTop = 25.5;
+  const double kMarginRight = 26.5;
+  const double kMarginBottom = 27.5;
+  const double kMarginLeft = 28.5;
+  RequestPrintWithCustomMargins(
+      kMarginTop, kMarginRight, kMarginBottom, kMarginLeft);
+  EXPECT_EQ(1, browser()->tab_count());
+
+  // Checking that sticky settings were saved correctly.
+  EXPECT_EQ(PrintPreviewHandler::last_used_color_model_, printing::COLOR);
+  EXPECT_EQ(PrintPreviewHandler::last_used_margins_type_,
+            printing::CUSTOM_MARGINS);
+  ASSERT_TRUE(PrintPreviewHandler::last_used_page_size_margins_);
+  CheckCustomMargins(kMarginTop, kMarginRight, kMarginBottom, kMarginLeft);
+}
+
+// Tests that margin settings are saved correctly when printing with default
+// margins selected.
+TEST_F(PrintPreviewHandlerTest, StickyMarginsDefault) {
+  RequestPrintWithDefaultMargins();
+  EXPECT_EQ(1, browser()->tab_count());
+
+  // Checking that sticky settings were saved correctly.
+  EXPECT_EQ(PrintPreviewHandler::last_used_color_model_, printing::COLOR);
+  EXPECT_EQ(PrintPreviewHandler::last_used_margins_type_,
+            printing::DEFAULT_MARGINS);
+  ASSERT_FALSE(PrintPreviewHandler::last_used_page_size_margins_);
+}
+
+// Tests that margin settings are saved correctly when printing with custom
+// margins selected and then again with default margins selected.
+TEST_F(PrintPreviewHandlerTest, StickyMarginsCustomThenDefault) {
+  const double kMarginTop = 125.5;
+  const double kMarginRight = 126.5;
+  const double kMarginBottom = 127.5;
+  const double kMarginLeft = 128.5;
+  RequestPrintWithCustomMargins(
+      kMarginTop, kMarginRight, kMarginBottom, kMarginLeft);
+  EXPECT_EQ(1, browser()->tab_count());
+  DeletePrintPreviewTab();
+  EXPECT_EQ(PrintPreviewHandler::last_used_margins_type_,
+            printing::CUSTOM_MARGINS);
+  ASSERT_TRUE(PrintPreviewHandler::last_used_page_size_margins_);
+  CheckCustomMargins(kMarginTop, kMarginRight, kMarginBottom, kMarginLeft);
+
+  OpenPrintPreviewTab();
+  EXPECT_EQ(2, browser()->tab_count());
+  RequestPrintWithDefaultMargins();
+
+  // Checking that sticky settings were saved correctly.
+  EXPECT_EQ(PrintPreviewHandler::last_used_color_model_, printing::COLOR);
+  EXPECT_EQ(PrintPreviewHandler::last_used_margins_type_,
+            printing::DEFAULT_MARGINS);
+  ASSERT_TRUE(PrintPreviewHandler::last_used_page_size_margins_);
+  CheckCustomMargins(kMarginTop, kMarginRight, kMarginBottom, kMarginLeft);
+}
+
+// Tests that margin settings are retrieved correctly after printing with custom
+// margins.
+TEST_F(PrintPreviewHandlerTest, GetLastUsedMarginSettingsCustom) {
+  const double kMarginTop = 125.5;
+  const double kMarginRight = 126.5;
+  const double kMarginBottom = 127.5;
+  const double kMarginLeft = 128.5;
+  RequestPrintWithCustomMargins(
+      kMarginTop, kMarginRight, kMarginBottom, kMarginLeft);
+  base::DictionaryValue initial_settings;
+  preview_ui_->handler_->GetLastUsedMarginSettings(&initial_settings);
+  int margins_type;
+  EXPECT_TRUE(initial_settings.GetInteger(printing::kSettingMarginsType,
+                                          &margins_type));
+  EXPECT_EQ(margins_type, printing::CUSTOM_MARGINS);
+  double margin_value;
+  EXPECT_TRUE(initial_settings.GetDouble(printing::kSettingMarginTop,
+                                         &margin_value));
+  EXPECT_EQ(kMarginTop, margin_value);
+  EXPECT_TRUE(initial_settings.GetDouble(printing::kSettingMarginRight,
+                                         &margin_value));
+  EXPECT_EQ(kMarginRight, margin_value);
+  EXPECT_TRUE(initial_settings.GetDouble(printing::kSettingMarginBottom,
+                                         &margin_value));
+  EXPECT_EQ(kMarginBottom, margin_value);
+  EXPECT_TRUE(initial_settings.GetDouble(printing::kSettingMarginLeft,
+                                         &margin_value));
+  EXPECT_EQ(kMarginLeft, margin_value);
+}
+
+// Tests that margin settings are retrieved correctly after printing with
+// default margins.
+TEST_F(PrintPreviewHandlerTest, GetLastUsedMarginSettingsDefault) {
+  RequestPrintWithDefaultMargins();
+  base::DictionaryValue initial_settings;
+  preview_ui_->handler_->GetLastUsedMarginSettings(&initial_settings);
+  int margins_type;
+  EXPECT_TRUE(initial_settings.GetInteger(printing::kSettingMarginsType,
+                                          &margins_type));
+  EXPECT_EQ(margins_type, printing::DEFAULT_MARGINS);
+  double margin_value;
+  EXPECT_FALSE(initial_settings.GetDouble(printing::kSettingMarginTop,
+                                          &margin_value));
+  EXPECT_FALSE(initial_settings.GetDouble(printing::kSettingMarginRight,
+                                          &margin_value));
+  EXPECT_FALSE(initial_settings.GetDouble(printing::kSettingMarginBottom,
+                                          &margin_value));
+  EXPECT_FALSE(initial_settings.GetDouble(printing::kSettingMarginLeft,
+                                          &margin_value));
+}