| 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 32 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 43 bool dialog_destroyed() { return dialog_destroyed_; } | 43 bool dialog_destroyed() { return dialog_destroyed_; } |
| 44 | 44 |
| 45 private: | 45 private: |
| 46 virtual void WebContentsDestroyed(WebContents* contents) OVERRIDE { | 46 virtual void WebContentsDestroyed(WebContents* contents) OVERRIDE { |
| 47 dialog_destroyed_ = true; | 47 dialog_destroyed_ = true; |
| 48 } | 48 } |
| 49 | 49 |
| 50 bool dialog_destroyed_; | 50 bool dialog_destroyed_; |
| 51 }; | 51 }; |
| 52 | 52 |
| 53 // Test to verify that when a initiator tab navigates, we can create a new | 53 // Test to verify that when a initiator navigates, we can create a new preview |
| 54 // preview dialog for the new tab contents. | 54 // dialog for the new tab contents. |
| 55 IN_PROC_BROWSER_TEST_F(PrintPreviewDialogControllerBrowserTest, | 55 IN_PROC_BROWSER_TEST_F(PrintPreviewDialogControllerBrowserTest, |
| 56 NavigateFromInitiatorTab) { | 56 NavigateFromInitiator) { |
| 57 // Create a reference to initiator tab contents. | 57 // Create a reference to initiator contents. |
| 58 WebContents* initiator_tab = | 58 WebContents* initiator = browser()->tab_strip_model()->GetActiveWebContents(); |
| 59 browser()->tab_strip_model()->GetActiveWebContents(); | 59 ASSERT_TRUE(initiator); |
| 60 ASSERT_TRUE(initiator_tab); | |
| 61 | 60 |
| 62 printing::PrintPreviewDialogController* dialog_controller = | 61 printing::PrintPreviewDialogController* dialog_controller = |
| 63 printing::PrintPreviewDialogController::GetInstance(); | 62 printing::PrintPreviewDialogController::GetInstance(); |
| 64 ASSERT_TRUE(dialog_controller); | 63 ASSERT_TRUE(dialog_controller); |
| 65 | 64 |
| 66 // Get the preview dialog for the initiator tab. | 65 // Get the preview dialog for the initiator. |
| 67 printing::PrintViewManager* print_view_manager = | 66 printing::PrintViewManager* print_view_manager = |
| 68 printing::PrintViewManager::FromWebContents(initiator_tab); | 67 printing::PrintViewManager::FromWebContents(initiator); |
| 69 print_view_manager->PrintPreviewNow(false); | 68 print_view_manager->PrintPreviewNow(false); |
| 70 WebContents* preview_dialog = | 69 WebContents* preview_dialog = |
| 71 dialog_controller->GetOrCreatePreviewDialog(initiator_tab); | 70 dialog_controller->GetOrCreatePreviewDialog(initiator); |
| 72 | 71 |
| 73 // Check a new print preview dialog got created. | 72 // Check a new print preview dialog got created. |
| 74 ASSERT_TRUE(preview_dialog); | 73 ASSERT_TRUE(preview_dialog); |
| 75 ASSERT_NE(initiator_tab, preview_dialog); | 74 ASSERT_NE(initiator, preview_dialog); |
| 76 | 75 |
| 77 // Navigate in the initiator tab. Make sure navigating destroys the print | 76 // Navigate in the initiator. Make sure navigating destroys the print preview |
| 78 // preview dialog. | 77 // dialog. |
| 79 PrintPreviewDialogDestroyedObserver observer(preview_dialog); | 78 PrintPreviewDialogDestroyedObserver observer(preview_dialog); |
| 80 GURL url(chrome::kChromeUINewTabURL); | 79 GURL url(chrome::kChromeUINewTabURL); |
| 81 ui_test_utils::NavigateToURL(browser(), url); | 80 ui_test_utils::NavigateToURL(browser(), url); |
| 82 ASSERT_TRUE(observer.dialog_destroyed()); | 81 ASSERT_TRUE(observer.dialog_destroyed()); |
| 83 | 82 |
| 84 // Get the print preview dialog for the initiator tab. | 83 // Get the print preview dialog for the initiator. |
| 85 print_view_manager->PrintPreviewNow(false); | 84 print_view_manager->PrintPreviewNow(false); |
| 86 WebContents* new_preview_dialog = | 85 WebContents* new_preview_dialog = |
| 87 dialog_controller->GetOrCreatePreviewDialog(initiator_tab); | 86 dialog_controller->GetOrCreatePreviewDialog(initiator); |
| 88 | 87 |
| 89 // Check a new preview dialog got created. | 88 // Check a new preview dialog got created. |
| 90 EXPECT_TRUE(new_preview_dialog); | 89 EXPECT_TRUE(new_preview_dialog); |
| 91 } | 90 } |
| 92 | 91 |
| 93 // Test to verify that after reloading the initiator tab, it creates a new | 92 // Test to verify that after reloading the initiator, it creates a new print |
| 94 // print preview dialog. | 93 // preview dialog. |
| 95 IN_PROC_BROWSER_TEST_F(PrintPreviewDialogControllerBrowserTest, | 94 IN_PROC_BROWSER_TEST_F(PrintPreviewDialogControllerBrowserTest, |
| 96 ReloadInitiatorTab) { | 95 ReloadInitiator) { |
| 97 // Create a reference to initiator tab contents. | 96 // Create a reference to initiator contents. |
| 98 WebContents* initiator_tab = | 97 WebContents* initiator = browser()->tab_strip_model()->GetActiveWebContents(); |
| 99 browser()->tab_strip_model()->GetActiveWebContents(); | 98 ASSERT_TRUE(initiator); |
| 100 ASSERT_TRUE(initiator_tab); | |
| 101 | 99 |
| 102 printing::PrintPreviewDialogController* dialog_controller = | 100 printing::PrintPreviewDialogController* dialog_controller = |
| 103 printing::PrintPreviewDialogController::GetInstance(); | 101 printing::PrintPreviewDialogController::GetInstance(); |
| 104 ASSERT_TRUE(dialog_controller); | 102 ASSERT_TRUE(dialog_controller); |
| 105 | 103 |
| 106 // Create a preview dialog for the initiator tab. | 104 // Create a preview dialog for the initiator. |
| 107 printing::PrintViewManager* print_view_manager = | 105 printing::PrintViewManager* print_view_manager = |
| 108 printing::PrintViewManager::FromWebContents(initiator_tab); | 106 printing::PrintViewManager::FromWebContents(initiator); |
| 109 print_view_manager->PrintPreviewNow(false); | 107 print_view_manager->PrintPreviewNow(false); |
| 110 WebContents* preview_dialog = | 108 WebContents* preview_dialog = |
| 111 dialog_controller->GetOrCreatePreviewDialog(initiator_tab); | 109 dialog_controller->GetOrCreatePreviewDialog(initiator); |
| 112 | 110 |
| 113 // Check a new print preview dialog got created. | 111 // Check a new print preview dialog got created. |
| 114 ASSERT_TRUE(preview_dialog); | 112 ASSERT_TRUE(preview_dialog); |
| 115 ASSERT_NE(initiator_tab, preview_dialog); | 113 ASSERT_NE(initiator, preview_dialog); |
| 116 | 114 |
| 117 // Reload the initiator tab. Make sure reloading destroys the print preview | 115 // Reload the initiator. Make sure reloading destroys the print preview |
| 118 // dialog. | 116 // dialog. |
| 119 PrintPreviewDialogDestroyedObserver dialog_destroyed_observer(preview_dialog); | 117 PrintPreviewDialogDestroyedObserver dialog_destroyed_observer(preview_dialog); |
| 120 content::WindowedNotificationObserver notification_observer( | 118 content::WindowedNotificationObserver notification_observer( |
| 121 content::NOTIFICATION_LOAD_STOP, | 119 content::NOTIFICATION_LOAD_STOP, |
| 122 content::NotificationService::AllSources()); | 120 content::NotificationService::AllSources()); |
| 123 chrome::Reload(browser(), CURRENT_TAB); | 121 chrome::Reload(browser(), CURRENT_TAB); |
| 124 notification_observer.Wait(); | 122 notification_observer.Wait(); |
| 125 ASSERT_TRUE(dialog_destroyed_observer.dialog_destroyed()); | 123 ASSERT_TRUE(dialog_destroyed_observer.dialog_destroyed()); |
| 126 | 124 |
| 127 // Create a preview dialog for the initiator tab. | 125 // Create a preview dialog for the initiator. |
| 128 print_view_manager->PrintPreviewNow(false); | 126 print_view_manager->PrintPreviewNow(false); |
| 129 WebContents* new_preview_dialog = | 127 WebContents* new_preview_dialog = |
| 130 dialog_controller->GetOrCreatePreviewDialog(initiator_tab); | 128 dialog_controller->GetOrCreatePreviewDialog(initiator); |
| 131 | 129 |
| 132 EXPECT_TRUE(new_preview_dialog); | 130 EXPECT_TRUE(new_preview_dialog); |
| 133 } | 131 } |
| OLD | NEW |