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
29class MockInstantServiceObserver : public InstantServiceObserver {
30 public:
31  MOCK_METHOD0(DefaultSearchProviderChanged, void());
32  MOCK_METHOD1(OmniboxStartMarginChanged, void(int));
33};
34
35class InstantServiceTest : public InstantUnitTestBase {
36 protected:
37  virtual void SetUp() OVERRIDE {
38    InstantUnitTestBase::SetUp();
39
40    instant_service_observer_.reset(new MockInstantServiceObserver());
41    instant_service_->AddObserver(instant_service_observer_.get());
42  }
43
44  virtual void TearDown() OVERRIDE {
45    instant_service_->RemoveObserver(instant_service_observer_.get());
46    InstantUnitTestBase::TearDown();
47  }
48
49  InstantSearchPrerenderer* GetInstantSearchPrerenderer() {
50    return instant_service_->instant_search_prerenderer();
51  }
52
53  void UpdateOmniboxStartMargin(int start_margin) {
54    instant_service_->OnOmniboxStartMarginChanged(start_margin);
55  }
56
57  scoped_ptr<MockInstantServiceObserver> instant_service_observer_;
58};
59
60class InstantServiceEnabledTest : public InstantServiceTest {
61 protected:
62  virtual void SetUp() OVERRIDE {
63    ASSERT_TRUE(base::FieldTrialList::CreateFieldTrial(
64        "EmbeddedSearch", "Group1 use_cacheable_ntp:1"));
65    InstantServiceTest::SetUp();
66  }
67};
68
69TEST_F(InstantServiceEnabledTest, DispatchDefaultSearchProviderChanged) {
70  EXPECT_CALL(*instant_service_observer_.get(), DefaultSearchProviderChanged())
71      .Times(1);
72
73  const std::string new_base_url = "https://bar.com/";
74  SetUserSelectedDefaultSearchProvider(new_base_url);
75}
76
77TEST_F(InstantServiceTest, DontDispatchGoogleURLUpdatedForNonGoogleURLs) {
78  EXPECT_CALL(*instant_service_observer_.get(), DefaultSearchProviderChanged())
79      .Times(1);
80  const std::string new_dsp_url = "https://bar.com/";
81  SetUserSelectedDefaultSearchProvider(new_dsp_url);
82  testing::Mock::VerifyAndClearExpectations(instant_service_observer_.get());
83
84  EXPECT_CALL(*instant_service_observer_.get(), DefaultSearchProviderChanged())
85      .Times(0);
86  const std::string new_base_url = "https://www.google.es/";
87  NotifyGoogleBaseURLUpdate(new_base_url);
88}
89
90TEST_F(InstantServiceTest, DispatchGoogleURLUpdated) {
91  EXPECT_CALL(*instant_service_observer_.get(), DefaultSearchProviderChanged())
92      .Times(1);
93
94  const std::string new_base_url = "https://www.google.es/";
95  NotifyGoogleBaseURLUpdate(new_base_url);
96}
97
98TEST_F(InstantServiceEnabledTest, SendsSearchURLsToRenderer) {
99  scoped_ptr<content::MockRenderProcessHost> rph(
100      new content::MockRenderProcessHost(profile()));
101  rph->sink().ClearMessages();
102  instant_service_->Observe(
103      content::NOTIFICATION_RENDERER_PROCESS_CREATED,
104      content::Source<content::MockRenderProcessHost>(rph.get()),
105      content::NotificationService::NoDetails());
106  EXPECT_EQ(1U, rph->sink().message_count());
107  const IPC::Message* msg = rph->sink().GetMessageAt(0);
108  ASSERT_TRUE(msg);
109  ChromeViewMsg_SetSearchURLs::Param params;
110  ChromeViewMsg_SetSearchURLs::Read(msg, &params);
111  std::vector<GURL> search_urls = params.a;
112  GURL new_tab_page_url = params.b;
113  EXPECT_EQ(2U, search_urls.size());
114  EXPECT_EQ("https://www.google.com/alt#quux=", search_urls[0].spec());
115  EXPECT_EQ("https://www.google.com/url?bar=", search_urls[1].spec());
116  EXPECT_EQ("https://www.google.com/newtab", new_tab_page_url.spec());
117}
118
119TEST_F(InstantServiceTest, InstantSearchEnabled) {
120  EXPECT_NE(static_cast<InstantSearchPrerenderer*>(NULL),
121            GetInstantSearchPrerenderer());
122}
123
124TEST_F(InstantServiceEnabledTest,
125       ResetInstantSearchPrerenderer_DefaultProviderChanged) {
126  EXPECT_CALL(*instant_service_observer_.get(), DefaultSearchProviderChanged())
127      .Times(2);
128
129  // Set a default search provider that doesn't support Instant.
130  TemplateURLData data;
131  data.SetURL("https://foobar.com/url?bar={searchTerms}");
132  TemplateURL* template_url = new TemplateURL(data);
133  // Takes ownership of |template_url|.
134  template_url_service_->Add(template_url);
135  template_url_service_->SetUserSelectedDefaultSearchProvider(template_url);
136
137  EXPECT_EQ(static_cast<InstantSearchPrerenderer*>(NULL),
138            GetInstantSearchPrerenderer());
139
140  // Set a default search provider that supports Instant and make sure
141  // InstantSearchPrerenderer is valid.
142  SetUserSelectedDefaultSearchProvider("https://google.com/");
143  EXPECT_NE(static_cast<InstantSearchPrerenderer*>(NULL),
144            GetInstantSearchPrerenderer());
145}
146
147TEST_F(InstantServiceEnabledTest,
148       ResetInstantSearchPrerenderer_GoogleBaseURLUpdated) {
149  EXPECT_CALL(*instant_service_observer_.get(), DefaultSearchProviderChanged())
150      .Times(1);
151
152  InstantSearchPrerenderer* old_prerenderer = GetInstantSearchPrerenderer();
153  EXPECT_TRUE(old_prerenderer != NULL);
154
155  const std::string new_base_url = "https://www.google.es/";
156  NotifyGoogleBaseURLUpdate(new_base_url);
157  EXPECT_NE(old_prerenderer, GetInstantSearchPrerenderer());
158}
159
160TEST_F(InstantServiceTest, OmniboxStartMarginChanged) {
161  int new_start_margin = 92;
162  EXPECT_CALL(*instant_service_observer_.get(),
163              OmniboxStartMarginChanged(new_start_margin)).Times(1);
164  UpdateOmniboxStartMargin(new_start_margin);
165}
166