components/search_engines/search_engines_test_util.cc

   1 // Copyright 2017 The Chromium Authors
   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 "components/search_engines/search_engines_test_util.h"
   6
   7 #include <utility>
   8
   9 #include "base/strings/utf_string_conversions.h"
  10 #include "components/search_engines/default_search_manager.h"
  11 #include "components/search_engines/template_url.h"
  12 #include "components/search_engines/template_url_data.h"
  13 #include "components/search_engines/template_url_data_util.h"
  14 #include "components/sync_preferences/testing_pref_service_syncable.h"
  15 #include "testing/gtest/include/gtest/gtest.h"
  16
  17 std::unique_ptr<TemplateURLData> GenerateDummyTemplateURLData(
  18     const std::string& keyword) {
  19   auto data = std::make_unique<TemplateURLData>();
  20   data->SetShortName(base::UTF8ToUTF16(keyword + "name"));
  21   data->SetKeyword(base::UTF8ToUTF16(keyword));
  22   data->SetURL(std::string("http://") + keyword + "foo/{searchTerms}");
  23   data->suggestions_url = std::string("http://") + keyword + "sugg";
  24   data->alternate_urls.push_back(std::string("http://") + keyword + "foo/alt");
  25   data->favicon_url = GURL("http://icon1");
  26   data->safe_for_autoreplace = true;
  27   data->input_encodings = {"UTF-8", "UTF-16"};
  28   data->date_created = base::Time();
  29   data->last_modified = base::Time();
  30   return data;
  31 }
  32
  33 void ExpectSimilar(const TemplateURLData* expected,
  34                    const TemplateURLData* actual) {
  35   ASSERT_TRUE(expected);
  36   ASSERT_TRUE(actual);
  37
  38   EXPECT_EQ(expected->short_name(), actual->short_name());
  39   EXPECT_EQ(expected->keyword(), actual->keyword());
  40   EXPECT_EQ(expected->url(), actual->url());
  41   EXPECT_EQ(expected->suggestions_url, actual->suggestions_url);
  42   EXPECT_EQ(expected->image_url, actual->image_url);
  43   EXPECT_EQ(expected->new_tab_url, actual->new_tab_url);
  44   EXPECT_EQ(expected->contextual_search_url, actual->contextual_search_url);
  45
  46   EXPECT_EQ(expected->search_url_post_params, actual->search_url_post_params);
  47   EXPECT_EQ(expected->suggestions_url_post_params,
  48             actual->suggestions_url_post_params);
  49   EXPECT_EQ(expected->image_url_post_params, actual->image_url_post_params);
  50
  51   EXPECT_EQ(expected->favicon_url, actual->favicon_url);
  52   EXPECT_EQ(expected->safe_for_autoreplace, actual->safe_for_autoreplace);
  53   EXPECT_EQ(expected->input_encodings, actual->input_encodings);
  54   EXPECT_EQ(expected->alternate_urls, actual->alternate_urls);
  55   EXPECT_EQ(expected->created_from_play_api, actual->created_from_play_api);
  56 }
  57
  58 void SetExtensionDefaultSearchInPrefs(
  59     sync_preferences::TestingPrefServiceSyncable* prefs,
  60     const TemplateURLData& data) {
  61   base::Value::Dict entry = TemplateURLDataToDictionary(data);
  62   prefs->SetExtensionPref(
  63       DefaultSearchManager::kDefaultSearchProviderDataPrefName,
  64       std::move(entry));
  65 }
  66
  67 void RemoveExtensionDefaultSearchFromPrefs(
  68     sync_preferences::TestingPrefServiceSyncable* prefs) {
  69   prefs->RemoveExtensionPref(
  70       DefaultSearchManager::kDefaultSearchProviderDataPrefName);
  71 }