Move KeywordProvider to components/omnibox

components/omnibox/keyword_provider_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/command_line.h"
 #include "base/message_loop/message_loop.h"
 #include "base/strings/utf_string_conversions.h"
-#include "chrome/browser/autocomplete/chrome_autocomplete_scheme_classifier.h"
-#include "chrome/browser/autocomplete/keyword_provider.h"
 #include "components/metrics/proto/omnibox_event.pb.h"
 #include "components/omnibox/autocomplete_match.h"
+#include "components/omnibox/autocomplete_scheme_classifier.h"
+#include "components/omnibox/keyword_provider.h"
 #include "components/search_engines/search_engines_switches.h"
 #include "components/search_engines/template_url.h"
 #include "components/search_engines/template_url_service.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "url/gurl.h"
 
 using base::ASCIIToUTF16;
 
+namespace {
+
+class TestingSchemeClassifier : public AutocompleteSchemeClassifier {
+ public:
+  virtual metrics::OmniboxInputType::Type GetInputTypeForScheme(
+      const std::string& scheme) const OVERRIDE {
+    if (net::URLRequest::IsHandledProtocol(scheme))
+      return metrics::OmniboxInputType::URL;
+    return metrics::OmniboxInputType::INVALID;
+  }
+};
+
+}  // namespace
+
 class KeywordProviderTest : public testing::Test {
  protected:
   template<class ResultType>
   struct MatchType {
     const ResultType member;
     bool allowed_to_be_default_match;
   };
 
   template<class ResultType>
   struct TestData {
@@ skipping 36 matching lines @@
   model_.reset(new TemplateURLService(kTestData, arraysize(kTestData)));
   kw_provider_ = new KeywordProvider(NULL, model_.get());
 }
 
 void KeywordProviderTest::TearDown() {
   model_.reset();
   kw_provider_ = NULL;
 }
 
 template<class ResultType>
-void KeywordProviderTest::RunTest(
-    TestData<ResultType>* keyword_cases,
-    int num_cases,
-    ResultType AutocompleteMatch::* member) {
+void KeywordProviderTest::RunTest(TestData<ResultType>* keyword_cases,
+                                  int num_cases,
+                                  ResultType AutocompleteMatch::* member) {
   ACMatches matches;
   for (int i = 0; i < num_cases; ++i) {
     SCOPED_TRACE(keyword_cases[i].input);
     AutocompleteInput input(keyword_cases[i].input, base::string16::npos,
                             base::string16(), GURL(),
                             metrics::OmniboxEventProto::INVALID_SPEC, true,
-                            false, true, true,
-                            ChromeAutocompleteSchemeClassifier(NULL));
+                            false, true, true, TestingSchemeClassifier());
     kw_provider_->Start(input, false);
     EXPECT_TRUE(kw_provider_->done());
     matches = kw_provider_->matches();
     ASSERT_EQ(keyword_cases[i].num_results, matches.size());
     for (size_t j = 0; j < matches.size(); ++j) {
       EXPECT_EQ(keyword_cases[i].output[j].member, matches[j].*member);
       EXPECT_EQ(keyword_cases[i].output[j].allowed_to_be_default_match,
                 matches[j].allowed_to_be_default_match);
     }
   }
@@ skipping 221 matching lines @@
     { "aa  foo", 5u, true, "aa.com?foo={searchTerms}", "foo", 1u },
 
     // Disallow exact keyword match.
     { "aa foo", base::string16::npos, false, "", "aa foo",
       base::string16::npos },
   };
   for (size_t i = 0; i < ARRAYSIZE_UNSAFE(cases); i++) {
     AutocompleteInput input(
         ASCIIToUTF16(cases[i].text), cases[i].cursor_position, base::string16(),
         GURL(), metrics::OmniboxEventProto::INVALID_SPEC, false, false,
-        cases[i].allow_exact_keyword_match, true,
-        ChromeAutocompleteSchemeClassifier(NULL));
+        cases[i].allow_exact_keyword_match, true, TestingSchemeClassifier());
     const TemplateURL* url =
         KeywordProvider::GetSubstitutingTemplateURLForInput(model_.get(),
                                                             &input);
     if (cases[i].expected_url.empty())
       EXPECT_FALSE(url);
     else
       EXPECT_EQ(cases[i].expected_url, url->url());
     EXPECT_EQ(ASCIIToUTF16(cases[i].updated_text), input.text());
     EXPECT_EQ(cases[i].updated_cursor_position, input.cursor_position());
   }
 }
 
 // If extra query params are specified on the command line, they should be
 // reflected (only) in the default search provider's destination URL.
 TEST_F(KeywordProviderTest, ExtraQueryParams) {
   CommandLine::ForCurrentProcess()->AppendSwitchASCII(
       switches::kExtraSearchQueryParams, "a=b");
 
   TestData<GURL> url_cases[] = {
     { ASCIIToUTF16("a 1 2 3"), 3,
       { { GURL("aa.com?a=b&foo=1+2+3"), false },
         { GURL("bogus URL 1+2+3"), false },
         { GURL("http://aaaa/?aaaa=1&b=1+2+3&c"), false } } },
   };
 
   RunTest<GURL>(url_cases, arraysize(url_cases),
                 &AutocompleteMatch::destination_url);
 }