instant_unittest_base.cc revision a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7 
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 "chrome/browser/search/instant_unittest_base.h"
6#include <string>
7
8#include "base/basictypes.h"
9#include "chrome/browser/chrome_notification_types.h"
10#include "chrome/browser/google/google_url_tracker.h"
11#include "chrome/browser/profiles/profile.h"
12#include "chrome/browser/search/instant_service.h"
13#include "chrome/browser/search/instant_service_factory.h"
14#include "chrome/browser/search/search.h"
15#include "chrome/browser/search_engines/search_terms_data.h"
16#include "chrome/browser/search_engines/template_url.h"
17#include "chrome/browser/search_engines/template_url_service.h"
18#include "chrome/browser/search_engines/template_url_service_factory.h"
19#include "chrome/common/pref_names.h"
20#include "chrome/test/base/browser_with_test_window_test.h"
21#include "chrome/test/base/testing_pref_service_syncable.h"
22#include "chrome/test/base/ui_test_utils.h"
23#include "components/variations/entropy_provider.h"
24#include "content/public/browser/notification_details.h"
25#include "content/public/browser/notification_service.h"
26#include "content/public/browser/notification_source.h"
27
28InstantUnitTestBase::InstantUnitTestBase() {
29  field_trial_list_.reset(new base::FieldTrialList(
30      new metrics::SHA1EntropyProvider("42")));
31}
32
33InstantUnitTestBase::~InstantUnitTestBase() {
34}
35
36void InstantUnitTestBase::SetUp() {
37  SetUpHelper();
38}
39
40void InstantUnitTestBase::SetUpWithoutCacheableNTP() {
41  ASSERT_TRUE(base::FieldTrialList::CreateFieldTrial(
42      "InstantExtended", "Group1 use_cacheable_ntp:0"));
43  SetUpHelper();
44}
45
46void InstantUnitTestBase::SetUpHelper() {
47  chrome::EnableQueryExtractionForTesting();
48  BrowserWithTestWindowTest::SetUp();
49
50  TemplateURLServiceFactory::GetInstance()->SetTestingFactoryAndUse(
51      profile(), &TemplateURLServiceFactory::BuildInstanceFor);
52  template_url_service_ = TemplateURLServiceFactory::GetForProfile(profile());
53  ui_test_utils::WaitForTemplateURLServiceToLoad(template_url_service_);
54
55  UIThreadSearchTermsData::SetGoogleBaseURL("https://www.google.com/");
56  TestingPrefServiceSyncable* pref_service = profile()->GetTestingPrefService();
57  pref_service->SetUserPref(prefs::kLastPromptedGoogleURL,
58                            new base::StringValue("https://www.google.com/"));
59  SetDefaultSearchProvider("{google:baseURL}");
60  instant_service_ = InstantServiceFactory::GetForProfile(profile());
61}
62
63void InstantUnitTestBase::TearDown() {
64  UIThreadSearchTermsData::SetGoogleBaseURL("");
65  BrowserWithTestWindowTest::TearDown();
66}
67
68void InstantUnitTestBase::SetDefaultSearchProvider(
69    const std::string& base_url) {
70  TemplateURLData data;
71  data.SetURL(base_url + "url?bar={searchTerms}");
72  data.instant_url = base_url +
73      "instant?{google:omniboxStartMarginParameter}{google:forceInstantResults}"
74      "foo=foo#foo=foo&strk";
75  data.new_tab_url = base_url + "newtab";
76  data.alternate_urls.push_back(base_url + "alt#quux={searchTerms}");
77  data.search_terms_replacement_key = "strk";
78
79  TemplateURL* template_url = new TemplateURL(profile(), data);
80  // Takes ownership of |template_url|.
81  template_url_service_->Add(template_url);
82  template_url_service_->SetDefaultSearchProvider(template_url);
83}
84
85void InstantUnitTestBase::NotifyGoogleBaseURLUpdate(
86    const std::string& new_google_base_url) {
87  // GoogleURLTracker is not created in tests.
88  // (See GoogleURLTrackerFactory::ServiceIsNULLWhileTesting())
89  // For determining google:baseURL for NTP, the following is used:
90  // UIThreadSearchTermsData::GoogleBaseURLValue()
91  // For simulating test behavior, this is overridden below.
92  UIThreadSearchTermsData::SetGoogleBaseURL(new_google_base_url);
93  GoogleURLTracker::UpdatedDetails details(GURL("https://www.google.com/"),
94                                           GURL(new_google_base_url));
95  content::NotificationService::current()->Notify(
96      chrome::NOTIFICATION_GOOGLE_URL_UPDATED,
97      content::Source<Profile>(profile()->GetOriginalProfile()),
98      content::Details<GoogleURLTracker::UpdatedDetails>(&details));
99}
100
101
102bool InstantUnitTestBase::IsInstantServiceObserver(
103    InstantServiceObserver* observer) {
104  return instant_service_->observers_.HasObserver(observer);
105}
106
107