src/chrome/browser/search/instant_service_unittest.cc

   1 // Copyright 2013 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 <string>
   6 #include <vector>
   7
   8 #include "base/memory/scoped_ptr.h"
   9 #include "base/metrics/field_trial.h"
  10 #include "base/strings/string_util.h"
  11 #include "chrome/browser/search/instant_service.h"
  12 #include "chrome/browser/search/instant_service_observer.h"
  13 #include "chrome/browser/search/instant_unittest_base.h"
  14 #include "chrome/browser/search/search.h"
  15 #include "chrome/browser/ui/browser.h"
  16 #include "chrome/browser/ui/browser_window.h"
  17 #include "chrome/browser/ui/host_desktop.h"
  18 #include "chrome/browser/ui/search/instant_search_prerenderer.h"
  19 #include "chrome/common/render_messages.h"
  20 #include "components/variations/entropy_provider.h"
  21 #include "content/public/browser/notification_service.h"
  22 #include "content/public/browser/notification_types.h"
  23 #include "content/public/test/mock_render_process_host.h"
  24 #include "ipc/ipc_message.h"
  25 #include "ipc/ipc_test_sink.h"
  26 #include "testing/gmock/include/gmock/gmock.h"
  27 #include "url/gurl.h"
  28
  29 class MockInstantServiceObserver : public InstantServiceObserver {
  30  public:
  31   MOCK_METHOD0(DefaultSearchProviderChanged, void());
  32   MOCK_METHOD0(GoogleURLUpdated, void());
  33   MOCK_METHOD1(OmniboxStartMarginChanged, void(int));
  34 };
  35
  36 class InstantServiceTest : public InstantUnitTestBase {
  37  protected:
  38   virtual void SetUp() OVERRIDE {
  39     InstantUnitTestBase::SetUp();
  40
  41     instant_service_observer_.reset(new MockInstantServiceObserver());
  42     instant_service_->AddObserver(instant_service_observer_.get());
  43   }
  44
  45   virtual void TearDown() OVERRIDE {
  46     instant_service_->RemoveObserver(instant_service_observer_.get());
  47     InstantUnitTestBase::TearDown();
  48   }
  49
  50   InstantSearchPrerenderer* GetInstantSearchPrerenderer() {
  51     return instant_service_->instant_search_prerenderer();
  52   }
  53
  54   void UpdateOmniboxStartMargin(int start_margin) {
  55     instant_service_->OnOmniboxStartMarginChanged(start_margin);
  56   }
  57
  58   scoped_ptr<MockInstantServiceObserver> instant_service_observer_;
  59 };
  60
  61 TEST_F(InstantServiceTest, DispatchDefaultSearchProviderChanged) {
  62   EXPECT_CALL(*instant_service_observer_.get(), DefaultSearchProviderChanged())
  63       .Times(1);
  64
  65   const std::string& new_base_url = "https://bar.com/";
  66   SetDefaultSearchProvider(new_base_url);
  67 }
  68
  69 TEST_F(InstantServiceTest, DispatchGoogleURLUpdated) {
  70   EXPECT_CALL(*instant_service_observer_.get(), GoogleURLUpdated()).Times(1);
  71
  72   const std::string& new_base_url = "https://www.google.es/";
  73   NotifyGoogleBaseURLUpdate(new_base_url);
  74 }
  75
  76 TEST_F(InstantServiceTest, SendsSearchURLsToRenderer) {
  77   ASSERT_TRUE(base::FieldTrialList::CreateFieldTrial("EmbeddedSearch",
  78       "Group1 use_cacheable_ntp:1"));
  79   scoped_ptr<content::MockRenderProcessHost> rph(
  80       new content::MockRenderProcessHost(profile()));
  81   rph->sink().ClearMessages();
  82   instant_service_->Observe(
  83       content::NOTIFICATION_RENDERER_PROCESS_CREATED,
  84       content::Source<content::MockRenderProcessHost>(rph.get()),
  85       content::NotificationService::NoDetails());
  86   EXPECT_EQ(1U, rph->sink().message_count());
  87   const IPC::Message* msg = rph->sink().GetMessageAt(0);
  88   ASSERT_TRUE(msg);
  89   std::vector<GURL> search_urls;
  90   GURL new_tab_page_url;
  91   ChromeViewMsg_SetSearchURLs::Read(msg, &search_urls, &new_tab_page_url);
  92   EXPECT_EQ(2U, search_urls.size());
  93   EXPECT_EQ("https://www.google.com/alt#quux=", search_urls[0].spec());
  94   EXPECT_EQ("https://www.google.com/url?bar=", search_urls[1].spec());
  95   EXPECT_EQ("https://www.google.com/newtab", new_tab_page_url.spec());
  96 }
  97
  98 TEST_F(InstantServiceTest, InstantSearchDisabled) {
  99   // 'prefetch_results' flag is not enabled in field trials. Make sure
 100   // InstantSearchPrerenderer is not initialized.
 101   EXPECT_EQ(static_cast<InstantSearchPrerenderer*>(NULL),
 102             GetInstantSearchPrerenderer());
 103 }
 104
 105 TEST_F(InstantServiceTest,
 106        ResetInstantSearchPrerenderer_DefaultProviderChanged) {
 107   ASSERT_TRUE(base::FieldTrialList::CreateFieldTrial(
 108       "EmbeddedSearch", "Group1 use_cacheable_ntp:1 prefetch_results:1"));
 109   EXPECT_CALL(*instant_service_observer_.get(), DefaultSearchProviderChanged())
 110       .Times(2);
 111
 112   // Set a default search provider that doesn't support Instant.
 113   TemplateURLData data;
 114   data.SetURL("https://foobar.com/url?bar={searchTerms}");
 115   TemplateURL* template_url = new TemplateURL(profile(), data);
 116   // Takes ownership of |template_url|.
 117   template_url_service_->Add(template_url);
 118   template_url_service_->SetDefaultSearchProvider(template_url);
 119
 120   EXPECT_EQ(static_cast<InstantSearchPrerenderer*>(NULL),
 121             GetInstantSearchPrerenderer());
 122
 123   // Set a default search provider that supports Instant and make sure
 124   // InstantSearchPrerenderer is valid.
 125   SetDefaultSearchProvider("https://google.com/");
 126   EXPECT_NE(static_cast<InstantSearchPrerenderer*>(NULL),
 127             GetInstantSearchPrerenderer());
 128 }
 129
 130 TEST_F(InstantServiceTest, ResetInstantSearchPrerenderer_GoogleBaseURLUpdated) {
 131   ASSERT_TRUE(base::FieldTrialList::CreateFieldTrial(
 132       "EmbeddedSearch", "Group1 use_cacheable_ntp:1 prefetch_results:1"));
 133   EXPECT_CALL(*instant_service_observer_.get(), DefaultSearchProviderChanged())
 134       .Times(1);
 135   EXPECT_CALL(*instant_service_observer_.get(), GoogleURLUpdated()).Times(1);
 136
 137   SetDefaultSearchProvider("https://google.com/");
 138   InstantSearchPrerenderer* old_prerenderer = GetInstantSearchPrerenderer();
 139   EXPECT_NE(static_cast<InstantSearchPrerenderer*>(NULL), old_prerenderer);
 140
 141   const std::string& new_base_url = "https://www.google.es/";
 142   NotifyGoogleBaseURLUpdate(new_base_url);
 143   EXPECT_NE(old_prerenderer, GetInstantSearchPrerenderer());
 144 }
 145
 146 TEST_F(InstantServiceTest, OmniboxStartMarginChanged) {
 147   int new_start_margin = 92;
 148   EXPECT_CALL(*instant_service_observer_.get(),
 149               OmniboxStartMarginChanged(new_start_margin)).Times(1);
 150   UpdateOmniboxStartMargin(new_start_margin);
 151 }