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#ifndef CHROME_BROWSER_SEARCH_INSTANT_UNITTEST_BASE_H_ 6#define CHROME_BROWSER_SEARCH_INSTANT_UNITTEST_BASE_H_ 7 8#include <string> 9 10#include "base/metrics/field_trial.h" 11#include "chrome/browser/search/instant_service.h" 12#include "chrome/test/base/browser_with_test_window_test.h" 13#include "components/search_engines/template_url_service.h" 14 15class InstantServiceObserver; 16 17// This class provides an extension on top of BrowserWithTestWindowTest, and 18// adds some utility methods which can be useful for various unit tests for 19// Embedded Search / Instant implementation classes. 20class InstantUnitTestBase : public BrowserWithTestWindowTest { 21 protected: 22 InstantUnitTestBase(); 23 virtual ~InstantUnitTestBase(); 24 25 virtual void SetUp() OVERRIDE; 26 virtual void TearDown() OVERRIDE; 27 28#if !defined(OS_IOS) && !defined(OS_ANDROID) 29 // Query extraction is always enabled on Android and iOS. 30 void SetUpWithoutQueryExtraction(); 31#endif 32 33 // Adds and sets the default search provider using the base_url. 34 // The base_url should have the http[s]:// prefix and a trailing / after the 35 // TLD. 36 // It will always use an instant-enabled configuration using a 37 // search_terms_replacement_key. 38 void SetUserSelectedDefaultSearchProvider(const std::string& base_url); 39 40 // Simulates a Google Base URL change as would happen in event of 41 // search-domain-check. Note that the GoogleURLTrackerFactory is disabled for 42 // tests, so this is required. 43 void NotifyGoogleBaseURLUpdate(const std::string& new_google_base_url); 44 45 bool IsInstantServiceObserver(InstantServiceObserver* observer); 46 47 InstantService* instant_service_; 48 TemplateURLService* template_url_service_; 49 scoped_ptr<base::FieldTrialList> field_trial_list_; 50 51 private: 52 // BrowserWithTestWindowTest override: 53 virtual TestingProfile* CreateProfile() OVERRIDE; 54 55 void SetUpHelper(); 56}; 57 58#endif // CHROME_BROWSER_SEARCH_INSTANT_UNITTEST_BASE_H_ 59