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

Author: rachelmcr

WooCommerce/Classes/ViewRelated/Orders/Order Creation/CustomerNoteSection/CustomerNoteSection.swift

@@ -83,6 +83,7 @@ private struct CustomerNoteSectionContent: View {
             }
             .buttonStyle(PlusButtonStyle())
             .padding([.leading, .bottom, .trailing])
+            .accessibilityIdentifier("add-customer-note-button")
         }
     }
 

WooCommerce/UITestsFoundation/Screens/Orders/NewOrderScreen.swift

@@ -31,6 +31,10 @@ public final class NewOrderScreen: ScreenObject {
         $0.buttons["add-fee-button"]
     }
 
+    private let addNoteButtonGetter: (XCUIApplication) -> XCUIElement = {
+        $0.buttons["add-customer-note-button"]
+    }
+
     private var createButton: XCUIElement { createButtonGetter(app) }
 
     /// Cancel button in the Navigation bar.
@@ -57,6 +61,10 @@ public final class NewOrderScreen: ScreenObject {
     ///
     private var addFeeButton: XCUIElement { addFeeButtonGetter(app) }
 
+    /// Add Note button in the Customer Note section.
+    ///
+    private var addNoteButton: XCUIElement { addNoteButtonGetter(app) }
+
     public init(app: XCUIApplication = XCUIApplication()) throws {
         try super.init(
             expectedElementGetters: [ createButtonGetter ],
@@ -106,6 +114,14 @@ public final class NewOrderScreen: ScreenObject {
         return try AddFeeScreen()
     }
 
+    /// Opens the Customer Note screen.
+    /// - Returns: Customer Note screen object.
+    @discardableResult
+    public func openCustomerNoteScreen() throws -> CustomerNoteScreen {
+        addNoteButton.tap()
+        return try CustomerNoteScreen()
+    }
+
 // MARK: - High-level Order Creation actions
 
     /// Creates a remote order with all of the entered order data.
@@ -161,6 +177,15 @@ public final class NewOrderScreen: ScreenObject {
             .confirmFee()
     }
 
+    /// Adds a note on the Customer Note screen.
+    /// - Parameter text: Text to enter as the customer note.
+    /// - Returns: New Order screen object.
+    public func addCustomerNote(_ text: String) throws -> NewOrderScreen {
+        return try openCustomerNoteScreen()
+            .enterNote(text)
+            .confirmNote()
+    }
+
     /// Cancels Order Creation process
     /// - Returns: Orders Screen object.
     @discardableResult