Handle Search suggest subtypes for Shortcuts DB

chrome/browser/history/shortcuts_backend_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/files/scoped_temp_dir.h"
 #include "base/message_loop/message_loop.h"
 #include "base/path_service.h"
 #include "base/strings/stringprintf.h"
 #include "base/strings/utf_string_conversions.h"
 #include "chrome/browser/history/shortcuts_backend.h"
@@ skipping 104 matching lines @@
     AutocompleteMatch::Type output_type;
   } type_cases[] = {
     { AutocompleteMatchType::URL_WHAT_YOU_TYPED,
       AutocompleteMatchType::HISTORY_URL },
     { AutocompleteMatchType::NAVSUGGEST,
       AutocompleteMatchType::HISTORY_URL },
     { AutocompleteMatchType::SEARCH_WHAT_YOU_TYPED,
       AutocompleteMatchType::SEARCH_HISTORY },
     { AutocompleteMatchType::SEARCH_SUGGEST,
       AutocompleteMatchType::SEARCH_HISTORY },
+    { AutocompleteMatchType::SEARCH_SUGGEST_ENTITY,
+      AutocompleteMatchType::SEARCH_HISTORY },
+    { AutocompleteMatchType::SEARCH_SUGGEST_INFINITE,
+      AutocompleteMatchType::SEARCH_HISTORY },
+    { AutocompleteMatchType::SEARCH_SUGGEST_PERSONALIZED,
+      AutocompleteMatchType::SEARCH_HISTORY },
+    { AutocompleteMatchType::SEARCH_SUGGEST_PROFILE,
+      AutocompleteMatchType::SEARCH_HISTORY },
   };
 
   for (size_t i = 0; i < ARRAYSIZE_UNSAFE(type_cases); ++i) {
     EXPECT_EQ(type_cases[i].output_type,
               ConvertedMatchType(type_cases[i].input_type));
   }
 }
 
+TEST_F(ShortcutsBackendTest, EntitySuggestionTest) {
+  AutocompleteMatch match;
+  match.fill_into_edit = base::UTF8ToUTF16("franklin d roosevelt");
+  match.type = AutocompleteMatchType::SEARCH_SUGGEST_ENTITY;
+  match.contents = base::UTF8ToUTF16("roosevelt");
+  match.contents_class =
+      AutocompleteMatch::ClassificationsFromString("0,0,5,2");
+  match.description = base::UTF8ToUTF16("Franklin D. Roosevelt");
+  match.description_class = AutocompleteMatch::ClassificationsFromString("0,4");
+  match.destination_url = GURL(
+      "http://www.google.com/search?q=franklin+d+roosevelt&gs_ssp=1234");
+  match.stripped_destination_url = GURL(
+      "http://www.google.com/search?q=franklin+d+roosevelt");
+
+  ShortcutsBackend::Shortcut::MatchCore match_core(match);
+
+  EXPECT_EQ(match.stripped_destination_url, match_core.destination_url);
+  EXPECT_EQ(match.fill_into_edit, match_core.contents);
+  EXPECT_TRUE(match_core.contents_class.empty());
+  EXPECT_EQ(base::string16(), match_core.description);
+  EXPECT_TRUE(match_core.description_class.empty());
+}
+
 TEST_F(ShortcutsBackendTest, AddAndUpdateShortcut) {
   InitBackend();
   EXPECT_FALSE(changed_notified_);
   ShortcutsBackend::Shortcut shortcut(
       "BD85DBA2-8C29-49F9-84AE-48E1E90880DF", base::ASCIIToUTF16("goog"),
       MatchCoreForTesting("http://www.google.com"), base::Time::Now(), 100);
   EXPECT_TRUE(backend_->AddShortcut(shortcut));
   EXPECT_TRUE(changed_notified_);
   changed_notified_ = false;
 
@@ skipping 54 matching lines @@
   std::vector<std::string> deleted_ids;
   deleted_ids.push_back(shortcut3.id);
   deleted_ids.push_back(shortcut4.id);
 
   EXPECT_TRUE(backend_->DeleteShortcutsWithIds(deleted_ids));
 
   ASSERT_EQ(0U, shortcuts.size());
 }
 
 }  // namespace history