Add screen object elements and methods to add fee to a new order

Author: rachelmcr

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

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

WooCommerce/UITestsFoundation/Screens/Orders/NewOrderScreen.swift

@@ -27,6 +27,10 @@ public final class NewOrderScreen: ScreenObject {
         $0.buttons["add-shipping-button"]
     }
 
+    private let addFeeButtonGetter: (XCUIApplication) -> XCUIElement = {
+        $0.buttons["add-fee-button"]
+    }
+
     private var createButton: XCUIElement { createButtonGetter(app) }
 
     /// Cancel button in the Navigation bar.
@@ -49,6 +53,10 @@ public final class NewOrderScreen: ScreenObject {
     ///
     private var addShippingButton: XCUIElement { addShippingButtonGetter(app) }
 
+    /// Add Fee button in the Payment section.
+    ///
+    private var addFeeButton: XCUIElement { addFeeButtonGetter(app) }
+
     public init(app: XCUIApplication = XCUIApplication()) throws {
         try super.init(
             expectedElementGetters: [ createButtonGetter ],
@@ -90,6 +98,14 @@ public final class NewOrderScreen: ScreenObject {
         return try AddShippingScreen()
     }
 
+    /// Opens the Add Fee screen.
+    /// - Returns: Add Fee screen object.
+    @discardableResult
+    public func openAddFeeScreen() throws -> AddFeeScreen {
+        addFeeButton.tap()
+        return try AddFeeScreen()
+    }
+
 // MARK: - High-level Order Creation actions
 
     /// Creates a remote order with all of the entered order data.
@@ -135,6 +151,16 @@ public final class NewOrderScreen: ScreenObject {
             .confirmShippingDetails()
     }
 
+    /// Adds a fee on the Add Fee screen.
+    /// - Parameters:
+    ///   - amount: Amount (in the store currency) to add as a fee.
+    /// - Returns: New Order screen object.
+    public func addFee(amount: String) throws -> NewOrderScreen {
+        return try openAddFeeScreen()
+            .enterFixedFee(amount: amount)
+            .confirmFee()
+    }
+
     /// Cancels Order Creation process
     /// - Returns: Orders Screen object.
     @discardableResult