1 // Copyright 2014 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 "chrome/browser/chrome_notification_types.h"
6 #include "chrome/browser/sessions/tab_restore_service.h"
7 #include "chrome/browser/sessions/tab_restore_service_delegate.h"
8 #include "chrome/browser/sessions/tab_restore_service_factory.h"
9 #include "chrome/browser/ui/browser.h"
10 #include "chrome/browser/ui/browser_commands.h"
11 #include "chrome/browser/ui/browser_list.h"
12 #include "chrome/browser/ui/tabs/tab_strip_model.h"
13 #include "chrome/browser/ui/toolbar/recent_tabs_sub_menu_model.h"
14 #include "chrome/test/base/in_process_browser_test.h"
15 #include "chrome/test/base/interactive_test_utils.h"
16 #include "content/public/browser/web_contents.h"
17 #include "content/public/common/url_constants.h"
18 #include "content/public/test/browser_test_utils.h"
19
20 typedef InProcessBrowserTest BrowserTabRestoreTest;
21
AwaitTabsReady(content::DOMMessageQueue * message_queue,int tabs)22 void AwaitTabsReady(content::DOMMessageQueue* message_queue, int tabs) {
23 for (int i = 0; i < tabs; ++i) {
24 std::string message;
25 EXPECT_TRUE(message_queue->WaitForMessage(&message));
26 EXPECT_EQ("\"READY\"", message);
27 }
28 }
29
CheckVisbility(TabStripModel * tab_strip_model,int visible_index)30 void CheckVisbility(TabStripModel* tab_strip_model, int visible_index) {
31 for (int i = 0; i < tab_strip_model->count(); ++i) {
32 content::WebContents* contents = tab_strip_model->GetWebContentsAt(i);
33 std::string document_visibility_state;
34 const char kGetStateJS[] = "window.domAutomationController.send("
35 "window.document.visibilityState);";
36 EXPECT_TRUE(content::ExecuteScriptAndExtractString(
37 contents, kGetStateJS, &document_visibility_state));
38 if (i == visible_index) {
39 EXPECT_EQ("visible", document_visibility_state);
40 } else {
41 EXPECT_EQ("hidden", document_visibility_state);
42 }
43 }
44 }
45
CreateTestTabs(Browser * browser)46 void CreateTestTabs(Browser* browser) {
47 GURL test_page(ui_test_utils::GetTestUrl(base::FilePath(),
48 base::FilePath(FILE_PATH_LITERAL("tab-restore-visibilty.html"))));
49 ui_test_utils::NavigateToURLWithDisposition(
50 browser, test_page, NEW_FOREGROUND_TAB,
51 ui_test_utils::BROWSER_TEST_WAIT_FOR_NAVIGATION);
52 ui_test_utils::NavigateToURLWithDisposition(
53 browser, test_page, NEW_BACKGROUND_TAB,
54 ui_test_utils::BROWSER_TEST_WAIT_FOR_NAVIGATION);
55 }
56
CloseBrowser(Browser * browser)57 void CloseBrowser(Browser* browser) {
58 content::WindowedNotificationObserver close_observer(
59 chrome::NOTIFICATION_BROWSER_CLOSED,
60 content::Source<Browser>(browser));
61 chrome::CloseWindow(browser);
62 close_observer.Wait();
63 }
64
IN_PROC_BROWSER_TEST_F(BrowserTabRestoreTest,RecentTabsMenuTabDisposition)65 IN_PROC_BROWSER_TEST_F(BrowserTabRestoreTest, RecentTabsMenuTabDisposition) {
66 // Create tabs.
67 CreateTestTabs(browser());
68 EXPECT_EQ(3, browser()->tab_strip_model()->count());
69
70 // Create a new browser.
71 ui_test_utils::NavigateToURLWithDisposition(
72 browser(),
73 GURL(url::kAboutBlankURL),
74 NEW_WINDOW,
75 ui_test_utils::BROWSER_TEST_WAIT_FOR_BROWSER);
76 BrowserList* active_browser_list =
77 BrowserList::GetInstance(browser()->host_desktop_type());
78 EXPECT_EQ(2u, active_browser_list->size());
79
80 // Close the first browser.
81 CloseBrowser(browser());
82 EXPECT_EQ(1u, active_browser_list->size());
83
84 // Restore tabs using the browser's recent tabs menu.
85 content::DOMMessageQueue queue;
86 Browser* browser = active_browser_list->get(0);
87 RecentTabsSubMenuModel menu(NULL, browser, NULL);
88 menu.ExecuteCommand(
89 RecentTabsSubMenuModel::GetFirstRecentTabsCommandId(), 0);
90 AwaitTabsReady(&queue, 2);
91
92 // There should be 3 restored tabs in the new browser.
93 EXPECT_EQ(2u, active_browser_list->size());
94 browser = active_browser_list->get(1);
95 EXPECT_EQ(3, browser->tab_strip_model()->count());
96
97 // The middle tab only should have visible disposition.
98 CheckVisbility(browser->tab_strip_model(), 1);
99 }
100
IN_PROC_BROWSER_TEST_F(BrowserTabRestoreTest,DelegateRestoreTabDisposition)101 IN_PROC_BROWSER_TEST_F(BrowserTabRestoreTest, DelegateRestoreTabDisposition) {
102 // Create tabs.
103 CreateTestTabs(browser());
104 EXPECT_EQ(3, browser()->tab_strip_model()->count());
105
106 // Create a new browser.
107 ui_test_utils::NavigateToURLWithDisposition(
108 browser(),
109 GURL(url::kAboutBlankURL),
110 NEW_WINDOW,
111 ui_test_utils::BROWSER_TEST_WAIT_FOR_BROWSER);
112 BrowserList* active_browser_list =
113 BrowserList::GetInstance(browser()->host_desktop_type());
114 EXPECT_EQ(2u, active_browser_list->size());
115
116 // Close the first browser.
117 CloseBrowser(browser());
118 EXPECT_EQ(1u, active_browser_list->size());
119
120 // Check the browser has a delegated restore service.
121 Browser* browser = active_browser_list->get(0);
122 TabRestoreService* service =
123 TabRestoreServiceFactory::GetForProfile(browser->profile());
124 bool has_tab_restore_service = !!service;
125 ASSERT_TRUE(has_tab_restore_service);
126 TabRestoreServiceDelegate* delegate =
127 TabRestoreServiceDelegate::FindDelegateForWebContents(
128 browser->tab_strip_model()->GetActiveWebContents());
129 bool has_tab_restore_delegate = !!delegate;
130 ASSERT_TRUE(has_tab_restore_delegate);
131
132 // Restore tabs using that delegated restore service.
133 content::DOMMessageQueue queue;
134 service->RestoreMostRecentEntry(
135 delegate, browser->host_desktop_type());
136 AwaitTabsReady(&queue, 2);
137
138 // There should be 3 restored tabs in the new browser.
139 EXPECT_EQ(2u, active_browser_list->size());
140 browser = active_browser_list->get(1);
141 EXPECT_EQ(3, browser->tab_strip_model()->count());
142
143 // The middle tab only should have visible disposition.
144 CheckVisbility(browser->tab_strip_model(), 1);
145 }
146