Index: chrome/browser/ui/webui/print_preview_ui_unittest.cc |
=================================================================== |
--- chrome/browser/ui/webui/print_preview_ui_unittest.cc (revision 110055) |
+++ chrome/browser/ui/webui/print_preview_ui_unittest.cc (working copy) |
@@ -8,6 +8,7 @@ |
#include "base/memory/ref_counted_memory.h" |
#include "chrome/browser/printing/print_preview_tab_controller.h" |
#include "chrome/browser/ui/browser_list.h" |
+#include "chrome/browser/ui/constrained_window_tab_helper.h" |
#include "chrome/browser/ui/tab_contents/tab_contents_wrapper.h" |
#include "chrome/browser/ui/webui/print_preview_ui.h" |
#include "chrome/common/chrome_switches.h" |
@@ -21,6 +22,10 @@ |
const unsigned char blob1[] = |
"12346102356120394751634516591348710478123649165419234519234512349134"; |
+size_t GetConstrainedWindowCount(TabContentsWrapper* tab) { |
+ return tab->constrained_window_tab_helper()->constrained_window_count(); |
+} |
+ |
} // namespace |
typedef BrowserWithTestWindowTest PrintPreviewUITest; |
@@ -36,6 +41,7 @@ |
TabContentsWrapper* initiator_tab = |
browser()->GetSelectedTabContentsWrapper(); |
ASSERT_TRUE(initiator_tab); |
+ EXPECT_EQ(0U, GetConstrainedWindowCount(initiator_tab)); |
scoped_refptr<printing::PrintPreviewTabController> |
controller(new printing::PrintPreviewTabController()); |
@@ -45,7 +51,8 @@ |
controller->GetOrCreatePreviewTab(initiator_tab); |
EXPECT_NE(initiator_tab, preview_tab); |
- EXPECT_EQ(2, browser()->tab_count()); |
+ EXPECT_EQ(1, browser()->tab_count()); |
+ EXPECT_EQ(1U, GetConstrainedWindowCount(initiator_tab)); |
PrintPreviewUI* preview_ui = |
reinterpret_cast<PrintPreviewUI*>(preview_tab->web_ui()); |
@@ -105,7 +112,8 @@ |
controller->GetOrCreatePreviewTab(initiator_tab); |
EXPECT_NE(initiator_tab, preview_tab); |
- EXPECT_EQ(2, browser()->tab_count()); |
+ EXPECT_EQ(1, browser()->tab_count()); |
+ EXPECT_EQ(1U, GetConstrainedWindowCount(initiator_tab)); |
PrintPreviewUI* preview_ui = |
reinterpret_cast<PrintPreviewUI*>(preview_tab->web_ui()); |
@@ -172,7 +180,8 @@ |
controller->GetOrCreatePreviewTab(initiator_tab); |
EXPECT_NE(initiator_tab, preview_tab); |
- EXPECT_EQ(2, browser()->tab_count()); |
+ EXPECT_EQ(1, browser()->tab_count()); |
+ EXPECT_EQ(1U, GetConstrainedWindowCount(initiator_tab)); |
PrintPreviewUI* preview_ui = |
reinterpret_cast<PrintPreviewUI*>(preview_tab->web_ui()); |