Print Preview: Store preview data in the print preview controller....

chrome/browser/printing/print_preview_tab_controller_unittest.cc

Index: chrome/browser/printing/print_preview_tab_controller_unittest.cc
===================================================================
--- chrome/browser/printing/print_preview_tab_controller_unittest.cc	(revision 71104)
+++ chrome/browser/printing/print_preview_tab_controller_unittest.cc	(working copy)
@@ -2,6 +2,7 @@
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
+#include "base/shared_memory.h"
 #include "chrome/browser/printing/print_preview_tab_controller.h"
 #include "chrome/browser/tab_contents/tab_contents.h"
 #include "chrome/browser/tabs/tab_strip_model.h"
@@ -112,3 +113,47 @@
       tab_contents_1, tab_contents_1->controller().window_id().id());
   EXPECT_EQ(preview_tab_1_index, browser()->selected_index());
 }
+
+// Tests {Get,Set}PrintPreviewData.
+TEST_F(PrintPreviewTabControllerTest, PrintPreviewData) {
+  ASSERT_TRUE(browser());
+  BrowserList::SetLastActive(browser());
+  ASSERT_TRUE(BrowserList::GetLastActive());
+
+  browser()->NewTab();
+  TabContents* initiator_tab = browser()->GetSelectedTabContents();
+  ASSERT_TRUE(initiator_tab);
+
+  scoped_refptr<printing::PrintPreviewTabController>
+      controller(new printing::PrintPreviewTabController());
+  ASSERT_TRUE(controller);
+
+  TabContents* preview_tab = controller->GetOrCreatePreviewTab(
+      initiator_tab, initiator_tab->controller().window_id().id());
+
+  EXPECT_NE(initiator_tab, preview_tab);
+  EXPECT_EQ(2, browser()->tab_count());
+
+  printing::PrintPreviewTabController::PrintPreviewData data;
+  EXPECT_FALSE(controller->GetPrintPreviewData(NULL, &data));
+  EXPECT_FALSE(controller->GetPrintPreviewData(initiator_tab, &data));
+  EXPECT_TRUE(controller->GetPrintPreviewData(preview_tab, &data));
+  EXPECT_EQ(NULL, data.first);
+  EXPECT_EQ(0U, data.second);
+
+  printing::PrintPreviewTabController::PrintPreviewData dummy_data =
+      std::make_pair(new base::SharedMemory(), 1234);
+
+  EXPECT_FALSE(controller->SetPrintPreviewData(NULL, dummy_data));
+  EXPECT_FALSE(controller->SetPrintPreviewData(initiator_tab, dummy_data));
+  EXPECT_TRUE(controller->SetPrintPreviewData(preview_tab, dummy_data));
+
+  EXPECT_FALSE(controller->GetPrintPreviewData(NULL, &data));
+  EXPECT_FALSE(controller->GetPrintPreviewData(initiator_tab, &data));
+  EXPECT_TRUE(controller->GetPrintPreviewData(preview_tab, &data));
+  EXPECT_EQ(dummy_data, data);
+
+  // This should not cause any memory leaks.
+  dummy_data.first = new base::SharedMemory();
+  EXPECT_TRUE(controller->SetPrintPreviewData(preview_tab, dummy_data));
+}
Property changes on: chrome/browser/printing/print_preview_tab_controller_unittest.cc
___________________________________________________________________
Added: svn:eol-style
   + LF