Prepare AutocompleteController for componentization

components/omnibox/base_search_provider_unittest.cc

 // Copyright 2014 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 "components/omnibox/base_search_provider.h"
 
 #include "base/strings/string16.h"
 #include "base/strings/utf_string_conversions.h"
 #include "components/omnibox/autocomplete_match.h"
 #include "components/omnibox/autocomplete_match_type.h"
-#include "components/omnibox/autocomplete_provider_client.h"
 #include "components/omnibox/autocomplete_scheme_classifier.h"
+#include "components/omnibox/mock_autocomplete_provider_client.h"
 #include "components/omnibox/search_suggestion_parser.h"
 #include "components/omnibox/suggestion_answer.h"
 #include "components/search_engines/search_terms_data.h"
 #include "components/search_engines/template_url_service.h"
 #include "components/search_engines/template_url_service_client.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using testing::NiceMock;
 using testing::Return;
 using testing::_;
 
-class MockAutocompleteProviderClient : public AutocompleteProviderClient {
- public:
-  MockAutocompleteProviderClient() {
-    template_url_service_.reset(new TemplateURLService(
-        nullptr, scoped_ptr<SearchTermsData>(new SearchTermsData), nullptr,
-        scoped_ptr<TemplateURLServiceClient>(), nullptr, nullptr,
-        base::Closure()));
-  }
-  MOCK_METHOD0(GetRequestContext, net::URLRequestContextGetter*());
-  MOCK_METHOD0(GetPrefs, PrefService*());
-  MOCK_CONST_METHOD0(GetSchemeClassifier,
-                     const AutocompleteSchemeClassifier&());
-  MOCK_METHOD0(GetAutocompleteClassifier, AutocompleteClassifier*());
-  MOCK_METHOD0(GetHistoryService, history::HistoryService*());
-
-  // Can't mock scoped_refptr :\.
-  scoped_refptr<history::TopSites> GetTopSites() override { return nullptr; }
-
-  MOCK_METHOD0(GetBookmarkModel, bookmarks::BookmarkModel*());
-  MOCK_METHOD0(GetInMemoryDatabase, history::URLDatabase*());
-
-  TemplateURLService* GetTemplateURLService() override {
-    return template_url_service_.get();
-  }
-  const TemplateURLService* GetTemplateURLService() const override {
-    return template_url_service_.get();
-  }
-
-  MOCK_CONST_METHOD0(GetSearchTermsData, const SearchTermsData&());
-
-  // Can't mock scoped_refptr :\.
-  scoped_refptr<ShortcutsBackend> GetShortcutsBackend() override {
-    return nullptr;
-  }
-  scoped_refptr<ShortcutsBackend> GetShortcutsBackendIfExists() override {
-    return nullptr;
-  }
-
-  MOCK_CONST_METHOD0(GetAcceptLanguages, std::string());
-  MOCK_CONST_METHOD0(IsOffTheRecord, bool());
-  MOCK_CONST_METHOD0(SearchSuggestEnabled, bool());
-  MOCK_CONST_METHOD0(BookmarkBarIsVisible, bool());
-  MOCK_CONST_METHOD0(TabSyncEnabledAndUnencrypted, bool());
-  MOCK_METHOD6(
-      Classify,
-      void(const base::string16& text,
-           bool prefer_keyword,
-           bool allow_exact_keyword_match,
-           metrics::OmniboxEventProto::PageClassification page_classification,
-           AutocompleteMatch* match,
-           GURL* alternate_nav_url));
-  MOCK_METHOD2(DeleteMatchingURLsForKeywordFromHistory,
-               void(history::KeywordID keyword_id, const base::string16& term));
-  MOCK_METHOD1(PrefetchImage, void(const GURL& url));
-
- private:
-  scoped_ptr<TemplateURLService> template_url_service_;
-
-  DISALLOW_COPY_AND_ASSIGN(MockAutocompleteProviderClient);
-};
-
 class TestBaseSearchProvider : public BaseSearchProvider {
  public:
   typedef BaseSearchProvider::MatchMap MatchMap;
 
   TestBaseSearchProvider(AutocompleteProvider::Type type,
                          AutocompleteProviderClient* client)
       : BaseSearchProvider(type, client) {}
   MOCK_METHOD1(DeleteMatch, void(const AutocompleteMatch& match));
   MOCK_CONST_METHOD1(AddProviderInfo, void(ProvidersInfo* provider_info));
   MOCK_CONST_METHOD1(GetTemplateURL, const TemplateURL*(bool is_keyword));
@@ skipping 24 matching lines @@
  private:
   DISALLOW_COPY_AND_ASSIGN(TestBaseSearchProvider);
 };
 
 class BaseSearchProviderTest : public testing::Test {
  public:
   ~BaseSearchProviderTest() override {}
 
  protected:
   void SetUp() override {
+    scoped_ptr<TemplateURLService> template_url_service(new TemplateURLService(
+        nullptr, scoped_ptr<SearchTermsData>(new SearchTermsData), nullptr,
+        scoped_ptr<TemplateURLServiceClient>(), nullptr, nullptr,
+        base::Closure()));
     client_.reset(new NiceMock<MockAutocompleteProviderClient>());
+    client_->set_template_url_service(template_url_service.Pass());
     provider_ = new NiceMock<TestBaseSearchProvider>(
         AutocompleteProvider::TYPE_SEARCH, client_.get());
   }
 
   scoped_refptr<NiceMock<TestBaseSearchProvider> > provider_;
   scoped_ptr<NiceMock<MockAutocompleteProviderClient>> client_;
 };
 
 TEST_F(BaseSearchProviderTest, PreserveAnswersWhenDeduplicating) {
   TemplateURLData data;
@@ skipping 73 matching lines @@
   EXPECT_TRUE(answer2->Equals(*match.answer));
   EXPECT_EQ(AutocompleteMatchType::SEARCH_HISTORY, match.type);
   EXPECT_EQ(1300, match.relevance);
 
   EXPECT_EQ(answer_contents, duplicate.answer_contents);
   EXPECT_EQ(answer_type, duplicate.answer_type);
   EXPECT_TRUE(answer->Equals(*duplicate.answer));
   EXPECT_EQ(AutocompleteMatchType::SEARCH_SUGGEST, duplicate.type);
   EXPECT_EQ(850, duplicate.relevance);
 }