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
3// found in the LICENSE file.
4
5#include "base/prefs/pref_service.h"
6#include "base/strings/utf_string_conversions.h"
7#include "chrome/app/chrome_command_ids.h"
8#include "chrome/browser/content_settings/host_content_settings_map.h"
9#include "chrome/browser/profiles/profile.h"
10#include "chrome/browser/task_manager/task_manager_browsertest_util.h"
11#include "chrome/browser/ui/browser.h"
12#include "chrome/browser/ui/browser_commands.h"
13#include "chrome/browser/ui/browser_dialogs.h"
14#include "chrome/browser/ui/tabs/tab_strip_model.h"
15#include "chrome/common/pref_names.h"
16#include "chrome/common/url_constants.h"
17#include "chrome/test/base/in_process_browser_test.h"
18#include "chrome/test/base/ui_test_utils.h"
19#include "content/public/browser/web_contents.h"
20#include "content/public/browser/web_contents_observer.h"
21#include "content/public/test/browser_test_utils.h"
22#include "content/public/test/test_navigation_observer.h"
23#include "content/public/test/test_utils.h"
24#include "url/url_constants.h"
25
26#if defined(OS_WIN)
27#include "ui/aura/window.h"
28#include "ui/aura/window_tree_host.h"
29#endif
30
31using task_manager::browsertest_util::MatchAboutBlankTab;
32using task_manager::browsertest_util::MatchAnyPrint;
33using task_manager::browsertest_util::MatchAnyTab;
34using task_manager::browsertest_util::MatchPrint;
35using task_manager::browsertest_util::WaitForTaskManagerRows;
36
37namespace {
38
39class PrintPreviewTest : public InProcessBrowserTest {
40 public:
41  PrintPreviewTest() {}
42
43  void Print() {
44    content::TestNavigationObserver nav_observer(NULL);
45    nav_observer.StartWatchingNewWebContents();
46    chrome::ExecuteCommand(browser(), IDC_PRINT);
47    nav_observer.Wait();
48    nav_observer.StopWatchingNewWebContents();
49  }
50};
51
52IN_PROC_BROWSER_TEST_F(PrintPreviewTest, PrintCommands) {
53  // We start off at about:blank page.
54  // Make sure there is 1 tab and print is enabled.
55  ASSERT_EQ(1, browser()->tab_strip_model()->count());
56
57  ASSERT_TRUE(chrome::IsCommandEnabled(browser(), IDC_PRINT));
58
59#if defined(DISABLE_BASIC_PRINTING)
60  bool is_basic_print_expected = false;
61#else
62  bool is_basic_print_expected = true;
63#endif  // DISABLE_BASIC_PRINTING
64
65  ASSERT_EQ(is_basic_print_expected,
66            chrome::IsCommandEnabled(browser(), IDC_BASIC_PRINT));
67
68  // Create the print preview dialog.
69  Print();
70
71  ASSERT_FALSE(chrome::IsCommandEnabled(browser(), IDC_PRINT));
72
73  ASSERT_EQ(is_basic_print_expected,
74            chrome::IsCommandEnabled(browser(), IDC_BASIC_PRINT));
75
76  content::TestNavigationObserver reload_observer(
77      browser()->tab_strip_model()->GetActiveWebContents());
78  chrome::Reload(browser(), CURRENT_TAB);
79  reload_observer.Wait();
80
81  ASSERT_TRUE(chrome::IsCommandEnabled(browser(), IDC_PRINT));
82
83  ASSERT_EQ(is_basic_print_expected,
84            chrome::IsCommandEnabled(browser(), IDC_BASIC_PRINT));
85}
86
87// Disable the test for mac, see http://crbug/367665.
88#if defined(OS_MACOSX) && !defined(OS_IOS)
89#define MAYBE_TaskManagerNewPrintPreview DISABLED_TaskManagerNewPrintPreview
90#else
91#define MAYBE_TaskManagerNewPrintPreview TaskManagerNewPrintPreview
92#endif
93IN_PROC_BROWSER_TEST_F(PrintPreviewTest, MAYBE_TaskManagerNewPrintPreview) {
94  chrome::ShowTaskManager(browser());  // Show task manager BEFORE print dialog.
95
96  ASSERT_NO_FATAL_FAILURE(WaitForTaskManagerRows(1, MatchAboutBlankTab()));
97  ASSERT_NO_FATAL_FAILURE(WaitForTaskManagerRows(1, MatchAnyTab()));
98  ASSERT_NO_FATAL_FAILURE(WaitForTaskManagerRows(0, MatchAnyPrint()));
99
100  // Create the print preview dialog.
101  Print();
102
103  ASSERT_NO_FATAL_FAILURE(WaitForTaskManagerRows(1, MatchAboutBlankTab()));
104  ASSERT_NO_FATAL_FAILURE(WaitForTaskManagerRows(1, MatchAnyTab()));
105  ASSERT_NO_FATAL_FAILURE(WaitForTaskManagerRows(1, MatchAnyPrint()));
106  ASSERT_NO_FATAL_FAILURE(
107      WaitForTaskManagerRows(1, MatchPrint(url::kAboutBlankURL)));
108}
109
110// http://crbug/367665.
111IN_PROC_BROWSER_TEST_F(PrintPreviewTest,
112                       DISABLED_TaskManagerExistingPrintPreview) {
113  // Create the print preview dialog.
114  Print();
115
116  chrome::ShowTaskManager(browser());  // Show task manager AFTER print dialog.
117
118  ASSERT_NO_FATAL_FAILURE(WaitForTaskManagerRows(1, MatchAboutBlankTab()));
119  ASSERT_NO_FATAL_FAILURE(WaitForTaskManagerRows(1, MatchAnyTab()));
120  ASSERT_NO_FATAL_FAILURE(WaitForTaskManagerRows(1, MatchAnyPrint()));
121  ASSERT_NO_FATAL_FAILURE(
122      WaitForTaskManagerRows(1, MatchPrint(url::kAboutBlankURL)));
123}
124
125#if defined(OS_WIN)
126// http://crbug.com/396360
127IN_PROC_BROWSER_TEST_F(PrintPreviewTest, DISABLED_NoCrashOnCloseWithOtherTabs) {
128  // Now print preview.
129  Print();
130
131  ui_test_utils::NavigateToURLWithDisposition(
132      browser(), GURL("about:blank"), NEW_FOREGROUND_TAB,
133      ui_test_utils::BROWSER_TEST_WAIT_FOR_NAVIGATION);
134
135  browser()->tab_strip_model()->ActivateTabAt(0, true);
136
137  // Navigate main tab to hide print preview.
138  ui_test_utils::NavigateToURL(browser(), GURL("about:blank"));
139
140  browser()->tab_strip_model()->ActivateTabAt(1, true);
141}
142#endif  // defined(OS_WIN)
143
144}  // namespace
145