blob: 56d7984793b3b431f46605d8e47ecbe03ce01b9b [file] [log] [blame]
// 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 <memory>
#include <string>
#include "base/files/scoped_temp_dir.h"
#include "base/macros.h"
#include "base/memory/ref_counted.h"
#include "base/strings/string16.h"
#include "components/search_engines/template_url.h"
#include "components/search_engines/template_url_data.h"
#include "components/search_engines/template_url_service_observer.h"
class GURL;
class KeywordWebDataService;
class TemplateURLService;
class TestingProfile;
class TestingSearchTermsData;
// Sets the managed preferences for the default search provider.
// enabled arg enables/disables use of managed engine by DefaultSearchManager.
void SetManagedDefaultSearchPreferences(const TemplateURLData& managed_data,
bool enabled,
TestingProfile* profile);
// Removes all the managed preferences for the default search provider.
void RemoveManagedDefaultSearchPreferences(TestingProfile* profile);
class TemplateURLServiceTestUtil : public TemplateURLServiceObserver {
~TemplateURLServiceTestUtil() override;
// TemplateURLServiceObserver implemementation.
void OnTemplateURLServiceChanged() override;
// Gets the observer count.
int GetObserverCount();
// Sets the observer count to 0.
void ResetObserverCount();
// Makes sure the load was successful and sent the correct notification.
void VerifyLoad();
// Makes the model believe it has been loaded (without actually doing the
// load). Since this avoids setting the built-in keyword version, the next
// load will do a merge from prepopulated data.
void ChangeModelToLoadState();
// Deletes the current model (and doesn't create a new one).
void ClearModel();
// Creates a new TemplateURLService.
void ResetModel(bool verify_load);
// Returns the search term from the last invocation of
// TemplateURLService::SetKeywordSearchTermsForURL and clears the search term.
base::string16 GetAndClearSearchTerm();
// Sets the google base url. |base_url| must be valid.
void SetGoogleBaseURL(const GURL& base_url);
// Adds extension controlled TemplateURL to the model and overrides default
// search pref in an extension controlled preferences, if extension wants to
// be default.
TemplateURL* AddExtensionControlledTURL(
std::unique_ptr<TemplateURL> extension_turl);
// Removes a TemplateURL controlled by |extension_id| from the model, and,
// if necessary, from the extension-controlled default search preference.
// This TemplateURL must exist.
void RemoveExtensionControlledTURL(const std::string& extension_id);
KeywordWebDataService* web_data_service() { return web_data_service_.get(); }
TemplateURLService* model() { return model_.get(); }
TestingProfile* profile() { return profile_.get(); }
std::unique_ptr<TestingProfile> profile_;
base::ScopedTempDir temp_dir_;
int changed_count_;
base::string16 search_term_;
scoped_refptr<KeywordWebDataService> web_data_service_;
TestingSearchTermsData* search_terms_data_;
std::unique_ptr<TemplateURLService> model_;