Home | History | Annotate | Download | only in search 
      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/browser/search_engines/template_url_service.h"
     13 #include "chrome/test/base/browser_with_test_window_test.h"
     14 
     15 class 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.
     20 class 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