Introduces a search term extraction mechanism working for arbitrary search providers.

chrome/browser/policy/policy_browsertest.cc

 // 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 <algorithm>
 #include <string>
 #include <vector>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
@@ skipping 615 matching lines @@
                POLICY_SCOPE_USER, base::Value::CreateBooleanValue(false));
   EXPECT_TRUE(service->GetDefaultSearchProvider());
   provider_.UpdateChromePolicy(policies);
   EXPECT_FALSE(service->GetDefaultSearchProvider());
   ui_test_utils::SendToOmniboxAndSubmit(location_bar, "should not work");
   // This means that submitting won't trigger any action.
   EXPECT_FALSE(model->CurrentMatch().destination_url.is_valid());
   EXPECT_EQ(GURL(chrome::kAboutBlankURL), web_contents->GetURL());
 }
 
+IN_PROC_BROWSER_TEST_F(PolicyTest, ReplaceSearchTerms) {
+  CommandLine::ForCurrentProcess()->AppendSwitch(
+      switches::kEnableInstantExtendedAPI);
+
+  // Verifies that a default search is made using the provider configured via
+  // policy. Also checks that default search can be completely disabled.
+  const string16 kKeyword(ASCIIToUTF16("testsearch"));
+  const std::string kSearchURL("http://search.example/search?q={searchTerms}");
+  const std::string kAlternateURL0(
+      "http://search.example/search#q={searchTerms}");
+  const std::string kAlternateURL1("http://search.example/#q={searchTerms}");
+
+  TemplateURLService* service = TemplateURLServiceFactory::GetForProfile(
+      browser()->profile());
+  ui_test_utils::WaitForTemplateURLServiceToLoad(service);
+  TemplateURL* default_search = service->GetDefaultSearchProvider();
+  ASSERT_TRUE(default_search);
+  EXPECT_NE(kKeyword, default_search->keyword());
+  EXPECT_NE(kSearchURL, default_search->url());
+  EXPECT_NE(kAlternateURL0, default_search->alternate_urls()[0]);
+  EXPECT_NE(kAlternateURL1, default_search->alternate_urls()[1]);
+
+  // Override the default search provider using policies.
+  PolicyMap policies;
+  policies.Set(key::kDefaultSearchProviderEnabled, POLICY_LEVEL_MANDATORY,
+               POLICY_SCOPE_USER, base::Value::CreateBooleanValue(true));
+  policies.Set(key::kDefaultSearchProviderKeyword, POLICY_LEVEL_MANDATORY,
+               POLICY_SCOPE_USER, base::Value::CreateStringValue(kKeyword));
+  policies.Set(key::kDefaultSearchProviderSearchURL, POLICY_LEVEL_MANDATORY,
+               POLICY_SCOPE_USER, base::Value::CreateStringValue(kSearchURL));
+  base::ListValue* alternate_urls = new base::ListValue();
+  alternate_urls->AppendString(kAlternateURL0);
+  alternate_urls->AppendString(kAlternateURL1);
+  policies.Set(key::kDefaultSearchProviderAlternateURLs, POLICY_LEVEL_MANDATORY,
+               POLICY_SCOPE_USER, alternate_urls);
+  provider_.UpdateChromePolicy(policies);
+  default_search = service->GetDefaultSearchProvider();
+  ASSERT_TRUE(default_search);
+  EXPECT_EQ(kKeyword, default_search->keyword());
+  EXPECT_EQ(kSearchURL, default_search->url());
+  EXPECT_EQ(kAlternateURL0, default_search->alternate_urls()[0]);
+  EXPECT_EQ(kAlternateURL1, default_search->alternate_urls()[1]);
+
+  // Verify that searching from the omnibox does search term replacement with
+  // first URL pattern.
+  chrome::FocusLocationBar(browser());
+  LocationBar* location_bar = browser()->window()->GetLocationBar();
+  ui_test_utils::SendToOmniboxAndSubmit(location_bar,
+      "http://search.example/#q=foobar");
+  OmniboxEditModel* model = location_bar->GetLocationEntry()->model();
+  EXPECT_TRUE(model->CurrentMatch().destination_url.is_valid());
+  EXPECT_EQ(ASCIIToUTF16("foobar"), model->CurrentMatch().contents);
+
+  // Verify that searching from the omnibox does search term replacement with
+  // second URL pattern.
+  chrome::FocusLocationBar(browser());
+  ui_test_utils::SendToOmniboxAndSubmit(location_bar,
+      "http://search.example/search#q=banana");
+  model = location_bar->GetLocationEntry()->model();
+  EXPECT_TRUE(model->CurrentMatch().destination_url.is_valid());
+  EXPECT_EQ(ASCIIToUTF16("banana"), model->CurrentMatch().contents);
+
+  // Verify that searching from the omnibox does search term replacement with
+  // standard search URL pattern.
+  chrome::FocusLocationBar(browser());
+  ui_test_utils::SendToOmniboxAndSubmit(location_bar,
+      "http://search.example/search?q=tractor+parts");
+  model = location_bar->GetLocationEntry()->model();
+  EXPECT_TRUE(model->CurrentMatch().destination_url.is_valid());
+  EXPECT_EQ(ASCIIToUTF16("tractor parts"), model->CurrentMatch().contents);
+
+  // Verify that searching from the omnibox prioritizes hash over query.
+  chrome::FocusLocationBar(browser());
+  ui_test_utils::SendToOmniboxAndSubmit(location_bar,
+      "http://search.example/search?q=tractor+parts#q=foobar");
+  model = location_bar->GetLocationEntry()->model();
+  EXPECT_TRUE(model->CurrentMatch().destination_url.is_valid());
+  EXPECT_EQ(ASCIIToUTF16("foobar"), model->CurrentMatch().contents);
+}
+
 // The linux and win  bots can't create a GL context. http://crbug.com/103379
 #if defined(OS_MACOSX)
 IN_PROC_BROWSER_TEST_F(PolicyTest, Disable3DAPIs) {
   ui_test_utils::NavigateToURL(browser(), GURL(chrome::kAboutBlankURL));
   // WebGL is enabled by default.
   content::WebContents* contents = chrome::GetActiveWebContents(browser());
   EXPECT_TRUE(IsWebGLEnabled(contents));
   // Disable with a policy.
   PolicyMap policies;
   policies.Set(key::kDisable3DAPIs, POLICY_LEVEL_MANDATORY,
@@ skipping 851 matching lines @@
 INSTANTIATE_TEST_CASE_P(
     RestoreOnStartupPolicyTestInstance,
     RestoreOnStartupPolicyTest,
     testing::Values(&RestoreOnStartupPolicyTest::HomepageIsNotNTP,
                     &RestoreOnStartupPolicyTest::HomepageIsNTP,
                     &RestoreOnStartupPolicyTest::ListOfURLs,
                     &RestoreOnStartupPolicyTest::NTP,
                     &RestoreOnStartupPolicyTest::Last));
 
 }  // namespace policy