[Payment Request] Displays appropriate footer if first transaction completed

ios/chrome/browser/ui/payments/payment_request_mediator_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.
 
 #import "ios/chrome/browser/ui/payments/payment_request_mediator.h"
 
 #import <Foundation/Foundation.h>
 
 #include "base/mac/foundation_util.h"
 #include "base/memory/ptr_util.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/test/scoped_task_environment.h"
 #include "components/autofill/core/browser/autofill_profile.h"
 #include "components/autofill/core/browser/autofill_test_utils.h"
 #include "components/autofill/core/browser/credit_card.h"
 #include "components/autofill/core/browser/test_personal_data_manager.h"
+#include "components/payments/core/payment_prefs.h"
+#include "components/payments/core/payments_test_util.h"
 #include "components/payments/core/strings_util.h"
+#include "components/prefs/pref_service.h"
 #include "components/signin/core/browser/signin_manager.h"
 #include "components/strings/grit/components_strings.h"
 #include "ios/chrome/browser/browser_state/test_chrome_browser_state.h"
 #include "ios/chrome/browser/payments/payment_request_test_util.h"
 #include "ios/chrome/browser/payments/payment_request_util.h"
 #include "ios/chrome/browser/payments/test_payment_request.h"
 #include "ios/chrome/browser/signin/fake_signin_manager_builder.h"
 #include "ios/chrome/browser/signin/signin_manager_factory.h"
 #import "ios/chrome/browser/ui/collection_view/cells/collection_view_detail_item.h"
 #import "ios/chrome/browser/ui/collection_view/cells/collection_view_footer_item.h"
@@ skipping 15 matching lines @@
 using ::payment_request_util::GetEmailLabelFromAutofillProfile;
 using ::payment_request_util::GetNameLabelFromAutofillProfile;
 using ::payment_request_util::GetPhoneNumberLabelFromAutofillProfile;
 using ::payment_request_util::GetShippingAddressLabelFromAutofillProfile;
 }  // namespace
 
 class PaymentRequestMediatorTest : public PlatformTest {
  protected:
   PaymentRequestMediatorTest()
       : autofill_profile_(autofill::test::GetFullProfile()),
-        credit_card_(autofill::test::GetCreditCard()) {
+        credit_card_(autofill::test::GetCreditCard()),
+        pref_service_(payments::test::PrefServiceForTesting()) {
     // Add testing profile and credit card to autofill::TestPersonalDataManager.
     personal_data_manager_.AddTestingProfile(&autofill_profile_);
     credit_card_.set_billing_address_id(autofill_profile_.guid());
     personal_data_manager_.AddTestingCreditCard(&credit_card_);
 
-    payment_request_ = base::MakeUnique<TestPaymentRequest>(
-        payment_request_test_util::CreateTestWebPaymentRequest(),
-        &personal_data_manager_);
-
     TestChromeBrowserState::Builder test_cbs_builder;
     test_cbs_builder.AddTestingFactory(ios::SigninManagerFactory::GetInstance(),
                                        &ios::BuildFakeSigninManager);
     chrome_browser_state_ = test_cbs_builder.Build();
+
+    payment_request_ = base::MakeUnique<TestPaymentRequest>(
+        payment_request_test_util::CreateTestWebPaymentRequest(),
+        chrome_browser_state_.get(), &personal_data_manager_);
+    payment_request_->SetPrefService(pref_service_.get());
+
     mediator_ = [[PaymentRequestMediator alloc]
-        initWithBrowserState:chrome_browser_state_.get()
-              paymentRequest:payment_request_.get()];
+        initWithPaymentRequest:payment_request_.get()];
   }
 
   PaymentRequestMediator* GetPaymentRequestMediator() { return mediator_; }
 
   base::test::ScopedTaskEnvironment scoped_task_evironment_;
 
   autofill::AutofillProfile autofill_profile_;
   autofill::CreditCard credit_card_;
+  std::unique_ptr<PrefService> pref_service_;
   autofill::TestPersonalDataManager personal_data_manager_;
   std::unique_ptr<TestPaymentRequest> payment_request_;
   std::unique_ptr<TestChromeBrowserState> chrome_browser_state_;
   PaymentRequestMediator* mediator_;
 };
 
 // Tests whether payment can be completed when expected.
 TEST_F(PaymentRequestMediatorTest, TestCanPay) {
   EXPECT_TRUE(payment_request_->selected_credit_card());
   EXPECT_TRUE(payment_request_->selected_shipping_profile());
@@ skipping 328 matching lines @@
       base::mac::ObjCCastStrict<CollectionViewDetailItem>(item);
   EXPECT_TRUE([add_contact_info_item.detailText
       isEqualToString:[l10n_util::GetNSString(IDS_ADD)
                           uppercaseStringWithLocale:[NSLocale currentLocale]]]);
   EXPECT_EQ(MDCCollectionViewCellAccessoryNone,
             add_contact_info_item.accessoryType);
 }
 
 // Tests that the Footer item is created as expected.
 TEST_F(PaymentRequestMediatorTest, TestFooterItem) {
+  // Make sure the first transaction has not completed yet.
+  pref_service_->SetBoolean(payments::kPaymentsFirstTransactionCompleted,
+                            false);
+
   // Make sure the user is signed out.
   SigninManager* signin_manager = ios::SigninManagerFactory::GetForBrowserState(
       chrome_browser_state_.get());
   if (signin_manager->IsAuthenticated()) {
     signin_manager->SignOut(signin_metrics::SIGNOUT_TEST,
                             signin_metrics::SignoutDelete::IGNORE_METRIC);
   }
 
   // Footer item should be of type CollectionViewFooterItem.
   id item = [GetPaymentRequestMediator() footerItem];
   ASSERT_TRUE([item isMemberOfClass:[CollectionViewFooterItem class]]);
   CollectionViewFooterItem* footer_item =
       base::mac::ObjCCastStrict<CollectionViewFooterItem>(item);
   EXPECT_TRUE([footer_item.text
       isEqualToString:l10n_util::GetNSString(
                           IDS_PAYMENTS_CARD_AND_ADDRESS_SETTINGS_SIGNED_OUT)]);
 
   // Fake a signed in user.
   signin_manager->SetAuthenticatedAccountInfo("12345", "username@example.com");
 
   item = [GetPaymentRequestMediator() footerItem];
   footer_item = base::mac::ObjCCastStrict<CollectionViewFooterItem>(item);
   EXPECT_TRUE([footer_item.text
       isEqualToString:l10n_util::GetNSStringF(
                           IDS_PAYMENTS_CARD_AND_ADDRESS_SETTINGS_SIGNED_IN,
                           base::ASCIIToUTF16("username@example.com"))]);
+
+  // Record that the first transaction completed.
+  pref_service_->SetBoolean(payments::kPaymentsFirstTransactionCompleted, true);
+
+  item = [GetPaymentRequestMediator() footerItem];
+  footer_item = base::mac::ObjCCastStrict<CollectionViewFooterItem>(item);
+  EXPECT_TRUE([footer_item.text
+      isEqualToString:l10n_util::GetNSString(
+                          IDS_PAYMENTS_CARD_AND_ADDRESS_SETTINGS)]);
+
+  // Sign the user out.
+  signin_manager->SignOut(signin_metrics::SIGNOUT_TEST,
+                          signin_metrics::SignoutDelete::IGNORE_METRIC);
+
+  // The signed in state has no effect on the footer text if the first
+  // transaction has completed.
+  footer_item = base::mac::ObjCCastStrict<CollectionViewFooterItem>(item);
+  EXPECT_TRUE([footer_item.text
+      isEqualToString:l10n_util::GetNSString(
+                          IDS_PAYMENTS_CARD_AND_ADDRESS_SETTINGS)]);
 }