| OLD | NEW |
| (Empty) |
| 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 #if !defined(GOOGLE_CHROME_BUILD) || defined(OS_CHROMEOS) | |
| 21 CommandLine::ForCurrentProcess()->AppendSwitch(switches::kEnablePrintPreview); | |
| 22 #endif | |
| 23 ASSERT_TRUE(browser()); | |
| 24 BrowserList::SetLastActive(browser()); | |
| 25 ASSERT_TRUE(BrowserList::GetLastActive()); | |
| 26 | |
| 27 browser()->NewTab(); | |
| 28 TabContents* initiator_tab = browser()->GetSelectedTabContents(); | |
| 29 ASSERT_TRUE(initiator_tab); | |
| 30 | |
| 31 scoped_refptr<printing::PrintPreviewTabController> | |
| 32 controller(new printing::PrintPreviewTabController()); | |
| 33 ASSERT_TRUE(controller); | |
| 34 | |
| 35 TabContents* preview_tab = controller->GetOrCreatePreviewTab(initiator_tab); | |
| 36 | |
| 37 EXPECT_NE(initiator_tab, preview_tab); | |
| 38 EXPECT_EQ(2, browser()->tab_count()); | |
| 39 | |
| 40 PrintPreviewUI* preview_ui = | |
| 41 reinterpret_cast<PrintPreviewUI*>(preview_tab->web_ui()); | |
| 42 ASSERT_TRUE(preview_ui != NULL); | |
| 43 PrintPreviewUIHTMLSource* html_source = preview_ui->html_source(); | |
| 44 | |
| 45 PrintPreviewUIHTMLSource::PrintPreviewData data; | |
| 46 html_source->GetPrintPreviewData(&data); | |
| 47 EXPECT_EQ(NULL, data.first); | |
| 48 EXPECT_EQ(0U, data.second); | |
| 49 | |
| 50 PrintPreviewUIHTMLSource::PrintPreviewData dummy_data = | |
| 51 std::make_pair(new base::SharedMemory(), 1234); | |
| 52 | |
| 53 html_source->SetPrintPreviewData(dummy_data); | |
| 54 html_source->GetPrintPreviewData(&data); | |
| 55 EXPECT_EQ(dummy_data, data); | |
| 56 | |
| 57 // This should not cause any memory leaks. | |
| 58 dummy_data.first = new base::SharedMemory(); | |
| 59 html_source->SetPrintPreviewData(dummy_data); | |
| 60 } | |
| OLD | NEW |