Print Preview: Handle a crashed initiator tab by showing a message in PP.

chrome/browser/printing/print_preview_tab_controller_browsertest.cc

Index: chrome/browser/printing/print_preview_tab_controller_browsertest.cc
diff --git a/chrome/browser/printing/print_preview_tab_controller_browsertest.cc b/chrome/browser/printing/print_preview_tab_controller_browsertest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..5663b256ece77f20411095f2be03fa5d6e96d7a1
--- /dev/null
+++ b/chrome/browser/printing/print_preview_tab_controller_browsertest.cc
@@ -0,0 +1,170 @@
+// 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 "chrome/browser/printing/print_preview_tab_controller.h"
+#include "chrome/browser/ui/browser.h"
+#include "chrome/browser/ui/browser_list.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/browser/tab_contents/tab_contents.h"
+#include "googleurl/src/gurl.h"
+
+namespace {
+
+typedef InProcessBrowserTest PrintPreviewTabControllerBrowserTest;
+
+// Test to verify that when a preview tab navigates, we can create a new print
+// preview tab for both initiator tab and new preview tab contents.
+IN_PROC_BROWSER_TEST_F(PrintPreviewTabControllerBrowserTest,
+                       NavigateFromPrintPreviewTab) {
+  ASSERT_TRUE(browser());
+  BrowserList::SetLastActive(browser());
+  ASSERT_TRUE(BrowserList::GetLastActive());
+
+  // Lets start with one window with one tab.
+  EXPECT_EQ(1u, BrowserList::size());
+  EXPECT_EQ(1, browser()->tab_count());
+
+  // Create a reference to initiator tab contents.
+  TabContents* initiator_tab = browser()->GetSelectedTabContents();
+  ASSERT_TRUE(initiator_tab);
+
+  scoped_refptr<printing::PrintPreviewTabController>
+     tab_controller(new printing::PrintPreviewTabController());
+  ASSERT_TRUE(tab_controller);
+
+  // Get the preview tab for initiator tab.
+  TabContents* preview_tab =
+    tab_controller->GetOrCreatePreviewTab(initiator_tab);
+
+  // New print preview tab is created. Current focus is on preview tab.
+  EXPECT_EQ(2, browser()->tab_count());
+  EXPECT_NE(initiator_tab, preview_tab);
+
+  GURL url(chrome::kAboutBlankURL);
+  ui_test_utils::NavigateToURL(browser(), url);
+  EXPECT_EQ(url, preview_tab->GetURL());
+
+  // Get the print preview tab for initiator tab.
+  TabContents* new_preview_tab =
+     tab_controller->GetOrCreatePreviewTab(initiator_tab);
+
+  // New preview tab is created.
+  EXPECT_EQ(3, browser()->tab_count());
+  EXPECT_NE(new_preview_tab, preview_tab);
+
+  // Get the print preview tab for old preview tab.
+  TabContents* newest_preview_tab =
+  tab_controller->GetOrCreatePreviewTab(preview_tab);
+
+  // Newest preview tab is created and the previously created preview tab is not
+  // merely activated.
+  EXPECT_EQ(4, browser()->tab_count());
+  EXPECT_NE(newest_preview_tab, new_preview_tab);
+}
+
+// Test to verify that when a initiator tab navigates, we can create a new
+// preview tab for the new tab contents. But we cannot create a preview tab for
+// the old preview tab.
+IN_PROC_BROWSER_TEST_F(PrintPreviewTabControllerBrowserTest,
+                       NavigateFromInitiatorTab) {
+  ASSERT_TRUE(browser());
+  BrowserList::SetLastActive(browser());
+  ASSERT_TRUE(BrowserList::GetLastActive());
+
+  // Lets start with one window with one tab.
+  EXPECT_EQ(1u, BrowserList::size());
+  EXPECT_EQ(1, browser()->tab_count());
+
+  // Create a reference to initiator tab contents.
+  TabContents* initiator_tab = browser()->GetSelectedTabContents();
+  ASSERT_TRUE(initiator_tab);
+
+  scoped_refptr<printing::PrintPreviewTabController>
+     tab_controller(new printing::PrintPreviewTabController());
+  ASSERT_TRUE(tab_controller);
+
+  // Get the preview tab for initiator tab.
+  TabContents* preview_tab =
+    tab_controller->GetOrCreatePreviewTab(initiator_tab);
+
+  // New print preview tab is created. Current focus is on preview tab.
+  EXPECT_EQ(2, browser()->tab_count());
+  EXPECT_NE(initiator_tab, preview_tab);
+
+  // Activate initiator tab.
+  browser()->ActivateTabAt(0, true);
+  GURL url(chrome::kChromeUINewTabURL);
+  ui_test_utils::NavigateToURL(browser(), url);
+
+  // Get the print preview tab for initiator tab.
+  TabContents* new_preview_tab =
+     tab_controller->GetOrCreatePreviewTab(initiator_tab);
+
+  // New preview tab is created.
+  EXPECT_EQ(3, browser()->tab_count());
+  EXPECT_NE(new_preview_tab, preview_tab);
+
+  // Get the print preview tab for old preview tab.
+  TabContents* newest_preview_tab =
+  tab_controller->GetOrCreatePreviewTab(preview_tab);
+
+  // Make sure preview tab is not created for |preview_tab|.
+  EXPECT_EQ(3, browser()->tab_count());
+  EXPECT_EQ(newest_preview_tab, preview_tab);
+}
+
+// Test to verify that even after reloading initiator tab and preview tab,
+// their association exists.
+IN_PROC_BROWSER_TEST_F(PrintPreviewTabControllerBrowserTest,
+                       ReloadInitiatorTabAndPreviewTab) {
+  ASSERT_TRUE(browser());
+  BrowserList::SetLastActive(browser());
+  ASSERT_TRUE(BrowserList::GetLastActive());
+
+  // Lets start with one window with one tab.
+  EXPECT_EQ(1u, BrowserList::size());
+  EXPECT_EQ(1, browser()->tab_count());
+
+  // Create a reference to initiator tab contents.
+  TabContents* initiator_tab = browser()->GetSelectedTabContents();
+  ASSERT_TRUE(initiator_tab);
+
+  scoped_refptr<printing::PrintPreviewTabController>
+     tab_controller(new printing::PrintPreviewTabController());
+  ASSERT_TRUE(tab_controller);
+
+  // Get the preview tab for initiator tab.
+  TabContents* preview_tab =
+    tab_controller->GetOrCreatePreviewTab(initiator_tab);
+
+  // New print preview tab is created. Current focus is on preview tab.
+  EXPECT_EQ(2, browser()->tab_count());
+  EXPECT_NE(initiator_tab, preview_tab);
+
+  // Activate initiator tab and reload.
+  browser()->ActivateTabAt(0, true);
+  browser()->Reload(CURRENT_TAB);
+
+  // Get the print preview tab for initiator tab.
+  TabContents* new_preview_tab =
+     tab_controller->GetOrCreatePreviewTab(initiator_tab);
+
+  // Old preview tab is activated.
+  EXPECT_EQ(2, browser()->tab_count());
+  EXPECT_EQ(new_preview_tab, preview_tab);
+
+  // Reload preview tab.
+  browser()->Reload(CURRENT_TAB);
+  // Get the print preview tab for old preview tab.
+  TabContents* newest_preview_tab =
+  tab_controller->GetOrCreatePreviewTab(preview_tab);
+
+  // Make sure new preview tab is not created for |preview_tab|.
+  EXPECT_EQ(2, browser()->tab_count());
+  EXPECT_EQ(newest_preview_tab, preview_tab);
+}
+
+}  // namespace