Reload Instant NTP and Instant-process tabs on search url change

chrome/browser/search/search_unittest.cc

 // Copyright (c) 2013 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/command_line.h"
 #include "base/metrics/field_trial.h"
 #include "base/metrics/histogram_base.h"
 #include "base/metrics/histogram_samples.h"
 #include "base/metrics/statistics_recorder.h"
 #include "base/prefs/pref_service.h"
 #include "chrome/browser/search/instant_service.h"
 #include "chrome/browser/search/instant_service_factory.h"
 #include "chrome/browser/search/search.h"
+#include "chrome/browser/search/search_test_utils.h"
 #include "chrome/browser/search_engines/search_terms_data.h"
 #include "chrome/browser/search_engines/template_url_service.h"
 #include "chrome/browser/search_engines/template_url_service_factory.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/metrics/entropy_provider.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/common/url_constants.h"
 #include "chrome/test/base/browser_with_test_window_test.h"
 #include "chrome/test/base/ui_test_utils.h"
@@ skipping 153 matching lines @@
  protected:
   virtual void SetUp() OVERRIDE {
     BrowserWithTestWindowTest::SetUp();
     field_trial_list_.reset(new base::FieldTrialList(
         new metrics::SHA1EntropyProvider("42")));
     TemplateURLServiceFactory::GetInstance()->SetTestingFactoryAndUse(
         profile(), &TemplateURLServiceFactory::BuildInstanceFor);
     TemplateURLService* template_url_service =
         TemplateURLServiceFactory::GetForProfile(profile());
     ui_test_utils::WaitForTemplateURLServiceToLoad(template_url_service);
-    SetSearchProvider();
-  }
-
-  void SetSearchProvider() {
-    TemplateURLService* template_url_service =
-        TemplateURLServiceFactory::GetForProfile(profile());
-    TemplateURLData data;
-    data.SetURL("http://foo.com/url?bar={searchTerms}");
-    data.instant_url = "http://foo.com/instant?"
-        "{google:omniboxStartMarginParameter}foo=foo#foo=foo&strk";
-    data.alternate_urls.push_back("http://foo.com/alt#quux={searchTerms}");
-    data.search_terms_replacement_key = "strk";
-
-    TemplateURL* template_url = new TemplateURL(profile(), data);
-    // Takes ownership of |template_url|.
-    template_url_service->Add(template_url);
-    template_url_service->SetDefaultSearchProvider(template_url);
-  }
-
-  // Build an Instant URL with or without a valid search terms replacement key
-  // as per |has_search_term_replacement_key|. Set that URL as the instant URL
-  // for the default search provider.
-  void SetDefaultInstantTemplateUrl(bool has_search_term_replacement_key) {
-    TemplateURLService* template_url_service =
-        TemplateURLServiceFactory::GetForProfile(profile());
-
-    static const char kInstantURLWithStrk[] =
-        "http://foo.com/instant?foo=foo#foo=foo&strk";
-    static const char kInstantURLNoStrk[] =
-        "http://foo.com/instant?foo=foo#foo=foo";
-
-    TemplateURLData data;
-    data.SetURL("http://foo.com/url?bar={searchTerms}");
-    data.instant_url = (has_search_term_replacement_key ?
-        kInstantURLWithStrk : kInstantURLNoStrk);
-    data.search_terms_replacement_key = "strk";
-
-    TemplateURL* template_url = new TemplateURL(profile(), data);
-    // Takes ownership of |template_url|.
-    template_url_service->Add(template_url);
-    template_url_service->SetDefaultSearchProvider(template_url);
+    search_test_utils::SetSearchProvider(profile());
   }
 
   bool InInstantProcess(const content::WebContents* contents) {
     InstantService* instant_service =
         InstantServiceFactory::GetForProfile(profile());
     return instant_service->IsInstantProcess(
         contents->GetRenderProcessHost()->GetID());
   }
 
   scoped_ptr<base::FieldTrialList> field_trial_list_;
@@ skipping 278 matching lines @@
         << test.url << " " << test.comment;
   }
 }
 
 TEST_F(SearchTest, GetInstantURLExtendedEnabled) {
   // Instant is disabled, so no Instant URL.
   EXPECT_EQ(GURL(), GetInstantURL(profile(), kDisableStartMargin));
 
   // Enable Instant. Still no Instant URL because "strk" is missing.
   EnableInstantExtendedAPIForTesting();
-  SetDefaultInstantTemplateUrl(false);
+  search_test_utils::SetDefaultInstantTemplateUrl(profile(), false);
   EXPECT_EQ(GURL(), GetInstantURL(profile(), kDisableStartMargin));
 
   // Set an Instant URL with a valid search terms replacement key.
-  SetDefaultInstantTemplateUrl(true);
+  search_test_utils::SetDefaultInstantTemplateUrl(profile(), true);
 
   // Now there should be a valid Instant URL. Note the HTTPS "upgrade".
   EXPECT_EQ(GURL("https://foo.com/instant?foo=foo#foo=foo&strk"),
             GetInstantURL(profile(), kDisableStartMargin));
 
   // Enable suggest. No difference.
   profile()->GetPrefs()->SetBoolean(prefs::kSearchSuggestEnabled, true);
   EXPECT_EQ(GURL("https://foo.com/instant?foo=foo#foo=foo&strk"),
             GetInstantURL(profile(), kDisableStartMargin));
 
@@ skipping 79 matching lines @@
 }
 
 TEST_F(SearchTest, ShouldShowInstantNTP_DisabledByInstantNewTabURLSwitch) {
   EnableInstantExtendedAPIForTesting();
   CommandLine::ForCurrentProcess()->AppendSwitchASCII(
       switches::kInstantNewTabURL, "http://example.com/newtab");
   EXPECT_FALSE(ShouldShowInstantNTP());
 }
 
 }  // namespace chrome