[Payments] Add the pay button, and control its enabled state

chrome/browser/ui/views/payments/payment_request_interactive_uitest_base.cc

 // 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 "chrome/browser/ui/views/payments/payment_request_interactive_uitest_base.h"
 
 #include <vector>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/memory/ptr_util.h"
 #include "base/run_loop.h"
 #include "base/strings/string16.h"
 #include "chrome/browser/autofill/personal_data_manager_factory.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/browser/ui/views/payments/payment_request_dialog_view_ids.h"
 #include "chrome/browser/ui/views/payments/validating_combobox.h"
 #include "chrome/browser/ui/views/payments/validating_textfield.h"
 #include "chrome/browser/ui/views/payments/view_stack.h"
 #include "chrome/test/base/interactive_test_utils.h"
 #include "chrome/test/base/ui_test_utils.h"
+#include "components/autofill/core/browser/autofill_profile.h"
+#include "components/autofill/core/browser/credit_card.h"
+#include "components/autofill/core/browser/personal_data_manager.h"
 #include "components/payments/content/payment_request.h"
 #include "components/payments/content/payment_request_web_contents_manager.h"
 #include "components/web_modal/web_contents_modal_dialog_manager.h"
 #include "content/public/browser/render_frame_host.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/common/content_switches.h"
 #include "content/public/test/browser_test_utils.h"
 #include "services/service_manager/public/cpp/interface_registry.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "ui/base/test/ui_controls.h"
 #include "ui/gfx/animation/test_animation_delegate.h"
+#include "ui/views/controls/button/button.h"
 #include "ui/views/controls/label.h"
 #include "ui/views/controls/styled_label.h"
 
 namespace payments {
 
 PersonalDataLoadedObserverMock::PersonalDataLoadedObserverMock() {}
 PersonalDataLoadedObserverMock::~PersonalDataLoadedObserverMock() {}
 
 PaymentRequestInteractiveTestBase::PaymentRequestInteractiveTestBase(
     const std::string& test_file_path)
@@ skipping 112 matching lines @@
     payment_requests_ptrs.push_back(p.first);
   return payment_requests_ptrs;
 }
 
 autofill::PersonalDataManager*
 PaymentRequestInteractiveTestBase::GetDataManager() {
   return autofill::PersonalDataManagerFactory::GetForProfile(
       Profile::FromBrowserContext(GetActiveWebContents()->GetBrowserContext()));
 }
 
+void PaymentRequestInteractiveTestBase::AddAutofillProfile(
+    const autofill::AutofillProfile& profile) {
+  autofill::PersonalDataManager* personal_data_manager = GetDataManager();
+  size_t profile_count = personal_data_manager->GetProfiles().size();
+
+  PersonalDataLoadedObserverMock personal_data_observer;
+  personal_data_manager->AddObserver(&personal_data_observer);
+  base::RunLoop data_loop;
+  EXPECT_CALL(personal_data_observer, OnPersonalDataChanged())
+      .WillOnce(QuitMessageLoop(&data_loop));
+  personal_data_manager->AddProfile(profile);
+  data_loop.Run();
+
+  personal_data_manager->RemoveObserver(&personal_data_observer);
+  EXPECT_EQ(profile_count + 1, personal_data_manager->GetProfiles().size());
+}
+
+void PaymentRequestInteractiveTestBase::AddCreditCard(
+    const autofill::CreditCard& card) {
+  autofill::PersonalDataManager* personal_data_manager = GetDataManager();
+  size_t card_count = personal_data_manager->GetCreditCards().size();
+
+  PersonalDataLoadedObserverMock personal_data_observer;
+  personal_data_manager->AddObserver(&personal_data_observer);
+  base::RunLoop data_loop;
+  EXPECT_CALL(personal_data_observer, OnPersonalDataChanged())
+      .WillOnce(QuitMessageLoop(&data_loop));
+  personal_data_manager->AddCreditCard(card);
+  data_loop.Run();
+
+  personal_data_manager->RemoveObserver(&personal_data_observer);
+  EXPECT_EQ(card_count + 1, personal_data_manager->GetCreditCards().size());
+}
+
 void PaymentRequestInteractiveTestBase::CreatePaymentRequestForTest(
     content::WebContents* web_contents,
     mojo::InterfaceRequest<payments::mojom::PaymentRequest> request) {
   DCHECK(web_contents);
   std::unique_ptr<TestChromePaymentRequestDelegate> delegate =
       base::MakeUnique<TestChromePaymentRequestDelegate>(
           web_contents, this /* observer */, this /* widget_observer */);
   delegate_ = delegate.get();
   PaymentRequestWebContentsManager::GetOrCreateForWebContents(web_contents)
       ->CreatePaymentRequest(web_contents, std::move(delegate),
@@ skipping 53 matching lines @@
 }
 
 bool PaymentRequestInteractiveTestBase::IsEditorComboboxInvalid(
     autofill::ServerFieldType type) {
   ValidatingCombobox* combobox = static_cast<ValidatingCombobox*>(
       delegate_->dialog_view()->GetViewByID(static_cast<int>(type)));
   DCHECK(combobox);
   return combobox->invalid();
 }
 
+bool PaymentRequestInteractiveTestBase::IsPayButtonEnabled() {
+  views::Button* button =
+      static_cast<views::Button*>(delegate_->dialog_view()->GetViewByID(
+          static_cast<int>(DialogViewID::PAYMENT_SHEET_PAY_BUTTON)));
+  DCHECK(button);
+  return button->enabled();
+}
+
 void PaymentRequestInteractiveTestBase::WaitForAnimation() {
   ViewStack* view_stack = dialog_view()->view_stack_for_testing();
   if (view_stack->slide_in_animator_->IsAnimating()) {
     view_stack->slide_in_animator_->SetAnimationDuration(1);
     view_stack->slide_in_animator_->SetAnimationDelegate(
         view_stack->top(), std::unique_ptr<gfx::AnimationDelegate>(
                                new gfx::TestAnimationDelegate()));
     base::RunLoop().Run();
   } else if (view_stack->slide_out_animator_->IsAnimating()) {
     view_stack->slide_out_animator_->SetAnimationDuration(1);
@@ skipping 46 matching lines @@
 
 void PaymentRequestInteractiveTestBase::ResetEventObserver(DialogEvent event) {
   event_observer_ = base::MakeUnique<DialogEventObserver>(event);
 }
 
 void PaymentRequestInteractiveTestBase::WaitForObservedEvent() {
   event_observer_->Wait();
 }
 
 }  // namespace payments