Print Preview: Make print preview tab modal.

chrome/browser/ui/webui/print_preview_handler_unittest.cc

 // 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) {
+    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;
 }
 
-}
+}  // namespace
 
 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());
@@ skipping 14 matching lines @@
     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_));
 
@@ skipping 28 matching lines @@
     // 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) {
+    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(
@@ skipping 11 matching lines @@
 
   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(
@@ skipping 31 matching lines @@
   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);
 }
 
@@ skipping 40 matching lines @@
   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));
 }