Adding new shipping address editor view to payment flow.

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

Index: chrome/browser/ui/views/payments/payment_request_browsertest_base.cc
diff --git a/chrome/browser/ui/views/payments/payment_request_browsertest_base.cc b/chrome/browser/ui/views/payments/payment_request_browsertest_base.cc
index c787046ff4ad0b35f58cac1f71b35e41ce096858..968166c2a058aafb1b4fb887b3e3706c38d629fe 100644
--- a/chrome/browser/ui/views/payments/payment_request_browsertest_base.cc
+++ b/chrome/browser/ui/views/payments/payment_request_browsertest_base.cc
@@ -4,6 +4,9 @@
 
 #include "chrome/browser/ui/views/payments/payment_request_browsertest_base.h"
 
+#include <memory>
+#include <string>
+#include <utility>
 #include <vector>
 
 #include "base/bind.h"
@@ -97,11 +100,21 @@ void PaymentRequestBrowserTestBase::OnPaymentMethodOpened() {
     event_observer_->Observe(DialogEvent::PAYMENT_METHOD_OPENED);
 }
 
+void PaymentRequestBrowserTestBase::OnShippingSectionOpened() {
+  if (event_observer_)
+    event_observer_->Observe(DialogEvent::SHIPPING_SECTION_OPENED);
+}
+
 void PaymentRequestBrowserTestBase::OnCreditCardEditorOpened() {
   if (event_observer_)
     event_observer_->Observe(DialogEvent::CREDIT_CARD_EDITOR_OPENED);
 }
 
+void PaymentRequestBrowserTestBase::OnShippingAddressEditorOpened() {
+  if (event_observer_)
+    event_observer_->Observe(DialogEvent::SHIPPING_ADDRESS_EDITOR_OPENED);
+}
+
 void PaymentRequestBrowserTestBase::OnBackNavigation() {
   if (event_observer_)
     event_observer_->Observe(DialogEvent::BACK_NAVIGATION);
@@ -112,6 +125,11 @@ void PaymentRequestBrowserTestBase::OnContactInfoOpened() {
     event_observer_->Observe(DialogEvent::CONTACT_INFO_OPENED);
 }
 
+void PaymentRequestBrowserTestBase::OnEditorViewUpdated() {
+  if (event_observer_)
+    event_observer_->Observe(DialogEvent::EDITOR_VIEW_UPDATED);
+}
+
 void PaymentRequestBrowserTestBase::OnWidgetDestroyed(views::Widget* widget) {
   if (event_observer_)
     event_observer_->Observe(DialogEvent::DIALOG_CLOSED);
@@ -162,12 +180,24 @@ void PaymentRequestBrowserTestBase::OpenPaymentMethodScreen() {
   ClickOnDialogViewAndWait(DialogViewID::PAYMENT_SHEET_PAYMENT_METHOD_SECTION);
 }
 
+void PaymentRequestBrowserTestBase::OpenShippingSectionScreen() {
+  ResetEventObserver(DialogEvent::SHIPPING_SECTION_OPENED);
+
+  ClickOnDialogViewAndWait(DialogViewID::PAYMENT_SHEET_SHIPPING_SECTION);
+}
+
 void PaymentRequestBrowserTestBase::OpenCreditCardEditorScreen() {
   ResetEventObserver(DialogEvent::CREDIT_CARD_EDITOR_OPENED);
 
   ClickOnDialogViewAndWait(DialogViewID::PAYMENT_METHOD_ADD_CARD_BUTTON);
 }
 
+void PaymentRequestBrowserTestBase::OpenShippingAddressEditorScreen() {
+  ResetEventObserver(DialogEvent::SHIPPING_ADDRESS_EDITOR_OPENED);
+
+  ClickOnDialogViewAndWait(DialogViewID::PAYMENT_METHOD_ADD_SHIPPING_BUTTON);
+}
+
 content::WebContents* PaymentRequestBrowserTestBase::GetActiveWebContents() {
   return browser()->tab_strip_model()->GetActiveWebContents();
 }