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