OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 "base/command_line.h" | 5 #include "base/command_line.h" |
6 #include "chrome/browser/printing/print_preview_dialog_controller.h" | 6 #include "chrome/browser/printing/print_preview_dialog_controller.h" |
7 #include "chrome/browser/printing/print_view_manager.h" | 7 #include "chrome/browser/printing/print_view_manager.h" |
8 #include "chrome/browser/ui/browser.h" | 8 #include "chrome/browser/ui/browser.h" |
9 #include "chrome/browser/ui/browser_commands.h" | 9 #include "chrome/browser/ui/browser_commands.h" |
10 #include "chrome/browser/ui/tabs/tab_strip_model.h" | 10 #include "chrome/browser/ui/tabs/tab_strip_model.h" |
(...skipping 13 matching lines...) Expand all Loading... |
24 PrintPreviewDialogControllerBrowserTest() {} | 24 PrintPreviewDialogControllerBrowserTest() {} |
25 virtual ~PrintPreviewDialogControllerBrowserTest() {} | 25 virtual ~PrintPreviewDialogControllerBrowserTest() {} |
26 | 26 |
27 virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE { | 27 virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE { |
28 #if !defined(GOOGLE_CHROME_BUILD) | 28 #if !defined(GOOGLE_CHROME_BUILD) |
29 command_line->AppendSwitch(switches::kEnablePrintPreview); | 29 command_line->AppendSwitch(switches::kEnablePrintPreview); |
30 #endif | 30 #endif |
31 } | 31 } |
32 }; | 32 }; |
33 | 33 |
34 class TabDestroyedObserver : public content::WebContentsObserver { | 34 class PrintPreviewDialogDestroyedObserver |
| 35 : public content::WebContentsObserver { |
35 public: | 36 public: |
36 explicit TabDestroyedObserver(WebContents* contents) | 37 explicit PrintPreviewDialogDestroyedObserver(WebContents* dialog) |
37 : content::WebContentsObserver(contents), | 38 : content::WebContentsObserver(dialog), |
38 tab_destroyed_(false) { | 39 dialog_destroyed_(false) { |
39 } | 40 } |
40 virtual ~TabDestroyedObserver() {} | 41 virtual ~PrintPreviewDialogDestroyedObserver() {} |
41 | 42 |
42 bool tab_destroyed() { return tab_destroyed_; } | 43 bool dialog_destroyed() { return dialog_destroyed_; } |
43 | 44 |
44 private: | 45 private: |
45 virtual void WebContentsDestroyed(WebContents* tab) OVERRIDE { | 46 virtual void WebContentsDestroyed(WebContents* contents) OVERRIDE { |
46 tab_destroyed_ = true; | 47 dialog_destroyed_ = true; |
47 } | 48 } |
48 | 49 |
49 bool tab_destroyed_; | 50 bool dialog_destroyed_; |
50 }; | 51 }; |
51 | 52 |
52 // Test to verify that when a initiator tab navigates, we can create a new | 53 // Test to verify that when a initiator tab navigates, we can create a new |
53 // preview tab for the new tab contents. | 54 // preview tab for the new tab contents. |
54 IN_PROC_BROWSER_TEST_F(PrintPreviewDialogControllerBrowserTest, | 55 IN_PROC_BROWSER_TEST_F(PrintPreviewDialogControllerBrowserTest, |
55 NavigateFromInitiatorTab) { | 56 NavigateFromInitiatorTab) { |
56 // Lets start with one tab. | 57 // Lets start with one tab. |
57 EXPECT_EQ(1, browser()->tab_strip_model()->count()); | 58 EXPECT_EQ(1, browser()->tab_strip_model()->count()); |
58 | 59 |
59 // Create a reference to initiator tab contents. | 60 // Create a reference to initiator tab contents. |
60 WebContents* initiator_tab = | 61 WebContents* initiator_tab = |
61 browser()->tab_strip_model()->GetActiveWebContents(); | 62 browser()->tab_strip_model()->GetActiveWebContents(); |
62 ASSERT_TRUE(initiator_tab); | 63 ASSERT_TRUE(initiator_tab); |
63 | 64 |
64 printing::PrintPreviewDialogController* tab_controller = | 65 printing::PrintPreviewDialogController* dialog_controller = |
65 printing::PrintPreviewDialogController::GetInstance(); | 66 printing::PrintPreviewDialogController::GetInstance(); |
66 ASSERT_TRUE(tab_controller); | 67 ASSERT_TRUE(dialog_controller); |
67 | 68 |
68 // Get the preview tab for initiator tab. | 69 // Get the preview tab for initiator tab. |
69 printing::PrintViewManager* print_view_manager = | 70 printing::PrintViewManager* print_view_manager = |
70 printing::PrintViewManager::FromWebContents(initiator_tab); | 71 printing::PrintViewManager::FromWebContents(initiator_tab); |
71 print_view_manager->PrintPreviewNow(false); | 72 print_view_manager->PrintPreviewNow(false); |
72 WebContents* preview_tab = | 73 WebContents* preview_tab = |
73 tab_controller->GetOrCreatePreviewTab(initiator_tab); | 74 dialog_controller->GetOrCreatePreviewDialog(initiator_tab); |
74 | 75 |
75 // New print preview tab is created. | 76 // New print preview tab is created. |
76 EXPECT_EQ(1, browser()->tab_strip_model()->count()); | 77 EXPECT_EQ(1, browser()->tab_strip_model()->count()); |
77 ASSERT_TRUE(preview_tab); | 78 ASSERT_TRUE(preview_tab); |
78 ASSERT_NE(initiator_tab, preview_tab); | 79 ASSERT_NE(initiator_tab, preview_tab); |
79 TabDestroyedObserver observer(preview_tab); | 80 PrintPreviewDialogDestroyedObserver observer(preview_tab); |
80 | 81 |
81 // Navigate in the initiator tab. | 82 // Navigate in the initiator tab. |
82 GURL url(chrome::kChromeUINewTabURL); | 83 GURL url(chrome::kChromeUINewTabURL); |
83 ui_test_utils::NavigateToURL(browser(), url); | 84 ui_test_utils::NavigateToURL(browser(), url); |
84 | 85 |
85 ASSERT_TRUE(observer.tab_destroyed()); | 86 ASSERT_TRUE(observer.dialog_destroyed()); |
86 | 87 |
87 // Get the print preview tab for initiator tab. | 88 // Get the print preview tab for initiator tab. |
88 print_view_manager->PrintPreviewNow(false); | 89 print_view_manager->PrintPreviewNow(false); |
89 WebContents* new_preview_tab = | 90 WebContents* new_preview_tab = |
90 tab_controller->GetOrCreatePreviewTab(initiator_tab); | 91 dialog_controller->GetOrCreatePreviewDialog(initiator_tab); |
91 | 92 |
92 // New preview tab is created. | 93 // New preview tab is created. |
93 EXPECT_EQ(1, browser()->tab_strip_model()->count()); | 94 EXPECT_EQ(1, browser()->tab_strip_model()->count()); |
94 EXPECT_TRUE(new_preview_tab); | 95 EXPECT_TRUE(new_preview_tab); |
95 } | 96 } |
96 | 97 |
97 // Test to verify that after reloading the initiator tab, it creates a new | 98 // Test to verify that after reloading the initiator tab, it creates a new |
98 // print preview tab. | 99 // print preview tab. |
99 IN_PROC_BROWSER_TEST_F(PrintPreviewDialogControllerBrowserTest, | 100 IN_PROC_BROWSER_TEST_F(PrintPreviewDialogControllerBrowserTest, |
100 ReloadInitiatorTab) { | 101 ReloadInitiatorTab) { |
101 // Lets start with one tab. | 102 // Lets start with one tab. |
102 EXPECT_EQ(1, browser()->tab_strip_model()->count()); | 103 EXPECT_EQ(1, browser()->tab_strip_model()->count()); |
103 | 104 |
104 // Create a reference to initiator tab contents. | 105 // Create a reference to initiator tab contents. |
105 WebContents* initiator_tab = | 106 WebContents* initiator_tab = |
106 browser()->tab_strip_model()->GetActiveWebContents(); | 107 browser()->tab_strip_model()->GetActiveWebContents(); |
107 ASSERT_TRUE(initiator_tab); | 108 ASSERT_TRUE(initiator_tab); |
108 | 109 |
109 printing::PrintPreviewDialogController* tab_controller = | 110 printing::PrintPreviewDialogController* dialog_controller = |
110 printing::PrintPreviewDialogController::GetInstance(); | 111 printing::PrintPreviewDialogController::GetInstance(); |
111 ASSERT_TRUE(tab_controller); | 112 ASSERT_TRUE(dialog_controller); |
112 | 113 |
113 // Get the preview tab for initiator tab. | 114 // Get the preview tab for initiator tab. |
114 printing::PrintViewManager* print_view_manager = | 115 printing::PrintViewManager* print_view_manager = |
115 printing::PrintViewManager::FromWebContents(initiator_tab); | 116 printing::PrintViewManager::FromWebContents(initiator_tab); |
116 print_view_manager->PrintPreviewNow(false); | 117 print_view_manager->PrintPreviewNow(false); |
117 WebContents* preview_tab = | 118 WebContents* preview_tab = |
118 tab_controller->GetOrCreatePreviewTab(initiator_tab); | 119 dialog_controller->GetOrCreatePreviewDialog(initiator_tab); |
119 | 120 |
120 // New print preview tab is created. | 121 // New print preview tab is created. |
121 EXPECT_EQ(1, browser()->tab_strip_model()->count()); | 122 EXPECT_EQ(1, browser()->tab_strip_model()->count()); |
122 ASSERT_TRUE(preview_tab); | 123 ASSERT_TRUE(preview_tab); |
123 ASSERT_NE(initiator_tab, preview_tab); | 124 ASSERT_NE(initiator_tab, preview_tab); |
124 TabDestroyedObserver tab_destroyed_observer(preview_tab); | 125 PrintPreviewDialogDestroyedObserver dialog_destroyed_observer(preview_tab); |
125 | 126 |
126 // Reload the initiator tab. | 127 // Reload the initiator tab. |
127 content::WindowedNotificationObserver notification_observer( | 128 content::WindowedNotificationObserver notification_observer( |
128 content::NOTIFICATION_LOAD_STOP, | 129 content::NOTIFICATION_LOAD_STOP, |
129 content::NotificationService::AllSources()); | 130 content::NotificationService::AllSources()); |
130 chrome::Reload(browser(), CURRENT_TAB); | 131 chrome::Reload(browser(), CURRENT_TAB); |
131 notification_observer.Wait(); | 132 notification_observer.Wait(); |
132 | 133 |
133 ASSERT_TRUE(tab_destroyed_observer.tab_destroyed()); | 134 ASSERT_TRUE(dialog_destroyed_observer.dialog_destroyed()); |
134 | 135 |
135 // Get the print preview tab for initiator tab. | 136 // Get the print preview tab for initiator tab. |
136 print_view_manager->PrintPreviewNow(false); | 137 print_view_manager->PrintPreviewNow(false); |
137 WebContents* new_preview_tab = | 138 WebContents* new_preview_tab = |
138 tab_controller->GetOrCreatePreviewTab(initiator_tab); | 139 dialog_controller->GetOrCreatePreviewDialog(initiator_tab); |
139 | 140 |
140 EXPECT_EQ(1, browser()->tab_strip_model()->count()); | 141 EXPECT_EQ(1, browser()->tab_strip_model()->count()); |
141 EXPECT_TRUE(new_preview_tab); | 142 EXPECT_TRUE(new_preview_tab); |
142 } | 143 } |
OLD | NEW |