Index: components/autofill/core/browser/autofill_test_utils.cc |
diff --git a/components/autofill/core/browser/autofill_test_utils.cc b/components/autofill/core/browser/autofill_test_utils.cc |
index 86f8ccb65de186153629ba8bc5f813ce0dc728e6..557fb8fb8c3cc8ec1aac0b3c7655b604c15de9ea 100644 |
--- a/components/autofill/core/browser/autofill_test_utils.cc |
+++ b/components/autofill/core/browser/autofill_test_utils.cc |
@@ -13,6 +13,7 @@ |
#include "components/autofill/core/browser/autofill_profile.h" |
#include "components/autofill/core/browser/credit_card.h" |
#include "components/autofill/core/browser/field_types.h" |
+#include "components/autofill/core/browser/webdata/autofill_table.h" |
#include "components/autofill/core/common/autofill_pref_names.h" |
#include "components/autofill/core/common/form_data.h" |
#include "components/autofill/core/common/form_field_data.h" |
@@ -219,5 +220,24 @@ void DisableSystemServices(PrefService* prefs) { |
#endif // defined(OS_MACOSX) && !defined(OS_IOS) |
} |
+void SetServerCreditCards(AutofillTable* table, |
+ const std::vector<CreditCard>& cards) { |
+ std::vector<CreditCard> as_masked_cards = cards; |
+ for (CreditCard& card : as_masked_cards) { |
+ card.set_record_type(CreditCard::MASKED_SERVER_CARD); |
+ std::string type = card.type(); |
+ card.SetNumber(card.LastFourDigits()); |
+ card.SetTypeForMaskedCard(type.c_str()); |
+ } |
+ table->SetServerCreditCards(as_masked_cards); |
+ |
+ for (const CreditCard& card : cards) { |
+ if (card.record_type() != CreditCard::FULL_SERVER_CARD) |
+ continue; |
+ |
+ table->UnmaskServerCreditCard(card.server_id(), card.number()); |
+ } |
+} |
+ |
} // namespace test |
} // namespace autofill |