1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
|
// Copyright 2017 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/browser/test_omnibox_client.h"
#include <memory>
#include <utility>
#include <vector>
#include "components/omnibox/browser/autocomplete_controller.h"
#include "components/omnibox/browser/autocomplete_scheme_classifier.h"
#include "components/omnibox/browser/mock_autocomplete_provider_client.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"
#include "third_party/skia/include/core/SkBitmap.h"
#include "ui/gfx/image/image.h"
#include "ui/gfx/image/image_skia.h"
TestOmniboxClient::TestOmniboxClient()
: session_id_(SessionID::FromSerializedValue(1)),
bookmark_model_(nullptr),
autocomplete_classifier_(
std::make_unique<AutocompleteController>(
CreateAutocompleteProviderClient(),
nullptr,
AutocompleteClassifier::DefaultOmniboxProviders()),
std::make_unique<TestSchemeClassifier>()) {}
TestOmniboxClient::~TestOmniboxClient() {
autocomplete_classifier_.Shutdown();
}
std::unique_ptr<AutocompleteProviderClient>
TestOmniboxClient::CreateAutocompleteProviderClient() {
std::unique_ptr<MockAutocompleteProviderClient> provider_client(
new MockAutocompleteProviderClient());
EXPECT_CALL(*provider_client, GetBuiltinURLs())
.WillRepeatedly(testing::Return(std::vector<base::string16>()));
EXPECT_CALL(*provider_client, GetSchemeClassifier())
.WillRepeatedly(testing::ReturnRef(scheme_classifier_));
std::unique_ptr<TemplateURLService> template_url_service(
new TemplateURLService(
nullptr, std::unique_ptr<SearchTermsData>(new SearchTermsData),
nullptr, std::unique_ptr<TemplateURLServiceClient>(), nullptr,
nullptr, base::Closure()));
// Save a reference to the created TemplateURLService for test use.
template_url_service_ = template_url_service.get();
provider_client->set_template_url_service(std::move(template_url_service));
return std::move(provider_client);
}
std::unique_ptr<OmniboxNavigationObserver>
TestOmniboxClient::CreateOmniboxNavigationObserver(
const base::string16& text,
const AutocompleteMatch& match,
const AutocompleteMatch& alternate_nav_match) {
alternate_nav_match_ = alternate_nav_match;
return nullptr;
}
bool TestOmniboxClient::IsPasteAndGoEnabled() const {
return true;
}
const SessionID& TestOmniboxClient::GetSessionID() const {
return session_id_;
}
void TestOmniboxClient::SetBookmarkModel(
bookmarks::BookmarkModel* bookmark_model) {
bookmark_model_ = bookmark_model;
}
bookmarks::BookmarkModel* TestOmniboxClient::GetBookmarkModel() {
return bookmark_model_;
}
TemplateURLService* TestOmniboxClient::GetTemplateURLService() {
DCHECK(template_url_service_);
return template_url_service_;
}
const AutocompleteSchemeClassifier& TestOmniboxClient::GetSchemeClassifier()
const {
return scheme_classifier_;
}
AutocompleteClassifier* TestOmniboxClient::GetAutocompleteClassifier() {
return &autocomplete_classifier_;
}
gfx::Image TestOmniboxClient::GetSizedIcon(
const gfx::VectorIcon& vector_icon_type,
SkColor vector_icon_color) const {
SkBitmap bitmap;
bitmap.allocN32Pixels(16, 16);
return gfx::Image(gfx::ImageSkia::CreateFrom1xBitmap(bitmap));
}
gfx::Image TestOmniboxClient::GetFaviconForPageUrl(
const GURL& page_url,
FaviconFetchedCallback on_favicon_fetched) {
page_url_for_last_favicon_request_ = page_url;
return gfx::Image();
}
GURL TestOmniboxClient::GetPageUrlForLastFaviconRequest() const {
return page_url_for_last_favicon_request_;
}
|
