Home | History | Annotate | Download | only in search_engines 
      1 // Copyright (c) 2012 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_engines/template_url_service_test_util.h"
      6 
      7 #include "base/message_loop/message_loop_proxy.h"
      8 #include "base/run_loop.h"
      9 #include "chrome/browser/search_engines/chrome_template_url_service_client.h"
     10 #include "chrome/test/base/testing_pref_service_syncable.h"
     11 #include "chrome/test/base/testing_profile.h"
     12 #include "components/search_engines/default_search_pref_test_util.h"
     13 #include "components/search_engines/keyword_table.h"
     14 #include "components/search_engines/keyword_web_data_service.h"
     15 #include "components/search_engines/template_url_service.h"
     16 #include "components/search_engines/testing_search_terms_data.h"
     17 #include "components/webdata/common/web_database_service.h"
     18 #include "testing/gtest/include/gtest/gtest.h"
     19 
     20 namespace {
     21 
     22 class TestingTemplateURLServiceClient : public ChromeTemplateURLServiceClient {
     23  public:
     24   TestingTemplateURLServiceClient(Profile* profile,
     25                                   base::string16* search_term)
     26       : ChromeTemplateURLServiceClient(profile),
     27         search_term_(search_term) {}
     28 
     29   virtual void SetKeywordSearchTermsForURL(
     30       const GURL& url,
     31       TemplateURLID id,
     32       const base::string16& term) OVERRIDE {
     33     *search_term_ = term;
     34   }
     35 
     36  private:
     37   base::string16* search_term_;
     38 
     39   DISALLOW_COPY_AND_ASSIGN(TestingTemplateURLServiceClient);
     40 };
     41 
     42 }  // namespace
     43 
     44 TemplateURLServiceTestUtil::TemplateURLServiceTestUtil()
     45     : changed_count_(0),
     46       search_terms_data_(NULL) {
     47   // Make unique temp directory.
     48   EXPECT_TRUE(temp_dir_.CreateUniqueTempDir());
     49   profile_.reset(new TestingProfile(temp_dir_.path()));
     50 
     51   scoped_refptr<WebDatabaseService> web_database_service =
     52       new WebDatabaseService(temp_dir_.path().AppendASCII("webdata"),
     53                              base::MessageLoopProxy::current(),
     54                              base::MessageLoopProxy::current());
     55   web_database_service->AddTable(
     56       scoped_ptr<WebDatabaseTable>(new KeywordTable()));
     57   web_database_service->LoadDatabase();
     58 
     59   web_data_service_ =  new KeywordWebDataService(
     60       web_database_service.get(), base::MessageLoopProxy::current(),
     61       KeywordWebDataService::ProfileErrorCallback());
     62   web_data_service_->Init();
     63 
     64   ResetModel(false);
     65 }
     66 
     67 TemplateURLServiceTestUtil::~TemplateURLServiceTestUtil() {
     68   ClearModel();
     69   profile_.reset();
     70 
     71   // Flush the message loop to make application verifiers happy.
     72   base::RunLoop().RunUntilIdle();
     73 }
     74 
     75 void TemplateURLServiceTestUtil::OnTemplateURLServiceChanged() {
     76   changed_count_++;
     77 }
     78 
     79 int TemplateURLServiceTestUtil::GetObserverCount() {
     80   return changed_count_;
     81 }
     82 
     83 void TemplateURLServiceTestUtil::ResetObserverCount() {
     84   changed_count_ = 0;
     85 }
     86 
     87 void TemplateURLServiceTestUtil::VerifyLoad() {
     88   ASSERT_FALSE(model()->loaded());
     89   model()->Load();
     90   base::RunLoop().RunUntilIdle();
     91   EXPECT_EQ(1, GetObserverCount());
     92   ResetObserverCount();
     93 }
     94 
     95 void TemplateURLServiceTestUtil::ChangeModelToLoadState() {
     96   model()->ChangeToLoadedState();
     97   // Initialize the web data service so that the database gets updated with
     98   // any changes made.
     99 
    100   model()->web_data_service_ = web_data_service_;
    101   base::RunLoop().RunUntilIdle();
    102 }
    103 
    104 void TemplateURLServiceTestUtil::ClearModel() {
    105   model_->Shutdown();
    106   model_.reset();
    107   search_terms_data_ = NULL;
    108 }
    109 
    110 void TemplateURLServiceTestUtil::ResetModel(bool verify_load) {
    111   if (model_)
    112     ClearModel();
    113   search_terms_data_ = new TestingSearchTermsData("http://www.google.com/");
    114   model_.reset(new TemplateURLService(
    115       profile()->GetPrefs(), scoped_ptr<SearchTermsData>(search_terms_data_),
    116       web_data_service_.get(),
    117       scoped_ptr<TemplateURLServiceClient>(
    118           new TestingTemplateURLServiceClient(profile(), &search_term_)),
    119       NULL, NULL, base::Closure()));
    120   model()->AddObserver(this);
    121   changed_count_ = 0;
    122   if (verify_load)
    123     VerifyLoad();
    124 }
    125 
    126 base::string16 TemplateURLServiceTestUtil::GetAndClearSearchTerm() {
    127   base::string16 search_term;
    128   search_term.swap(search_term_);
    129   return search_term;
    130 }
    131 
    132 void TemplateURLServiceTestUtil::SetGoogleBaseURL(const GURL& base_url) {
    133   DCHECK(base_url.is_valid());
    134   search_terms_data_->set_google_base_url(base_url.spec());
    135   model_->GoogleBaseURLChanged();
    136 }
    137 
    138 void TemplateURLServiceTestUtil::SetManagedDefaultSearchPreferences(
    139     bool enabled,
    140     const std::string& name,
    141     const std::string& keyword,
    142     const std::string& search_url,
    143     const std::string& suggest_url,
    144     const std::string& icon_url,
    145     const std::string& encodings,
    146     const std::string& alternate_url,
    147     const std::string& search_terms_replacement_key) {
    148   DefaultSearchPrefTestUtil::SetManagedPref(
    149       profile()->GetTestingPrefService(),
    150       enabled, name, keyword, search_url, suggest_url, icon_url, encodings,
    151       alternate_url, search_terms_replacement_key);
    152 }
    153 
    154 void TemplateURLServiceTestUtil::RemoveManagedDefaultSearchPreferences() {
    155   DefaultSearchPrefTestUtil::RemoveManagedPref(
    156       profile()->GetTestingPrefService());
    157 }
    158