[Payment Request] Records the use of data models used in Payment Request.

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 "base/test/scoped_task_environment.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/chrome/browser/payments/test_payment_request.h"
 #include "ios/web/public/payments/payment_request.h"
+#include "testing/gmock/include/gmock/gmock-matchers.h"
+#include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 #if !defined(__has_feature) || !__has_feature(objc_arc)
 #error "This file requires ARC support."
 #endif
 
+namespace {
+
+class MockTestPersonalDataManager : public autofill::TestPersonalDataManager {
+ public:
+  MockTestPersonalDataManager() : TestPersonalDataManager() {}
+  MOCK_METHOD1(RecordUseOf, void(const autofill::AutofillDataModel&));
+};
+
+MATCHER_P(GuidMatches, guid, "") {
+  return arg.guid() == guid;
+}
+
+}  // namespace
+
 class PaymentRequestTest : public testing::Test {
  protected:
   // Returns PaymentDetails with one shipping option that's selected.
   web::PaymentDetails CreateDetailsWithShippingOption() {
     web::PaymentDetails details;
     std::vector<web::PaymentShippingOption> shipping_options;
     web::PaymentShippingOption option1;
     option1.id = base::UTF8ToUTF16("option:1");
     option1.selected = true;
     shipping_options.push_back(std::move(option1));
@@ skipping 502 matching lines @@
 
   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.
   TestPaymentRequest payment_request(web_payment_request,
                                      &personal_data_manager);
   EXPECT_EQ(credit_card.guid(), payment_request.selected_credit_card()->guid());
 }
+
+// Test that the use counts of the data models are updated as expected when
+// different autofill profiles are used as the shipping address and the contact
+// info.
+TEST_F(PaymentRequestTest, RecordUseStats_RequestShippingAndContactInfo) {
+  MockTestPersonalDataManager personal_data_manager;
+  // Add a profile that is incomplete for a contact info, but is used more
+  // frequently so is selected as the default shipping address.
+  autofill::AutofillProfile address = autofill::test::GetFullProfile();
+  address.SetInfo(autofill::AutofillType(autofill::EMAIL_ADDRESS),
+                  base::string16(), "en-US");
+  address.set_use_count(10U);
+  personal_data_manager.AddTestingProfile(&address);
+  autofill::AutofillProfile contact_info = autofill::test::GetFullProfile2();
+  contact_info.set_use_count(5U);
+  personal_data_manager.AddTestingProfile(&contact_info);
+  autofill::CreditCard credit_card = autofill::test::GetCreditCard();
+  personal_data_manager.AddTestingCreditCard(&credit_card);
+  credit_card.set_billing_address_id(address.guid());
+
+  web::PaymentRequest web_payment_request =
+      payment_request_test_util::CreateTestWebPaymentRequest();
+
+  TestPaymentRequest payment_request(web_payment_request,
+                                     &personal_data_manager);
+  EXPECT_EQ(address.guid(),
+            payment_request.selected_shipping_profile()->guid());
+  EXPECT_EQ(contact_info.guid(),
+            payment_request.selected_contact_profile()->guid());
+  EXPECT_EQ(credit_card.guid(), payment_request.selected_credit_card()->guid());
+
+  EXPECT_CALL(personal_data_manager, RecordUseOf(GuidMatches(address.guid())))
+      .Times(1);
+  EXPECT_CALL(personal_data_manager,
+              RecordUseOf(GuidMatches(contact_info.guid())))
+      .Times(1);
+  EXPECT_CALL(personal_data_manager,
+              RecordUseOf(GuidMatches(credit_card.guid())))
+      .Times(1);
+
+  payment_request.RecordUseStats();
+}
+
+// Test that the use counts of the data models are updated as expected when the
+// same autofill profile is used as the shipping address and the contact info.
+TEST_F(PaymentRequestTest, RecordUseStats_SameShippingAndContactInfoProfile) {
+  MockTestPersonalDataManager personal_data_manager;
+  autofill::AutofillProfile address = autofill::test::GetFullProfile();
+  personal_data_manager.AddTestingProfile(&address);
+  autofill::CreditCard credit_card = autofill::test::GetCreditCard();
+  personal_data_manager.AddTestingCreditCard(&credit_card);
+  credit_card.set_billing_address_id(address.guid());
+
+  web::PaymentRequest web_payment_request =
+      payment_request_test_util::CreateTestWebPaymentRequest();
+
+  TestPaymentRequest payment_request(web_payment_request,
+                                     &personal_data_manager);
+  EXPECT_EQ(address.guid(),
+            payment_request.selected_shipping_profile()->guid());
+  EXPECT_EQ(address.guid(), payment_request.selected_contact_profile()->guid());
+  EXPECT_EQ(credit_card.guid(), payment_request.selected_credit_card()->guid());
+
+  // Even though |address| is used for contact info, shipping address, and
+  // credit_card's billing address, the stats should be updated only once.
+  EXPECT_CALL(personal_data_manager, RecordUseOf(GuidMatches(address.guid())))
+      .Times(1);
+  EXPECT_CALL(personal_data_manager,
+              RecordUseOf(GuidMatches(credit_card.guid())))
+      .Times(1);
+
+  payment_request.RecordUseStats();
+}
+
+// Test that the use counts of the data models are updated as expected when no
+// contact information is requested.
+TEST_F(PaymentRequestTest, RecordUseStats_RequestShippingOnly) {
+  MockTestPersonalDataManager personal_data_manager;
+  autofill::AutofillProfile address = autofill::test::GetFullProfile();
+  personal_data_manager.AddTestingProfile(&address);
+  autofill::CreditCard credit_card = autofill::test::GetCreditCard();
+  personal_data_manager.AddTestingCreditCard(&credit_card);
+  credit_card.set_billing_address_id(address.guid());
+
+  web::PaymentRequest web_payment_request =
+      payment_request_test_util::CreateTestWebPaymentRequest();
+  web_payment_request.options.request_payer_name = false;
+  web_payment_request.options.request_payer_email = false;
+  web_payment_request.options.request_payer_phone = false;
+
+  TestPaymentRequest payment_request(web_payment_request,
+                                     &personal_data_manager);
+  EXPECT_EQ(address.guid(),
+            payment_request.selected_shipping_profile()->guid());
+  EXPECT_EQ(nullptr, payment_request.selected_contact_profile());
+  EXPECT_EQ(credit_card.guid(), payment_request.selected_credit_card()->guid());
+
+  EXPECT_CALL(personal_data_manager, RecordUseOf(GuidMatches(address.guid())))
+      .Times(1);
+  EXPECT_CALL(personal_data_manager,
+              RecordUseOf(GuidMatches(credit_card.guid())))
+      .Times(1);
+
+  payment_request.RecordUseStats();
+}
+
+// Test that the use counts of the data models are updated as expected when no
+// shipping information is requested.
+TEST_F(PaymentRequestTest, RecordUseStats_RequestContactInfoOnly) {
+  MockTestPersonalDataManager personal_data_manager;
+  autofill::AutofillProfile address = autofill::test::GetFullProfile();
+  personal_data_manager.AddTestingProfile(&address);
+  autofill::CreditCard credit_card = autofill::test::GetCreditCard();
+  personal_data_manager.AddTestingCreditCard(&credit_card);
+  credit_card.set_billing_address_id(address.guid());
+
+  web::PaymentRequest web_payment_request =
+      payment_request_test_util::CreateTestWebPaymentRequest();
+  web_payment_request.options.request_shipping = false;
+
+  TestPaymentRequest payment_request(web_payment_request,
+                                     &personal_data_manager);
+  EXPECT_EQ(nullptr, payment_request.selected_shipping_profile());
+  EXPECT_EQ(address.guid(), payment_request.selected_contact_profile()->guid());
+  EXPECT_EQ(credit_card.guid(), payment_request.selected_credit_card()->guid());
+
+  EXPECT_CALL(personal_data_manager, RecordUseOf(GuidMatches(address.guid())))
+      .Times(1);
+  EXPECT_CALL(personal_data_manager,
+              RecordUseOf(GuidMatches(credit_card.guid())))
+      .Times(1);
+
+  payment_request.RecordUseStats();
+}
+
+// Test that the use counts of the data models are updated as expected when no
+// shipping or contact information is requested.
+TEST_F(PaymentRequestTest, RecordUseStats_NoShippingOrContactInfoRequested) {
+  MockTestPersonalDataManager personal_data_manager;
+  autofill::AutofillProfile address = autofill::test::GetFullProfile();
+  personal_data_manager.AddTestingProfile(&address);
+  autofill::CreditCard credit_card = autofill::test::GetCreditCard();
+  personal_data_manager.AddTestingCreditCard(&credit_card);
+  credit_card.set_billing_address_id(address.guid());
+
+  web::PaymentRequest web_payment_request =
+      payment_request_test_util::CreateTestWebPaymentRequest();
+  web_payment_request.options.request_shipping = false;
+  web_payment_request.options.request_payer_name = false;
+  web_payment_request.options.request_payer_email = false;
+  web_payment_request.options.request_payer_phone = false;
+
+  TestPaymentRequest payment_request(web_payment_request,
+                                     &personal_data_manager);
+  EXPECT_EQ(nullptr, payment_request.selected_shipping_profile());
+  EXPECT_EQ(nullptr, payment_request.selected_contact_profile());
+  EXPECT_EQ(credit_card.guid(), payment_request.selected_credit_card()->guid());
+
+  EXPECT_CALL(personal_data_manager, RecordUseOf(GuidMatches(address.guid())))
+      .Times(0);
+  EXPECT_CALL(personal_data_manager,
+              RecordUseOf(GuidMatches(credit_card.guid())))
+      .Times(1);
+
+  payment_request.RecordUseStats();
+}