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
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
|
// Copyright 2014 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.
#ifndef COMPONENTS_AUTOFILL_CORE_BROWSER_TEST_AUTOFILL_CLIENT_H_
#define COMPONENTS_AUTOFILL_CORE_BROWSER_TEST_AUTOFILL_CLIENT_H_
#include <memory>
#include <string>
#include <utility>
#include <vector>
#include "base/compiler_specific.h"
#include "base/i18n/rtl.h"
#include "base/macros.h"
#include "build/build_config.h"
#include "components/autofill/core/browser/autofill_client.h"
#include "components/autofill/core/browser/mock_autocomplete_history_manager.h"
#include "components/autofill/core/browser/payments/test_payments_client.h"
#include "components/autofill/core/browser/test_address_normalizer.h"
#include "components/autofill/core/browser/test_form_data_importer.h"
#include "components/autofill/core/browser/test_legacy_strike_database.h"
#include "components/autofill/core/browser/test_personal_data_manager.h"
#include "components/autofill/core/browser/test_strike_database.h"
#include "components/prefs/pref_service.h"
#include "components/ukm/test_ukm_recorder.h"
#include "services/identity/public/cpp/identity_test_environment.h"
namespace autofill {
// This class is for easier writing of tests.
class TestAutofillClient : public AutofillClient {
public:
TestAutofillClient();
~TestAutofillClient() override;
// AutofillClient:
PersonalDataManager* GetPersonalDataManager() override;
AutocompleteHistoryManager* GetAutocompleteHistoryManager() override;
PrefService* GetPrefs() override;
syncer::SyncService* GetSyncService() override;
identity::IdentityManager* GetIdentityManager() override;
FormDataImporter* GetFormDataImporter() override;
payments::PaymentsClient* GetPaymentsClient() override;
LegacyStrikeDatabase* GetLegacyStrikeDatabase() override;
StrikeDatabase* GetStrikeDatabase() override;
ukm::UkmRecorder* GetUkmRecorder() override;
ukm::SourceId GetUkmSourceId() override;
AddressNormalizer* GetAddressNormalizer() override;
security_state::SecurityLevel GetSecurityLevelForUmaHistograms() override;
void ShowAutofillSettings(bool show_credit_card_settings) override;
void ShowUnmaskPrompt(const CreditCard& card,
UnmaskCardReason reason,
base::WeakPtr<CardUnmaskDelegate> delegate) override;
void OnUnmaskVerificationResult(PaymentsRpcResult result) override;
void ShowLocalCardMigrationDialog(
base::OnceClosure show_migration_dialog_closure) override;
void ConfirmMigrateLocalCardToCloud(
std::unique_ptr<base::DictionaryValue> legal_message,
const std::string& user_email,
const std::vector<MigratableCreditCard>& migratable_credit_cards,
LocalCardMigrationCallback start_migrating_cards_callback) override;
void ShowLocalCardMigrationResults(
const bool has_server_error,
const base::string16& tip_message,
const std::vector<MigratableCreditCard>& migratable_credit_cards,
MigrationDeleteCardCallback delete_local_card_callback) override;
void ConfirmSaveAutofillProfile(const AutofillProfile& profile,
base::OnceClosure callback) override;
void ConfirmSaveCreditCardLocally(
const CreditCard& card,
SaveCreditCardOptions options,
LocalSaveCardPromptCallback callback) override;
#if defined(OS_ANDROID)
void ConfirmAccountNameFixFlow(
base::OnceCallback<void(const base::string16&)> callback) override;
void ConfirmExpirationDateFixFlow(
const CreditCard& card,
base::OnceCallback<void(const base::string16&, const base::string16&)>
callback) override;
#endif // defined(OS_ANDROID)
void ConfirmSaveCreditCardToCloud(
const CreditCard& card,
std::unique_ptr<base::DictionaryValue> legal_message,
SaveCreditCardOptions options,
UploadSaveCardPromptCallback callback) override;
void ConfirmCreditCardFillAssist(const CreditCard& card,
base::OnceClosure callback) override;
bool HasCreditCardScanFeature() override;
void ScanCreditCard(const CreditCardScanCallback& callback) override;
void ShowAutofillPopup(
const gfx::RectF& element_bounds,
base::i18n::TextDirection text_direction,
const std::vector<Suggestion>& suggestions,
bool autoselect_first_suggestion,
base::WeakPtr<AutofillPopupDelegate> delegate) override;
void UpdateAutofillPopupDataListValues(
const std::vector<base::string16>& values,
const std::vector<base::string16>& labels) override;
void HideAutofillPopup() override;
bool IsAutocompleteEnabled() override;
void PropagateAutofillPredictions(
content::RenderFrameHost* rfh,
const std::vector<autofill::FormStructure*>& forms) override;
void DidFillOrPreviewField(const base::string16& autofilled_value,
const base::string16& profile_full_name) override;
// By default, TestAutofillClient will report that the context is
// secure. This can be adjusted by calling set_form_origin() with an
// http:// URL.
bool IsContextSecure() override;
bool ShouldShowSigninPromo() override;
bool AreServerCardsSupported() override;
void ExecuteCommand(int id) override;
// RiskDataLoader:
void LoadRiskData(
base::OnceCallback<void(const std::string&)> callback) override;
// Initializes UKM source from form_origin_. This needs to be called
// in unittests after calling Purge for ukm recorder to re-initialize
// sources.
void InitializeUKMSources();
void SetPrefs(std::unique_ptr<PrefService> prefs) {
prefs_ = std::move(prefs);
}
void set_test_legacy_strike_database(
std::unique_ptr<TestLegacyStrikeDatabase> test_legacy_strike_database) {
test_legacy_strike_database_ = std::move(test_legacy_strike_database);
}
void set_test_strike_database(
std::unique_ptr<TestStrikeDatabase> test_strike_database) {
test_strike_database_ = std::move(test_strike_database);
}
void set_test_payments_client(
std::unique_ptr<payments::TestPaymentsClient> payments_client) {
payments_client_ = std::move(payments_client);
}
void set_test_form_data_importer(
std::unique_ptr<TestFormDataImporter> form_data_importer) {
form_data_importer_ = std::move(form_data_importer);
}
void set_form_origin(const GURL& url);
void set_sync_service(syncer::SyncService* test_sync_service) {
test_sync_service_ = test_sync_service;
}
void set_security_level(security_state::SecurityLevel security_level) {
security_level_ = security_level;
}
bool ConfirmSaveCardLocallyWasCalled() {
return confirm_save_credit_card_locally_called_;
}
bool get_offer_to_save_credit_card_bubble_was_shown() {
return offer_to_save_credit_card_bubble_was_shown_.value();
}
MockAutocompleteHistoryManager* GetMockAutocompleteHistoryManager() {
return &mock_autocomplete_history_manager_;
}
void set_migration_card_selections(
const std::vector<std::string>& migration_card_selection) {
migration_card_selection_ = migration_card_selection;
}
GURL form_origin() { return form_origin_; }
static void UpdateSourceURL(ukm::UkmRecorder* ukm_recorder,
ukm::SourceId source_id,
GURL url);
private:
identity::IdentityTestEnvironment identity_test_env_;
syncer::SyncService* test_sync_service_ = nullptr;
TestAddressNormalizer test_address_normalizer_;
TestPersonalDataManager test_personal_data_manager_;
MockAutocompleteHistoryManager mock_autocomplete_history_manager_;
// NULL by default.
std::unique_ptr<PrefService> prefs_;
std::unique_ptr<TestLegacyStrikeDatabase> test_legacy_strike_database_;
std::unique_ptr<TestStrikeDatabase> test_strike_database_;
std::unique_ptr<payments::PaymentsClient> payments_client_;
std::unique_ptr<FormDataImporter> form_data_importer_;
GURL form_origin_;
ukm::SourceId source_id_ = -1;
security_state::SecurityLevel security_level_ =
security_state::SecurityLevel::NONE;
bool confirm_save_credit_card_locally_called_ = false;
// Populated if save was offered. True if bubble was shown, false otherwise.
base::Optional<bool> offer_to_save_credit_card_bubble_was_shown_;
// Populated if name fix flow was offered. True if bubble was shown, false
// otherwise.
base::Optional<bool> credit_card_name_fix_flow_bubble_was_shown_;
std::vector<std::string> migration_card_selection_;
DISALLOW_COPY_AND_ASSIGN(TestAutofillClient);
};
} // namespace autofill
#endif // COMPONENTS_AUTOFILL_CORE_BROWSER_TEST_AUTOFILL_CLIENT_H_
|
