Consolidate search terms extraction and Instant process determination.

chrome/browser/ui/toolbar/toolbar_model_unittest.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 "chrome/browser/ui/toolbar/toolbar_model.h"
 
 #include "base/command_line.h"
 #include "base/utf_string_conversions.h"
+#include "chrome/browser/instant/instant_service.h"
+#include "chrome/browser/instant/instant_service_factory.h"
 #include "chrome/browser/search_engines/template_url.h"
 #include "chrome/browser/search_engines/template_url_service.h"
 #include "chrome/browser/search_engines/template_url_service_factory.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/search/search.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/browser/ui/toolbar/toolbar_model.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/test/base/browser_with_test_window_test.h"
+#include "content/public/browser/render_process_host.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/common/url_constants.h"
 
 using content::OpenURLParams;
 using content::Referrer;
 using content::WebContents;
 
 namespace {
 
 struct TestItem {
@@ skipping 91 matching lines @@
     TemplateURLServiceFactory::GetInstance()->SetTestingFactoryAndUse(
         profile(), &TemplateURLServiceFactory::BuildInstanceFor);
   }
   virtual void TearDown() OVERRIDE { BrowserWithTestWindowTest::TearDown(); }
 
  protected:
 
   void ResetDefaultTemplateURL() {
     TemplateURLData data;
     data.SetURL("http://google.com/search?q={searchTerms}");
+    data.instant_url = "http://does/not/exist";
     data.search_terms_replacement_key = "{google:instantExtendedEnabledKey}";
     TemplateURL* search_template_url = new TemplateURL(profile(), data);
     TemplateURLService* template_url_service =
         TemplateURLServiceFactory::GetForProfile(profile());
     template_url_service->Add(search_template_url);
     template_url_service->SetDefaultSearchProvider(search_template_url);
     ASSERT_NE(0, search_template_url->id());
     template_url_service->Load();
   }
 
@@ skipping 12 matching lines @@
   void NavigateAndCheckTextImpl(const GURL& url,
                                 bool can_replace,
                                 const string16 expected_text,
                                 bool would_replace,
                                 bool should_display) {
     WebContents* contents = browser()->tab_strip_model()->GetWebContentsAt(0);
     browser()->OpenURL(OpenURLParams(
         url, Referrer(), CURRENT_TAB, content::PAGE_TRANSITION_TYPED,
         false));
 
+    // Query terms replacement requires that the renderer process be a
+    // recognized Instant renderer. Fake it.
+    InstantService* instant_service =
+        InstantServiceFactory::GetForProfile(profile());
+    int process_id = contents->GetRenderProcessHost()->GetID();
+    instant_service->AddInstantProcess(process_id);
+
     ToolbarModel* toolbar_model = browser()->toolbar_model();
 
     // Check while loading.
     EXPECT_EQ(should_display, toolbar_model->ShouldDisplayURL());
     EXPECT_EQ(expected_text, toolbar_model->GetText(can_replace));
     EXPECT_EQ(would_replace,
               toolbar_model->WouldReplaceSearchURLWithSearchTerms());
 
     // Check after commit.
     CommitPendingLoad(&contents->GetController());
@@ skipping 29 matching lines @@
   AddTab(browser(), GURL(chrome::kAboutBlankURL));
   for (size_t i = 0; i < arraysize(test_items); ++i) {
     const TestItem& test_item = test_items[i];
     NavigateAndCheckText(test_item.url,
                          test_item.expected_text,
                          test_item.expected_replace_text_active,
                          test_item.would_replace,
                          test_item.should_display);
   }
 }