Adds screen object elements and methods to add shipping to a new order

Author: rachelmcr

WooCommerce/Classes/ViewRelated/Orders/Order Creation/PaymentSection/OrderPaymentSection.swift

@@ -67,6 +67,7 @@ struct OrderPaymentSection: View {
             }
             .buttonStyle(PlusButtonStyle())
             .padding()
+            .accessibilityIdentifier("add-shipping-button")
         }
     }
 

WooCommerce/UITestsFoundation/Screens/Orders/NewOrderScreen.swift

@@ -23,6 +23,10 @@ public final class NewOrderScreen: ScreenObject {
         $0.staticTexts["Add Customer Details"]
     }
 
+    private let addShippingButtonGetter: (XCUIApplication) -> XCUIElement = {
+        $0.buttons["add-shipping-button"]
+    }
+
     private var createButton: XCUIElement { createButtonGetter(app) }
 
     /// Cancel button in the Navigation bar.
@@ -41,6 +45,10 @@ public final class NewOrderScreen: ScreenObject {
     ///
     private var addCustomerDetailsButton: XCUIElement { addCustomerDetailsButtonGetter(app) }
 
+    /// Add Shipping button in the Payment section.
+    ///
+    private var addShippingButton: XCUIElement { addShippingButtonGetter(app) }
+
     public init(app: XCUIApplication = XCUIApplication()) throws {
         try super.init(
             expectedElementGetters: [ createButtonGetter ],
@@ -74,6 +82,14 @@ public final class NewOrderScreen: ScreenObject {
         return try CustomerDetailsScreen()
     }
 
+    /// Opens the Add Shipping screen.
+    /// - Returns: Add Shipping screen object.
+    @discardableResult
+    public func openAddShippingScreen() throws -> AddShippingScreen {
+        addShippingButton.tap()
+        return try AddShippingScreen()
+    }
+
 // MARK: - High-level Order Creation actions
 
     /// Creates a remote order with all of the entered order data.
@@ -107,6 +123,18 @@ public final class NewOrderScreen: ScreenObject {
             .enterCustomerDetails(name: name)
     }
 
+    /// Adds shipping on the Add Shipping screen.
+    /// - Parameters:
+    ///   - amount: Amount (in the store currency) to add for shipping.
+    ///   - name: Name of the shipping method (e.g. "Free Shipping" or "Flat Rate").
+    /// - Returns: New Order screen object.
+    public func addShipping(amount: String, name: String) throws -> NewOrderScreen {
+        return try openAddShippingScreen()
+            .enterShippingAmount(amount)
+            .enterShippingName(name)
+            .confirmShippingDetails()
+    }
+
     /// Cancels Order Creation process
     /// - Returns: Orders Screen object.
     @discardableResult