Moved WebDataService to ProfileKeyedService

chrome/browser/ui/search_engines/keyword_editor_controller_unittest.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "base/message_loop.h"
 #include "base/string16.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/search_engines/template_url.h"
 #include "chrome/browser/search_engines/template_url_service.h"
 #include "chrome/browser/search_engines/template_url_service_factory.h"
 #include "chrome/browser/ui/search_engines/keyword_editor_controller.h"
 #include "chrome/browser/ui/search_engines/template_url_table_model.h"
+#include "chrome/browser/webdata/web_data_service_factory.h"
 #include "chrome/common/chrome_notification_types.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/test/base/testing_pref_service.h"
 #include "chrome/test/base/testing_profile.h"
 #include "content/public/browser/notification_service.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "ui/base/models/table_model_observer.h"
 
 static const string16 kA(ASCIIToUTF16("a"));
 static const string16 kA1(ASCIIToUTF16("a1"));
 static const string16 kB(ASCIIToUTF16("b"));
 static const string16 kB1(ASCIIToUTF16("b1"));
 
 // Base class for keyword editor tests. Creates a profile containing an
 // empty TemplateURLService.
 class KeywordEditorControllerTest : public testing::Test,
                                     public ui::TableModelObserver {
  public:
   // Initializes all of the state.
   void Init(bool simulate_load_failure);
 
-  virtual void SetUp() {
-    Init(false);
-  }
-
-  virtual void OnModelChanged() {
-    model_changed_count_++;
-  }
-
-  virtual void OnItemsChanged(int start, int length) {
-    items_changed_count_++;
-  }
-
-  virtual void OnItemsAdded(int start, int length) {
-    added_count_++;
-  }
-
-  virtual void OnItemsRemoved(int start, int length) {
-    removed_count_++;
-  }
+  virtual void SetUp() OVERRIDE;
+  virtual void OnModelChanged() OVERRIDE;
+  virtual void OnItemsChanged(int start, int length) OVERRIDE;
+  virtual void OnItemsAdded(int start, int length) OVERRIDE;
+  virtual void OnItemsRemoved(int start, int length) OVERRIDE;
 
   void VerifyChangeCount(int model_changed_count, int item_changed_count,
-                         int added_count, int removed_count) {
-    ASSERT_EQ(model_changed_count, model_changed_count_);
-    ASSERT_EQ(item_changed_count, items_changed_count_);
-    ASSERT_EQ(added_count, added_count_);
-    ASSERT_EQ(removed_count, removed_count_);
-    ClearChangeCount();
-  }
-
-  void ClearChangeCount() {
-    model_changed_count_ = items_changed_count_ = added_count_ =
-        removed_count_ = 0;
-  }
-
-  void SimulateDefaultSearchIsManaged(const std::string& url) {
-    ASSERT_FALSE(url.empty());
-    TestingPrefService* service = profile_->GetTestingPrefService();
-    service->SetManagedPref(prefs::kDefaultSearchProviderEnabled,
-                            Value::CreateBooleanValue(true));
-    service->SetManagedPref(prefs::kDefaultSearchProviderSearchURL,
-                            Value::CreateStringValue(url));
-    service->SetManagedPref(prefs::kDefaultSearchProviderName,
-                            Value::CreateStringValue("managed"));
-    // Clear the IDs that are not specified via policy.
-    service->SetManagedPref(prefs::kDefaultSearchProviderID,
-                            new StringValue(std::string()));
-    service->SetManagedPref(prefs::kDefaultSearchProviderPrepopulateID,
-                            new StringValue(std::string()));
-    model_->Observe(chrome::NOTIFICATION_DEFAULT_SEARCH_POLICY_CHANGED,
-                    content::NotificationService::AllSources(),
-                    content::NotificationService::NoDetails());
-  }
-
-  TemplateURLTableModel* table_model() const {
-    return controller_->table_model();
-  }
+                         int added_count, int removed_count);
+  void ClearChangeCount();
+  void SimulateDefaultSearchIsManaged(const std::string& url);
+  TemplateURLTableModel* table_model() const;
 
  protected:
   MessageLoopForUI message_loop_;
   scoped_ptr<TestingProfile> profile_;
   scoped_ptr<KeywordEditorController> controller_;
   TemplateURLService* model_;
 
   int model_changed_count_;
   int items_changed_count_;
   int added_count_;
   int removed_count_;
 };
 
 void KeywordEditorControllerTest::Init(bool simulate_load_failure) {
   ClearChangeCount();
 
   // If init is called twice, make sure that the controller is destroyed before
   // the profile is.
   controller_.reset();
   profile_.reset(new TestingProfile());
   profile_->CreateTemplateURLService();
 
+  WebDataServiceFactory::GetInstance()->SetTestingFactory(profile_.get(), NULL);
+
   model_ = TemplateURLServiceFactory::GetForProfile(profile_.get());
   if (simulate_load_failure)
     model_->OnWebDataServiceRequestDone(0, NULL);
 
   controller_.reset(new KeywordEditorController(profile_.get()));
   controller_->table_model()->SetObserver(this);
 }
 
+void KeywordEditorControllerTest::SetUp() {
+  Init(false);
+}
+
+void KeywordEditorControllerTest::OnModelChanged() {
+  model_changed_count_++;
+}
+
+void KeywordEditorControllerTest::OnItemsChanged(int start, int length) {
+  items_changed_count_++;
+}
+
+void KeywordEditorControllerTest::OnItemsAdded(int start, int length) {
+  added_count_++;
+}
+
+void KeywordEditorControllerTest::OnItemsRemoved(int start, int length) {
+  removed_count_++;
+}
+
+void KeywordEditorControllerTest::VerifyChangeCount(int model_changed_count,
+                                                    int item_changed_count,
+                                                    int added_count,
+                                                    int removed_count) {
+  ASSERT_EQ(model_changed_count, model_changed_count_);
+  ASSERT_EQ(item_changed_count, items_changed_count_);
+  ASSERT_EQ(added_count, added_count_);
+  ASSERT_EQ(removed_count, removed_count_);
+  ClearChangeCount();
+}
+
+void KeywordEditorControllerTest::ClearChangeCount() {
+  model_changed_count_ = items_changed_count_ = added_count_ =
+      removed_count_ = 0;
+}
+
+void KeywordEditorControllerTest::SimulateDefaultSearchIsManaged(
+    const std::string& url) {
+  ASSERT_FALSE(url.empty());
+  TestingPrefService* service = profile_->GetTestingPrefService();
+  service->SetManagedPref(prefs::kDefaultSearchProviderEnabled,
+                          Value::CreateBooleanValue(true));
+  service->SetManagedPref(prefs::kDefaultSearchProviderSearchURL,
+                          Value::CreateStringValue(url));
+  service->SetManagedPref(prefs::kDefaultSearchProviderName,
+                          Value::CreateStringValue("managed"));
+  // Clear the IDs that are not specified via policy.
+  service->SetManagedPref(prefs::kDefaultSearchProviderID,
+                          new StringValue(std::string()));
+  service->SetManagedPref(prefs::kDefaultSearchProviderPrepopulateID,
+                          new StringValue(std::string()));
+  model_->Observe(chrome::NOTIFICATION_DEFAULT_SEARCH_POLICY_CHANGED,
+                  content::NotificationService::AllSources(),
+                  content::NotificationService::NoDetails());
+}
+
+TemplateURLTableModel* KeywordEditorControllerTest::table_model() const {
+  return controller_->table_model();
+}
+
 // Tests adding a TemplateURL.
 TEST_F(KeywordEditorControllerTest, Add) {
   controller_->AddTemplateURL(kA, kB, "http://c");
 
   // Verify the observer was notified.
   VerifyChangeCount(0, 0, 1, 0);
   if (HasFatalFailure())
     return;
 
   // Verify the TableModel has the new data.
@@ skipping 115 matching lines @@
   TemplateURL* turl = new TemplateURL(data);
   model_->Add(turl);
 
   // Table model should have updated.
   VerifyChangeCount(1, 0, 0, 0);
 
   // And should contain the newly added TemplateURL.
   ASSERT_EQ(1, table_model()->RowCount());
   ASSERT_EQ(0, table_model()->IndexOfTemplateURL(turl));
 }