test respond flow

Author: kiftio

Tests/ShopifyCheckoutSheetKitTests/CheckoutAddressChangeStartTests.swift

@@ -25,6 +25,49 @@
 import XCTest
 
 class CheckoutAddressChangeStartTests: XCTestCase {
+    // MARK: - Response Tests
+
+    func testRespondWithSendsJavaScriptToWebView() throws {
+        let mockWebView = MockWebView()
+        let params = CheckoutAddressChangeStartParams(
+            addressType: "shipping",
+            cart: createTestCart()
+        )
+        let request = CheckoutAddressChangeStart(id: "test-id-789", params: params)
+        request.webview = mockWebView
+
+        let payload = CheckoutAddressChangeStartResponsePayload(
+            cart: CartInput(
+                delivery: CartDeliveryInput(
+                    addresses: [
+                        CartSelectableAddressInput(
+                            address: CartDeliveryAddressInput(countryCode: "US")
+                        )
+                    ]
+                )
+            )
+        )
+
+        let expectation = expectation(description: "JavaScript executed")
+        mockWebView.evaluateJavaScriptExpectation = expectation
+
+        try request.respondWith(payload: payload)
+
+        waitForExpectations(timeout: 2.0)
+
+        // Verify the JavaScript was executed and contains the expected JSON-RPC response
+        XCTAssertNotNil(mockWebView.capturedJavaScript, "JavaScript should have been executed")
+
+        let capturedJS = mockWebView.capturedJavaScript ?? ""
+
+        // Verify the response contains expected JSON-RPC fields
+        XCTAssertTrue(capturedJS.contains("window.postMessage"), "Should call window.postMessage")
+        XCTAssertTrue(capturedJS.contains("\"jsonrpc\":\"2.0\""), "Should include JSON-RPC version")
+        XCTAssertTrue(capturedJS.contains("\"id\":\"test-id-789\""), "Should include request ID")
+        XCTAssertTrue(capturedJS.contains("\"result\""), "Should include result field")
+        XCTAssertTrue(capturedJS.contains("\"cart\""), "Should include cart in result")
+    }
+
     // MARK: - Validation Tests
 
     func testValidateAcceptsValid2CharacterCountryCode() throws {

Tests/ShopifyCheckoutSheetKitTests/CheckoutSubmitStartTests.swift

@@ -25,6 +25,40 @@
 import XCTest
 
 class CheckoutSubmitStartTests: XCTestCase {
+    // MARK: - Response Tests
+
+    func testRespondWithSendsJavaScriptToWebView() throws {
+        let mockWebView = MockWebView()
+        let params = CheckoutSubmitStartParams(
+            cart: createTestCart(),
+            checkout: Checkout(id: "test-checkout-123")
+        )
+        let request = CheckoutSubmitStart(id: "test-id-456", params: params)
+        request.webview = mockWebView
+
+        let payload = CheckoutSubmitStartResponsePayload(payment: nil, cart: nil, errors: nil)
+
+        let expectation = expectation(description: "JavaScript executed")
+        mockWebView.evaluateJavaScriptExpectation = expectation
+
+        try request.respondWith(payload: payload)
+
+        waitForExpectations(timeout: 2.0)
+
+        // Verify the JavaScript was executed and contains the expected JSON-RPC response
+        XCTAssertNotNil(mockWebView.capturedJavaScript, "JavaScript should have been executed")
+
+        let capturedJS = mockWebView.capturedJavaScript ?? ""
+
+        // Verify the response contains expected JSON-RPC fields
+        XCTAssertTrue(capturedJS.contains("window.postMessage"), "Should call window.postMessage")
+        XCTAssertTrue(capturedJS.contains("\"jsonrpc\":\"2.0\""), "Should include JSON-RPC version")
+        XCTAssertTrue(capturedJS.contains("\"id\":\"test-id-456\""), "Should include request ID")
+        XCTAssertTrue(capturedJS.contains("\"result\""), "Should include result field")
+    }
+
+    // MARK: - Decoding Tests
+
     func testDecodesCheckoutSessionId() throws {
         let json = """
         {

Tests/ShopifyCheckoutSheetKitTests/Mocks/MockCheckoutWebViewDelegate.swift

@@ -95,4 +95,8 @@ class MockCheckoutWebViewDelegate: CheckoutWebViewDelegate {
     func checkoutViewDidRequestCardChange(event _: CheckoutCardChangeRequested) {
         // No-op for tests unless explicitly asserted
     }
+
+    func checkoutViewDidStartSubmit(event _: CheckoutSubmitStart) {
+        // No-op for tests unless explicitly asserted
+    }
 }

Tests/ShopifyCheckoutSheetKitTests/Mocks/MockWebView.swift

@@ -30,8 +30,13 @@ class MockWebView: CheckoutWebView {
 
     var evaluateJavaScriptExpectation: XCTestExpectation?
 
+    var capturedJavaScript: String?
+
     override func evaluateJavaScript(_ javaScriptString: String) async throws -> Any {
-        if javaScriptString == expectedScript {
+        capturedJavaScript = javaScriptString
+        if !javaScriptString.isEmpty {
+            evaluateJavaScriptExpectation?.fulfill()
+        } else if javaScriptString == expectedScript {
             evaluateJavaScriptExpectation?.fulfill()
         }
         return true