1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 #include "base/command_line.h" 6 #include "base/shared_memory.h" 7 #include "chrome/browser/printing/print_preview_tab_controller.h" 8 #include "chrome/browser/ui/browser_list.h" 9 #include "chrome/browser/ui/webui/print_preview_ui.h" 10 #include "chrome/browser/ui/webui/print_preview_ui_html_source.h" 11 #include "chrome/common/chrome_switches.h" 12 #include "chrome/test/browser_with_test_window_test.h" 13 #include "chrome/test/testing_profile.h" 14 #include "content/browser/tab_contents/tab_contents.h" 15 16 typedef BrowserWithTestWindowTest PrintPreviewUIHTMLSourceTest; 17 18 // Create/Get a preview tab for initiator tab. 19 TEST_F(PrintPreviewUIHTMLSourceTest, PrintPreviewData) { 20 // TODO(thestig) Remove when print preview is enabled by default. 21 CommandLine::ForCurrentProcess()->AppendSwitch(switches::kEnablePrintPreview); 22 ASSERT_TRUE(browser()); 23 BrowserList::SetLastActive(browser()); 24 ASSERT_TRUE(BrowserList::GetLastActive()); 25 26 browser()->NewTab(); 27 TabContents* initiator_tab = browser()->GetSelectedTabContents(); 28 ASSERT_TRUE(initiator_tab); 29 30 scoped_refptr<printing::PrintPreviewTabController> 31 controller(new printing::PrintPreviewTabController()); 32 ASSERT_TRUE(controller); 33 34 TabContents* preview_tab = controller->GetOrCreatePreviewTab(initiator_tab); 35 36 EXPECT_NE(initiator_tab, preview_tab); 37 EXPECT_EQ(2, browser()->tab_count()); 38 39 PrintPreviewUI* preview_ui = 40 reinterpret_cast<PrintPreviewUI*>(preview_tab->web_ui()); 41 ASSERT_TRUE(preview_ui != NULL); 42 PrintPreviewUIHTMLSource* html_source = preview_ui->html_source(); 43 44 PrintPreviewUIHTMLSource::PrintPreviewData data; 45 html_source->GetPrintPreviewData(&data); 46 EXPECT_EQ(NULL, data.first); 47 EXPECT_EQ(0U, data.second); 48 49 PrintPreviewUIHTMLSource::PrintPreviewData dummy_data = 50 std::make_pair(new base::SharedMemory(), 1234); 51 52 html_source->SetPrintPreviewData(dummy_data); 53 html_source->GetPrintPreviewData(&data); 54 EXPECT_EQ(dummy_data, data); 55 56 // This should not cause any memory leaks. 57 dummy_data.first = new base::SharedMemory(); 58 html_source->SetPrintPreviewData(dummy_data); 59 } 60