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/command_line.h"
6 #include "base/format_macros.h"
7 #include "base/path_service.h"
8 #include "base/strings/stringprintf.h"
9 #include "base/strings/utf_string_conversions.h"
10 #include "chrome/browser/autocomplete/autocomplete_input.h"
11 #include "chrome/browser/autocomplete/autocomplete_match.h"
12 #include "chrome/browser/autocomplete/autocomplete_provider.h"
13 #include "chrome/browser/extensions/extension_browsertest.h"
14 #include "chrome/browser/extensions/extension_service.h"
15 #include "chrome/browser/extensions/unpacked_installer.h"
16 #include "chrome/browser/history/history_service.h"
17 #include "chrome/browser/history/history_service_factory.h"
18 #include "chrome/browser/profiles/profile.h"
19 #include "chrome/browser/search_engines/template_url_service_factory.h"
20 #include "chrome/browser/ui/browser.h"
21 #include "chrome/browser/ui/browser_commands.h"
22 #include "chrome/browser/ui/browser_tabstrip.h"
23 #include "chrome/browser/ui/browser_window.h"
24 #include "chrome/browser/ui/omnibox/location_bar.h"
25 #include "chrome/browser/ui/omnibox/omnibox_popup_model.h"
26 #include "chrome/browser/ui/omnibox/omnibox_view.h"
27 #include "chrome/browser/ui/tabs/tab_strip_model.h"
28 #include "chrome/common/chrome_paths.h"
29 #include "chrome/common/url_constants.h"
30 #include "chrome/test/base/in_process_browser_test.h"
31 #include "chrome/test/base/test_switches.h"
32 #include "chrome/test/base/ui_test_utils.h"
33 #include "components/metrics/proto/omnibox_event.pb.h"
34 #include "content/public/browser/notification_service.h"
35 #include "content/public/browser/notification_types.h"
36 #include "testing/gtest/include/gtest/gtest.h"
37
38 namespace {
39
AutocompleteResultAsString(const AutocompleteResult & result)40 base::string16 AutocompleteResultAsString(const AutocompleteResult& result) {
41 std::string output(base::StringPrintf("{%" PRIuS "} ", result.size()));
42 for (size_t i = 0; i < result.size(); ++i) {
43 AutocompleteMatch match = result.match_at(i);
44 output.append(base::StringPrintf("[\"%s\" by \"%s\"] ",
45 base::UTF16ToUTF8(match.contents).c_str(),
46 match.provider->GetName()));
47 }
48 return base::UTF8ToUTF16(output);
49 }
50
51 } // namespace
52
53 class AutocompleteBrowserTest : public ExtensionBrowserTest {
54 protected:
WaitForTemplateURLServiceToLoad()55 void WaitForTemplateURLServiceToLoad() {
56 ui_test_utils::WaitForTemplateURLServiceToLoad(
57 TemplateURLServiceFactory::GetForProfile(browser()->profile()));
58 }
59
GetLocationBar() const60 LocationBar* GetLocationBar() const {
61 return browser()->window()->GetLocationBar();
62 }
63
GetAutocompleteController() const64 AutocompleteController* GetAutocompleteController() const {
65 return GetLocationBar()->GetOmniboxView()->model()->popup_model()->
66 autocomplete_controller();
67 }
68 };
69
IN_PROC_BROWSER_TEST_F(AutocompleteBrowserTest,Basic)70 IN_PROC_BROWSER_TEST_F(AutocompleteBrowserTest, Basic) {
71 #if defined(OS_WIN) && defined(USE_ASH)
72 // Disable this test in Metro+Ash for now (http://crbug.com/262796).
73 if (CommandLine::ForCurrentProcess()->HasSwitch(switches::kAshBrowserTests))
74 return;
75 #endif
76
77 WaitForTemplateURLServiceToLoad();
78 LocationBar* location_bar = GetLocationBar();
79 OmniboxView* omnibox_view = location_bar->GetOmniboxView();
80
81 EXPECT_FALSE(location_bar->GetDestinationURL().is_valid());
82 EXPECT_EQ(base::UTF8ToUTF16(url::kAboutBlankURL), omnibox_view->GetText());
83 // TODO(phajdan.jr): check state of IsSelectAll when it's consistent across
84 // platforms.
85
86 location_bar->FocusLocation(true);
87
88 EXPECT_FALSE(location_bar->GetDestinationURL().is_valid());
89 EXPECT_EQ(base::UTF8ToUTF16(url::kAboutBlankURL), omnibox_view->GetText());
90 EXPECT_TRUE(omnibox_view->IsSelectAll());
91
92 omnibox_view->SetUserText(base::ASCIIToUTF16("chrome"));
93
94 EXPECT_FALSE(location_bar->GetDestinationURL().is_valid());
95 EXPECT_EQ(base::ASCIIToUTF16("chrome"), omnibox_view->GetText());
96 EXPECT_FALSE(omnibox_view->IsSelectAll());
97
98 omnibox_view->RevertAll();
99
100 EXPECT_FALSE(location_bar->GetDestinationURL().is_valid());
101 EXPECT_EQ(base::UTF8ToUTF16(url::kAboutBlankURL), omnibox_view->GetText());
102 EXPECT_FALSE(omnibox_view->IsSelectAll());
103
104 omnibox_view->SetUserText(base::ASCIIToUTF16("chrome"));
105 location_bar->Revert();
106
107 EXPECT_FALSE(location_bar->GetDestinationURL().is_valid());
108 EXPECT_EQ(base::UTF8ToUTF16(url::kAboutBlankURL), omnibox_view->GetText());
109 EXPECT_FALSE(omnibox_view->IsSelectAll());
110 }
111
112 // Autocomplete test is flaky on ChromeOS.
113 // http://crbug.com/52928
114 #if defined(OS_CHROMEOS)
115 #define MAYBE_Autocomplete DISABLED_Autocomplete
116 #else
117 #define MAYBE_Autocomplete Autocomplete
118 #endif
119
IN_PROC_BROWSER_TEST_F(AutocompleteBrowserTest,MAYBE_Autocomplete)120 IN_PROC_BROWSER_TEST_F(AutocompleteBrowserTest, MAYBE_Autocomplete) {
121 #if defined(OS_WIN) && defined(USE_ASH)
122 // Disable this test in Metro+Ash for now (http://crbug.com/262796).
123 if (CommandLine::ForCurrentProcess()->HasSwitch(switches::kAshBrowserTests))
124 return;
125 #endif
126
127 WaitForTemplateURLServiceToLoad();
128 // The results depend on the history backend being loaded. Make sure it is
129 // loaded so that the autocomplete results are consistent.
130 ui_test_utils::WaitForHistoryToLoad(
131 HistoryServiceFactory::GetForProfile(browser()->profile(),
132 Profile::EXPLICIT_ACCESS));
133
134 LocationBar* location_bar = GetLocationBar();
135 OmniboxView* omnibox_view = location_bar->GetOmniboxView();
136 AutocompleteController* autocomplete_controller = GetAutocompleteController();
137
138 {
139 omnibox_view->model()->SetInputInProgress(true);
140 autocomplete_controller->Start(AutocompleteInput(
141 base::ASCIIToUTF16("chrome"), base::string16::npos, base::string16(),
142 GURL(), metrics::OmniboxEventProto::NTP, true, false, true, false));
143
144 EXPECT_TRUE(autocomplete_controller->done());
145 EXPECT_FALSE(location_bar->GetDestinationURL().is_valid());
146 EXPECT_TRUE(omnibox_view->GetText().empty());
147 EXPECT_TRUE(omnibox_view->IsSelectAll());
148 const AutocompleteResult& result = autocomplete_controller->result();
149 // We get two matches because we have a provider for extension apps and the
150 // Chrome Web Store is a built-in Extension app. For this test, we only care
151 // about the other match existing.
152 ASSERT_GE(result.size(), 1U) << AutocompleteResultAsString(result);
153 AutocompleteMatch match = result.match_at(0);
154 EXPECT_EQ(AutocompleteMatchType::SEARCH_WHAT_YOU_TYPED, match.type);
155 EXPECT_FALSE(match.deletable);
156 }
157
158 {
159 location_bar->Revert();
160 EXPECT_FALSE(location_bar->GetDestinationURL().is_valid());
161 EXPECT_EQ(base::UTF8ToUTF16(url::kAboutBlankURL), omnibox_view->GetText());
162 EXPECT_FALSE(omnibox_view->IsSelectAll());
163 const AutocompleteResult& result = autocomplete_controller->result();
164 EXPECT_TRUE(result.empty()) << AutocompleteResultAsString(result);
165 }
166 }
167
IN_PROC_BROWSER_TEST_F(AutocompleteBrowserTest,TabAwayRevertSelect)168 IN_PROC_BROWSER_TEST_F(AutocompleteBrowserTest, TabAwayRevertSelect) {
169 #if defined(OS_WIN) && defined(USE_ASH)
170 // Disable this test in Metro+Ash for now (http://crbug.com/262796).
171 if (CommandLine::ForCurrentProcess()->HasSwitch(switches::kAshBrowserTests))
172 return;
173 #endif
174
175 WaitForTemplateURLServiceToLoad();
176 // http://code.google.com/p/chromium/issues/detail?id=38385
177 // Make sure that tabbing away from an empty omnibox causes a revert
178 // and select all.
179 LocationBar* location_bar = GetLocationBar();
180 OmniboxView* omnibox_view = location_bar->GetOmniboxView();
181 EXPECT_EQ(base::UTF8ToUTF16(url::kAboutBlankURL), omnibox_view->GetText());
182 omnibox_view->SetUserText(base::string16());
183 content::WindowedNotificationObserver observer(
184 content::NOTIFICATION_LOAD_STOP,
185 content::NotificationService::AllSources());
186 chrome::AddSelectedTabWithURL(browser(),
187 GURL(url::kAboutBlankURL),
188 content::PAGE_TRANSITION_AUTO_TOPLEVEL);
189 observer.Wait();
190 EXPECT_EQ(base::UTF8ToUTF16(url::kAboutBlankURL), omnibox_view->GetText());
191 chrome::CloseTab(browser());
192 EXPECT_EQ(base::UTF8ToUTF16(url::kAboutBlankURL), omnibox_view->GetText());
193 EXPECT_TRUE(omnibox_view->IsSelectAll());
194 }
195
IN_PROC_BROWSER_TEST_F(AutocompleteBrowserTest,FocusSearch)196 IN_PROC_BROWSER_TEST_F(AutocompleteBrowserTest, FocusSearch) {
197 #if defined(OS_WIN) && defined(USE_ASH)
198 // Disable this test in Metro+Ash for now (http://crbug.com/262796).
199 if (CommandLine::ForCurrentProcess()->HasSwitch(switches::kAshBrowserTests))
200 return;
201 #endif
202
203 WaitForTemplateURLServiceToLoad();
204 LocationBar* location_bar = GetLocationBar();
205 OmniboxView* omnibox_view = location_bar->GetOmniboxView();
206
207 // Focus search when omnibox is blank
208 {
209 EXPECT_FALSE(location_bar->GetDestinationURL().is_valid());
210 EXPECT_EQ(base::UTF8ToUTF16(url::kAboutBlankURL), omnibox_view->GetText());
211
212 location_bar->FocusSearch();
213 EXPECT_FALSE(location_bar->GetDestinationURL().is_valid());
214 EXPECT_EQ(base::ASCIIToUTF16("?"), omnibox_view->GetText());
215
216 size_t selection_start, selection_end;
217 omnibox_view->GetSelectionBounds(&selection_start, &selection_end);
218 EXPECT_EQ(1U, selection_start);
219 EXPECT_EQ(1U, selection_end);
220 }
221
222 // Focus search when omnibox is _not_ alread in forced query mode.
223 {
224 omnibox_view->SetUserText(base::ASCIIToUTF16("foo"));
225 EXPECT_FALSE(location_bar->GetDestinationURL().is_valid());
226 EXPECT_EQ(base::ASCIIToUTF16("foo"), omnibox_view->GetText());
227
228 location_bar->FocusSearch();
229 EXPECT_FALSE(location_bar->GetDestinationURL().is_valid());
230 EXPECT_EQ(base::ASCIIToUTF16("?"), omnibox_view->GetText());
231
232 size_t selection_start, selection_end;
233 omnibox_view->GetSelectionBounds(&selection_start, &selection_end);
234 EXPECT_EQ(1U, selection_start);
235 EXPECT_EQ(1U, selection_end);
236 }
237
238 // Focus search when omnibox _is_ already in forced query mode, but no query
239 // has been typed.
240 {
241 omnibox_view->SetUserText(base::ASCIIToUTF16("?"));
242 EXPECT_FALSE(location_bar->GetDestinationURL().is_valid());
243 EXPECT_EQ(base::ASCIIToUTF16("?"), omnibox_view->GetText());
244
245 location_bar->FocusSearch();
246 EXPECT_FALSE(location_bar->GetDestinationURL().is_valid());
247 EXPECT_EQ(base::ASCIIToUTF16("?"), omnibox_view->GetText());
248
249 size_t selection_start, selection_end;
250 omnibox_view->GetSelectionBounds(&selection_start, &selection_end);
251 EXPECT_EQ(1U, selection_start);
252 EXPECT_EQ(1U, selection_end);
253 }
254
255 // Focus search when omnibox _is_ already in forced query mode, and some query
256 // has been typed.
257 {
258 omnibox_view->SetUserText(base::ASCIIToUTF16("?foo"));
259 EXPECT_FALSE(location_bar->GetDestinationURL().is_valid());
260 EXPECT_EQ(base::ASCIIToUTF16("?foo"), omnibox_view->GetText());
261
262 location_bar->FocusSearch();
263 EXPECT_FALSE(location_bar->GetDestinationURL().is_valid());
264 EXPECT_EQ(base::ASCIIToUTF16("?foo"), omnibox_view->GetText());
265
266 size_t selection_start, selection_end;
267 omnibox_view->GetSelectionBounds(&selection_start, &selection_end);
268 EXPECT_EQ(1U, std::min(selection_start, selection_end));
269 EXPECT_EQ(4U, std::max(selection_start, selection_end));
270 }
271
272 // Focus search when omnibox is in forced query mode with leading whitespace.
273 {
274 omnibox_view->SetUserText(base::ASCIIToUTF16(" ?foo"));
275 EXPECT_FALSE(location_bar->GetDestinationURL().is_valid());
276 EXPECT_EQ(base::ASCIIToUTF16(" ?foo"), omnibox_view->GetText());
277
278 location_bar->FocusSearch();
279 EXPECT_FALSE(location_bar->GetDestinationURL().is_valid());
280 EXPECT_EQ(base::ASCIIToUTF16(" ?foo"), omnibox_view->GetText());
281
282 size_t selection_start, selection_end;
283 omnibox_view->GetSelectionBounds(&selection_start, &selection_end);
284 EXPECT_EQ(4U, std::min(selection_start, selection_end));
285 EXPECT_EQ(7U, std::max(selection_start, selection_end));
286 }
287 }
288