OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include <vector> | 5 #include <vector> |
6 | 6 |
7 #include "base/command_line.h" | 7 #include "base/command_line.h" |
8 #include "base/memory/ref_counted_memory.h" | 8 #include "base/memory/ref_counted_memory.h" |
9 #include "chrome/browser/printing/print_preview_tab_controller.h" | 9 #include "chrome/browser/printing/print_preview_tab_controller.h" |
10 #include "chrome/browser/ui/browser_list.h" | 10 #include "chrome/browser/ui/browser_list.h" |
11 #include "chrome/browser/ui/constrained_window_tab_helper.h" | |
12 #include "chrome/browser/ui/tab_contents/tab_contents_wrapper.h" | 11 #include "chrome/browser/ui/tab_contents/tab_contents_wrapper.h" |
13 #include "chrome/browser/ui/webui/print_preview_ui.h" | 12 #include "chrome/browser/ui/webui/print_preview_ui.h" |
14 #include "chrome/common/chrome_switches.h" | 13 #include "chrome/common/chrome_switches.h" |
15 #include "chrome/test/base/browser_with_test_window_test.h" | 14 #include "chrome/test/base/browser_with_test_window_test.h" |
16 #include "chrome/test/base/testing_profile.h" | 15 #include "chrome/test/base/testing_profile.h" |
17 #include "content/browser/tab_contents/tab_contents.h" | 16 #include "content/browser/tab_contents/tab_contents.h" |
18 #include "printing/print_job_constants.h" | 17 #include "printing/print_job_constants.h" |
19 | 18 |
20 namespace { | 19 namespace { |
21 | 20 |
22 const unsigned char blob1[] = | 21 const unsigned char blob1[] = |
23 "12346102356120394751634516591348710478123649165419234519234512349134"; | 22 "12346102356120394751634516591348710478123649165419234519234512349134"; |
24 | 23 |
25 size_t GetConstrainedWindowCount(TabContentsWrapper* tab) { | |
26 return tab->constrained_window_tab_helper()->constrained_window_count(); | |
27 } | |
28 | |
29 } // namespace | 24 } // namespace |
30 | 25 |
31 typedef BrowserWithTestWindowTest PrintPreviewUITest; | 26 typedef BrowserWithTestWindowTest PrintPreviewUITest; |
32 | 27 |
33 // Create/Get a preview tab for initiator tab. | 28 // Create/Get a preview tab for initiator tab. |
34 TEST_F(PrintPreviewUITest, PrintPreviewData) { | 29 TEST_F(PrintPreviewUITest, PrintPreviewData) { |
35 CommandLine::ForCurrentProcess()->AppendSwitch(switches::kEnablePrintPreview); | 30 CommandLine::ForCurrentProcess()->AppendSwitch(switches::kEnablePrintPreview); |
36 ASSERT_TRUE(browser()); | 31 ASSERT_TRUE(browser()); |
37 BrowserList::SetLastActive(browser()); | 32 BrowserList::SetLastActive(browser()); |
38 ASSERT_TRUE(BrowserList::GetLastActive()); | 33 ASSERT_TRUE(BrowserList::GetLastActive()); |
39 | 34 |
40 browser()->NewTab(); | 35 browser()->NewTab(); |
41 TabContentsWrapper* initiator_tab = | 36 TabContentsWrapper* initiator_tab = |
42 browser()->GetSelectedTabContentsWrapper(); | 37 browser()->GetSelectedTabContentsWrapper(); |
43 ASSERT_TRUE(initiator_tab); | 38 ASSERT_TRUE(initiator_tab); |
44 EXPECT_EQ(0U, GetConstrainedWindowCount(initiator_tab)); | |
45 | 39 |
46 scoped_refptr<printing::PrintPreviewTabController> | 40 scoped_refptr<printing::PrintPreviewTabController> |
47 controller(new printing::PrintPreviewTabController()); | 41 controller(new printing::PrintPreviewTabController()); |
48 ASSERT_TRUE(controller); | 42 ASSERT_TRUE(controller); |
49 | 43 |
50 TabContentsWrapper* preview_tab = | 44 TabContentsWrapper* preview_tab = |
51 controller->GetOrCreatePreviewTab(initiator_tab); | 45 controller->GetOrCreatePreviewTab(initiator_tab); |
52 | 46 |
53 EXPECT_NE(initiator_tab, preview_tab); | 47 EXPECT_NE(initiator_tab, preview_tab); |
54 EXPECT_EQ(1, browser()->tab_count()); | 48 EXPECT_EQ(2, browser()->tab_count()); |
55 EXPECT_EQ(1U, GetConstrainedWindowCount(initiator_tab)); | |
56 | 49 |
57 PrintPreviewUI* preview_ui = | 50 PrintPreviewUI* preview_ui = |
58 reinterpret_cast<PrintPreviewUI*>(preview_tab->web_ui()); | 51 reinterpret_cast<PrintPreviewUI*>(preview_tab->web_ui()); |
59 ASSERT_TRUE(preview_ui != NULL); | 52 ASSERT_TRUE(preview_ui != NULL); |
60 | 53 |
61 scoped_refptr<RefCountedBytes> data; | 54 scoped_refptr<RefCountedBytes> data; |
62 preview_ui->GetPrintPreviewDataForIndex( | 55 preview_ui->GetPrintPreviewDataForIndex( |
63 printing::COMPLETE_PREVIEW_DOCUMENT_INDEX, | 56 printing::COMPLETE_PREVIEW_DOCUMENT_INDEX, |
64 &data); | 57 &data); |
65 EXPECT_EQ(NULL, data.get()); | 58 EXPECT_EQ(NULL, data.get()); |
(...skipping 39 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
105 ASSERT_TRUE(initiator_tab); | 98 ASSERT_TRUE(initiator_tab); |
106 | 99 |
107 scoped_refptr<printing::PrintPreviewTabController> | 100 scoped_refptr<printing::PrintPreviewTabController> |
108 controller(new printing::PrintPreviewTabController()); | 101 controller(new printing::PrintPreviewTabController()); |
109 ASSERT_TRUE(controller); | 102 ASSERT_TRUE(controller); |
110 | 103 |
111 TabContentsWrapper* preview_tab = | 104 TabContentsWrapper* preview_tab = |
112 controller->GetOrCreatePreviewTab(initiator_tab); | 105 controller->GetOrCreatePreviewTab(initiator_tab); |
113 | 106 |
114 EXPECT_NE(initiator_tab, preview_tab); | 107 EXPECT_NE(initiator_tab, preview_tab); |
115 EXPECT_EQ(1, browser()->tab_count()); | 108 EXPECT_EQ(2, browser()->tab_count()); |
116 EXPECT_EQ(1U, GetConstrainedWindowCount(initiator_tab)); | |
117 | 109 |
118 PrintPreviewUI* preview_ui = | 110 PrintPreviewUI* preview_ui = |
119 reinterpret_cast<PrintPreviewUI*>(preview_tab->web_ui()); | 111 reinterpret_cast<PrintPreviewUI*>(preview_tab->web_ui()); |
120 ASSERT_TRUE(preview_ui != NULL); | 112 ASSERT_TRUE(preview_ui != NULL); |
121 | 113 |
122 scoped_refptr<RefCountedBytes> data; | 114 scoped_refptr<RefCountedBytes> data; |
123 preview_ui->GetPrintPreviewDataForIndex(printing::FIRST_PAGE_INDEX, &data); | 115 preview_ui->GetPrintPreviewDataForIndex(printing::FIRST_PAGE_INDEX, &data); |
124 EXPECT_EQ(NULL, data.get()); | 116 EXPECT_EQ(NULL, data.get()); |
125 | 117 |
126 std::vector<unsigned char> preview_data(blob1, blob1 + sizeof(blob1)); | 118 std::vector<unsigned char> preview_data(blob1, blob1 + sizeof(blob1)); |
(...skipping 46 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
173 ASSERT_TRUE(initiator_tab); | 165 ASSERT_TRUE(initiator_tab); |
174 | 166 |
175 scoped_refptr<printing::PrintPreviewTabController> | 167 scoped_refptr<printing::PrintPreviewTabController> |
176 controller(new printing::PrintPreviewTabController()); | 168 controller(new printing::PrintPreviewTabController()); |
177 ASSERT_TRUE(controller); | 169 ASSERT_TRUE(controller); |
178 | 170 |
179 TabContentsWrapper* preview_tab = | 171 TabContentsWrapper* preview_tab = |
180 controller->GetOrCreatePreviewTab(initiator_tab); | 172 controller->GetOrCreatePreviewTab(initiator_tab); |
181 | 173 |
182 EXPECT_NE(initiator_tab, preview_tab); | 174 EXPECT_NE(initiator_tab, preview_tab); |
183 EXPECT_EQ(1, browser()->tab_count()); | 175 EXPECT_EQ(2, browser()->tab_count()); |
184 EXPECT_EQ(1U, GetConstrainedWindowCount(initiator_tab)); | |
185 | 176 |
186 PrintPreviewUI* preview_ui = | 177 PrintPreviewUI* preview_ui = |
187 reinterpret_cast<PrintPreviewUI*>(preview_tab->web_ui()); | 178 reinterpret_cast<PrintPreviewUI*>(preview_tab->web_ui()); |
188 ASSERT_TRUE(preview_ui != NULL); | 179 ASSERT_TRUE(preview_ui != NULL); |
189 | 180 |
190 // Test with invalid |preview_ui_addr|. | 181 // Test with invalid |preview_ui_addr|. |
191 bool cancel = false; | 182 bool cancel = false; |
192 preview_ui->GetCurrentPrintPreviewStatus("invalid", 0, &cancel); | 183 preview_ui->GetCurrentPrintPreviewStatus("invalid", 0, &cancel); |
193 EXPECT_TRUE(cancel); | 184 EXPECT_TRUE(cancel); |
194 | 185 |
(...skipping 18 matching lines...) Expand all Loading... |
213 cancel = false; | 204 cancel = false; |
214 preview_ui->GetCurrentPrintPreviewStatus(preview_ui_addr, kFirstRequestId, | 205 preview_ui->GetCurrentPrintPreviewStatus(preview_ui_addr, kFirstRequestId, |
215 &cancel); | 206 &cancel); |
216 EXPECT_TRUE(cancel); | 207 EXPECT_TRUE(cancel); |
217 | 208 |
218 cancel = true; | 209 cancel = true; |
219 preview_ui->GetCurrentPrintPreviewStatus(preview_ui_addr, kSecondRequestId, | 210 preview_ui->GetCurrentPrintPreviewStatus(preview_ui_addr, kSecondRequestId, |
220 &cancel); | 211 &cancel); |
221 EXPECT_FALSE(cancel); | 212 EXPECT_FALSE(cancel); |
222 } | 213 } |
OLD | NEW |