| // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "base/strings/string16.h" |
| #include "base/strings/utf_string_conversions.h" |
| #include "chrome/browser/autocomplete/autocomplete_controller.h" |
| #include "chrome/browser/autocomplete/autocomplete_match.h" |
| #include "chrome/browser/autocomplete/autocomplete_result.h" |
| #include "chrome/browser/autocomplete/chrome_autocomplete_scheme_classifier.h" |
| #include "chrome/browser/extensions/api/omnibox/omnibox_api_testbase.h" |
| #include "chrome/browser/profiles/profile.h" |
| #include "chrome/browser/search_engines/template_url_service_factory.h" |
| #include "chrome/browser/ui/browser.h" |
| #include "chrome/browser/ui/location_bar/location_bar.h" |
| #include "chrome/browser/ui/omnibox/omnibox_view.h" |
| #include "chrome/test/base/ui_test_utils.h" |
| #include "components/autocomplete/autocomplete_input.h" |
| #include "components/metrics/proto/omnibox_event.pb.h" |
| #include "ui/base/window_open_disposition.h" |
| |
| using base::ASCIIToUTF16; |
| using metrics::OmniboxEventProto; |
| |
| IN_PROC_BROWSER_TEST_F(OmniboxApiTest, Basic) { |
| ASSERT_TRUE(RunExtensionTest("omnibox")) << message_; |
| |
| // The results depend on the TemplateURLService being loaded. Make sure it is |
| // loaded so that the autocomplete results are consistent. |
| Profile* profile = browser()->profile(); |
| ui_test_utils::WaitForTemplateURLServiceToLoad( |
| TemplateURLServiceFactory::GetForProfile(profile)); |
| |
| AutocompleteController* autocomplete_controller = |
| GetAutocompleteController(browser()); |
| |
| // Test that our extension's keyword is suggested to us when we partially type |
| // it. |
| { |
| autocomplete_controller->Start(AutocompleteInput( |
| ASCIIToUTF16("keywor"), base::string16::npos, base::string16(), GURL(), |
| OmniboxEventProto::NTP, true, false, true, true, |
| ChromeAutocompleteSchemeClassifier(profile))); |
| WaitForAutocompleteDone(autocomplete_controller); |
| EXPECT_TRUE(autocomplete_controller->done()); |
| |
| // Now, peek into the controller to see if it has the results we expect. |
| // First result should be to search for what was typed, second should be to |
| // enter "extension keyword" mode. |
| const AutocompleteResult& result = autocomplete_controller->result(); |
| ASSERT_EQ(2U, result.size()) << AutocompleteResultAsString(result); |
| AutocompleteMatch match = result.match_at(0); |
| EXPECT_EQ(AutocompleteMatchType::SEARCH_WHAT_YOU_TYPED, match.type); |
| EXPECT_FALSE(match.deletable); |
| |
| match = result.match_at(1); |
| EXPECT_EQ(ASCIIToUTF16("keyword"), match.keyword); |
| } |
| |
| // Test that our extension can send suggestions back to us. |
| { |
| autocomplete_controller->Start(AutocompleteInput( |
| ASCIIToUTF16("keyword suggestio"), base::string16::npos, |
| base::string16(), GURL(), OmniboxEventProto::NTP, true, false, true, |
| true, ChromeAutocompleteSchemeClassifier(profile))); |
| WaitForAutocompleteDone(autocomplete_controller); |
| EXPECT_TRUE(autocomplete_controller->done()); |
| |
| // Now, peek into the controller to see if it has the results we expect. |
| // First result should be to invoke the keyword with what we typed, 2-4 |
| // should be to invoke with suggestions from the extension, and the last |
| // should be to search for what we typed. |
| const AutocompleteResult& result = autocomplete_controller->result(); |
| ASSERT_EQ(5U, result.size()) << AutocompleteResultAsString(result); |
| |
| EXPECT_EQ(ASCIIToUTF16("keyword"), result.match_at(0).keyword); |
| EXPECT_EQ(ASCIIToUTF16("keyword suggestio"), |
| result.match_at(0).fill_into_edit); |
| EXPECT_EQ(AutocompleteMatchType::SEARCH_OTHER_ENGINE, |
| result.match_at(0).type); |
| EXPECT_EQ(AutocompleteProvider::TYPE_KEYWORD, |
| result.match_at(0).provider->type()); |
| EXPECT_EQ(ASCIIToUTF16("keyword"), result.match_at(1).keyword); |
| EXPECT_EQ(ASCIIToUTF16("keyword suggestion1"), |
| result.match_at(1).fill_into_edit); |
| EXPECT_EQ(AutocompleteProvider::TYPE_KEYWORD, |
| result.match_at(1).provider->type()); |
| EXPECT_EQ(ASCIIToUTF16("keyword"), result.match_at(2).keyword); |
| EXPECT_EQ(ASCIIToUTF16("keyword suggestion2"), |
| result.match_at(2).fill_into_edit); |
| EXPECT_EQ(AutocompleteProvider::TYPE_KEYWORD, |
| result.match_at(2).provider->type()); |
| EXPECT_EQ(ASCIIToUTF16("keyword"), result.match_at(3).keyword); |
| EXPECT_EQ(ASCIIToUTF16("keyword suggestion3"), |
| result.match_at(3).fill_into_edit); |
| EXPECT_EQ(AutocompleteProvider::TYPE_KEYWORD, |
| result.match_at(3).provider->type()); |
| |
| base::string16 description = |
| ASCIIToUTF16("Description with style: <match>, [dim], (url till end)"); |
| EXPECT_EQ(description, result.match_at(1).contents); |
| ASSERT_EQ(6u, result.match_at(1).contents_class.size()); |
| |
| EXPECT_EQ(0u, |
| result.match_at(1).contents_class[0].offset); |
| EXPECT_EQ(ACMatchClassification::NONE, |
| result.match_at(1).contents_class[0].style); |
| |
| EXPECT_EQ(description.find('<'), |
| result.match_at(1).contents_class[1].offset); |
| EXPECT_EQ(ACMatchClassification::MATCH, |
| result.match_at(1).contents_class[1].style); |
| |
| EXPECT_EQ(description.find('>') + 1u, |
| result.match_at(1).contents_class[2].offset); |
| EXPECT_EQ(ACMatchClassification::NONE, |
| result.match_at(1).contents_class[2].style); |
| |
| EXPECT_EQ(description.find('['), |
| result.match_at(1).contents_class[3].offset); |
| EXPECT_EQ(ACMatchClassification::DIM, |
| result.match_at(1).contents_class[3].style); |
| |
| EXPECT_EQ(description.find(']') + 1u, |
| result.match_at(1).contents_class[4].offset); |
| EXPECT_EQ(ACMatchClassification::NONE, |
| result.match_at(1).contents_class[4].style); |
| |
| EXPECT_EQ(description.find('('), |
| result.match_at(1).contents_class[5].offset); |
| EXPECT_EQ(ACMatchClassification::URL, |
| result.match_at(1).contents_class[5].style); |
| |
| AutocompleteMatch match = result.match_at(4); |
| EXPECT_EQ(AutocompleteMatchType::SEARCH_WHAT_YOU_TYPED, match.type); |
| EXPECT_EQ(AutocompleteProvider::TYPE_SEARCH, |
| result.match_at(4).provider->type()); |
| EXPECT_FALSE(match.deletable); |
| } |
| |
| // Flaky, see http://crbug.com/167158 |
| /* |
| { |
| LocationBar* location_bar = GetLocationBar(browser()); |
| ResultCatcher catcher; |
| OmniboxView* omnibox_view = location_bar->GetOmniboxView(); |
| omnibox_view->OnBeforePossibleChange(); |
| omnibox_view->SetUserText(ASCIIToUTF16("keyword command")); |
| omnibox_view->OnAfterPossibleChange(); |
| location_bar->AcceptInput(); |
| // This checks that the keyword provider (via javascript) |
| // gets told to navigate to the string "command". |
| EXPECT_TRUE(catcher.GetNextResult()) << catcher.message(); |
| } |
| */ |
| } |
| |
| IN_PROC_BROWSER_TEST_F(OmniboxApiTest, OnInputEntered) { |
| ASSERT_TRUE(RunExtensionTest("omnibox")) << message_; |
| Profile* profile = browser()->profile(); |
| ui_test_utils::WaitForTemplateURLServiceToLoad( |
| TemplateURLServiceFactory::GetForProfile(profile)); |
| |
| LocationBar* location_bar = GetLocationBar(browser()); |
| OmniboxView* omnibox_view = location_bar->GetOmniboxView(); |
| ResultCatcher catcher; |
| AutocompleteController* autocomplete_controller = |
| GetAutocompleteController(browser()); |
| omnibox_view->OnBeforePossibleChange(); |
| omnibox_view->SetUserText(ASCIIToUTF16("keyword command")); |
| omnibox_view->OnAfterPossibleChange(); |
| |
| autocomplete_controller->Start(AutocompleteInput( |
| ASCIIToUTF16("keyword command"), base::string16::npos, base::string16(), |
| GURL(), OmniboxEventProto::NTP, true, false, true, true, |
| ChromeAutocompleteSchemeClassifier(profile))); |
| omnibox_view->model()->AcceptInput(CURRENT_TAB, false); |
| WaitForAutocompleteDone(autocomplete_controller); |
| EXPECT_TRUE(autocomplete_controller->done()); |
| EXPECT_TRUE(catcher.GetNextResult()) << catcher.message(); |
| |
| omnibox_view->OnBeforePossibleChange(); |
| omnibox_view->SetUserText(ASCIIToUTF16("keyword newtab")); |
| omnibox_view->OnAfterPossibleChange(); |
| WaitForAutocompleteDone(autocomplete_controller); |
| EXPECT_TRUE(autocomplete_controller->done()); |
| |
| autocomplete_controller->Start(AutocompleteInput( |
| ASCIIToUTF16("keyword newtab"), base::string16::npos, base::string16(), |
| GURL(), OmniboxEventProto::NTP, true, false, true, true, |
| ChromeAutocompleteSchemeClassifier(profile))); |
| omnibox_view->model()->AcceptInput(NEW_FOREGROUND_TAB, false); |
| WaitForAutocompleteDone(autocomplete_controller); |
| EXPECT_TRUE(autocomplete_controller->done()); |
| EXPECT_TRUE(catcher.GetNextResult()) << catcher.message(); |
| } |
| |
| // Tests that we get suggestions from and send input to the incognito context |
| // of an incognito split mode extension. |
| // http://crbug.com/100927 |
| // Test is flaky: http://crbug.com/101219 |
| IN_PROC_BROWSER_TEST_F(OmniboxApiTest, DISABLED_IncognitoSplitMode) { |
| Profile* profile = browser()->profile(); |
| ResultCatcher catcher_incognito; |
| catcher_incognito.RestrictToProfile(profile->GetOffTheRecordProfile()); |
| |
| ASSERT_TRUE(RunExtensionTestIncognito("omnibox")) << message_; |
| |
| // Open an incognito window and wait for the incognito extension process to |
| // respond. |
| Browser* incognito_browser = CreateIncognitoBrowser(); |
| ASSERT_TRUE(catcher_incognito.GetNextResult()) << catcher_incognito.message(); |
| |
| // The results depend on the TemplateURLService being loaded. Make sure it is |
| // loaded so that the autocomplete results are consistent. |
| ui_test_utils::WaitForTemplateURLServiceToLoad( |
| TemplateURLServiceFactory::GetForProfile(browser()->profile())); |
| |
| LocationBar* location_bar = GetLocationBar(incognito_browser); |
| AutocompleteController* autocomplete_controller = |
| GetAutocompleteController(incognito_browser); |
| |
| // Test that we get the incognito-specific suggestions. |
| { |
| autocomplete_controller->Start(AutocompleteInput( |
| ASCIIToUTF16("keyword suggestio"), base::string16::npos, |
| base::string16(), GURL(), OmniboxEventProto::NTP, true, false, true, |
| true, ChromeAutocompleteSchemeClassifier(profile))); |
| WaitForAutocompleteDone(autocomplete_controller); |
| EXPECT_TRUE(autocomplete_controller->done()); |
| |
| // First result should be to invoke the keyword with what we typed, 2-4 |
| // should be to invoke with suggestions from the extension, and the last |
| // should be to search for what we typed. |
| const AutocompleteResult& result = autocomplete_controller->result(); |
| ASSERT_EQ(5U, result.size()) << AutocompleteResultAsString(result); |
| ASSERT_FALSE(result.match_at(0).keyword.empty()); |
| EXPECT_EQ(ASCIIToUTF16("keyword suggestion3 incognito"), |
| result.match_at(3).fill_into_edit); |
| } |
| |
| // Test that our input is sent to the incognito context. The test will do a |
| // text comparison and succeed only if "command incognito" is sent to the |
| // incognito context. |
| { |
| ResultCatcher catcher; |
| autocomplete_controller->Start(AutocompleteInput( |
| ASCIIToUTF16("keyword command incognito"), base::string16::npos, |
| base::string16(), GURL(), OmniboxEventProto::NTP, true, false, true, |
| true, ChromeAutocompleteSchemeClassifier(profile))); |
| location_bar->AcceptInput(); |
| EXPECT_TRUE(catcher.GetNextResult()) << catcher.message(); |
| } |
| } |