[Payment Request] Makes sure only complete credit cards can be selected

ios/chrome/browser/payments/payment_request_unittest.mm

 // 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 "ios/chrome/browser/payments/payment_request.h"
 
 #include "base/strings/utf_string_conversions.h"
 #include "components/autofill/core/browser/autofill_test_utils.h"
 #include "components/autofill/core/browser/autofill_type.h"
 #include "components/autofill/core/browser/field_types.h"
 #include "components/autofill/core/browser/test_personal_data_manager.h"
 #include "components/payments/core/currency_formatter.h"
 #include "components/payments/core/payment_method_data.h"
 #include "ios/chrome/browser/application_context.h"
+#include "ios/chrome/browser/payments/payment_request_test_util.h"
 #include "ios/web/public/payments/payment_request.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 #if !defined(__has_feature) || !__has_feature(objc_arc)
 #error "This file requires ARC support."
 #endif
 
 class PaymentRequestTest : public testing::Test {
  protected:
   // Returns PaymentDetails with one shipping option that's selected.
@@ skipping 410 matching lines @@
   // address1 has more use counts, and even though it has no phone number, it's
   // still selected as the contact profile because merchant doesn't require
   // phone. address2 is selected as the shipping profile because it's the most
   // complete for shipping.
   PaymentRequest payment_request(web_payment_request, &personal_data_manager);
   EXPECT_EQ(address2.guid(),
             payment_request.selected_shipping_profile()->guid());
   EXPECT_EQ(address1.guid(),
             payment_request.selected_contact_profile()->guid());
 }
+
+// Test that loading payment methods when none are available works as expected.
+TEST_F(PaymentRequestTest, SelectedPaymentMethod_NoPaymentMethods) {
+  autofill::TestPersonalDataManager personal_data_manager;
+  web::PaymentRequest web_payment_request =
+      payment_request_test_util::CreateTestWebPaymentRequest();
+
+  // No payment methods are selected because none are available!
+  PaymentRequest payment_request(web_payment_request, &personal_data_manager);
+  EXPECT_EQ(nullptr, payment_request.selected_credit_card());
+}
+
+// Test that loading expired credit cards works as expected.
+TEST_F(PaymentRequestTest, SelectedPaymentMethod_ExpiredCard) {
+  autofill::TestPersonalDataManager personal_data_manager;
+  autofill::AutofillProfile billing_address = autofill::test::GetFullProfile();
+  personal_data_manager.AddTestingProfile(&billing_address);
+  autofill::CreditCard credit_card = autofill::test::GetCreditCard();
+  personal_data_manager.AddTestingCreditCard(&credit_card);
+  credit_card.SetExpirationYear(2016);  // Expired.
+  credit_card.set_billing_address_id(billing_address.guid());
+
+  web::PaymentRequest web_payment_request =
+      payment_request_test_util::CreateTestWebPaymentRequest();
+
+  // credit_card is selected because expired cards are valid for payment.
+  PaymentRequest payment_request(web_payment_request, &personal_data_manager);
+  EXPECT_EQ(credit_card.guid(), payment_request.selected_credit_card()->guid());
+}
+
+// Test that loading complete payment methods works as expected.
+TEST_F(PaymentRequestTest, SelectedPaymentMethod_Complete) {
+  autofill::TestPersonalDataManager personal_data_manager;
+  autofill::AutofillProfile billing_address = autofill::test::GetFullProfile();
+  personal_data_manager.AddTestingProfile(&billing_address);
+  autofill::CreditCard credit_card = autofill::test::GetCreditCard();
+  credit_card.set_use_count(5U);
+  personal_data_manager.AddTestingCreditCard(&credit_card);
+  credit_card.set_billing_address_id(billing_address.guid());
+  autofill::CreditCard credit_card2 = autofill::test::GetCreditCard2();
+  credit_card2.set_use_count(15U);
+  personal_data_manager.AddTestingCreditCard(&credit_card2);
+  credit_card2.set_billing_address_id(billing_address.guid());
+
+  web::PaymentRequest web_payment_request =
+      payment_request_test_util::CreateTestWebPaymentRequest();
+
+  // credit_card2 is selected because it has the most use count (Frecency
+  // model).
+  PaymentRequest payment_request(web_payment_request, &personal_data_manager);
+  EXPECT_EQ(credit_card2.guid(),
+            payment_request.selected_credit_card()->guid());
+}
+
+// Test that loading incomplete payment methods works as expected.
+TEST_F(PaymentRequestTest, SelectedPaymentMethod_Incomplete) {
+  autofill::TestPersonalDataManager personal_data_manager;
+  autofill::AutofillProfile billing_address = autofill::test::GetFullProfile();
+  personal_data_manager.AddTestingProfile(&billing_address);
+  autofill::CreditCard credit_card = autofill::test::GetCreditCard();
+  credit_card.set_use_count(5U);
+  personal_data_manager.AddTestingCreditCard(&credit_card);
+  credit_card.set_billing_address_id(billing_address.guid());
+  autofill::CreditCard credit_card2 = autofill::test::GetCreditCard2();
+  credit_card2.set_use_count(15U);
+  personal_data_manager.AddTestingCreditCard(&credit_card2);
+
+  web::PaymentRequest web_payment_request =
+      payment_request_test_util::CreateTestWebPaymentRequest();
+
+  // Even though credit_card2 has more use counts, credit_card is selected
+  // because it is complete.
+  PaymentRequest payment_request(web_payment_request, &personal_data_manager);
+  EXPECT_EQ(credit_card.guid(), payment_request.selected_credit_card()->guid());
+}