Handle origin address selection

woocommerce · Jan 8, 2025 · a3611c5 · a3611c5
1 parent 32559f8
commit a3611c5
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 1 deletion.
diff --git a/...g Create Shipping Labels/WooShippingAddresses/WooShippingOriginAddressListViewModel.swift b/...g Create Shipping Labels/WooShippingAddresses/WooShippingOriginAddressListViewModel.swift
@@ -5,6 +5,9 @@ final class WooShippingOriginAddressListViewModel: ObservableObject {
     let addresses: [WooShippingOriginAddress]
     @Published private(set) var selectedAddressID: String?
 
+    /// Closure (set externally) called when an address is selected.
+    var onSelect: ((WooShippingOriginAddress) -> Void)?
+
     init(addresses: [WooShippingOriginAddress],
          selectedAddressID: String? = nil) {
         self.addresses = addresses
@@ -22,6 +25,7 @@ final class WooShippingOriginAddressListViewModel: ObservableObject {
             return
         }
         selectedAddressID = address.id
+        onSelect?(address)
     }
 }
 

diff --git a/...Shipping Labels/WooShipping Create Shipping Labels/WooShippingCreateLabelsViewModel.swift b/...Shipping Labels/WooShipping Create Shipping Labels/WooShippingCreateLabelsViewModel.swift
@@ -245,7 +245,11 @@ private extension WooShippingCreateLabelsViewModel {
             switch result {
             case .success(let addresses):
                 selectedOriginAddress = addresses.first(where: \.defaultAddress)
-                originAddresses = WooShippingOriginAddressListViewModel(addresses: addresses, selectedAddressID: selectedOriginAddress?.id)
+                originAddresses = WooShippingOriginAddressListViewModel(addresses: addresses,
+                                                                        selectedAddressID: selectedOriginAddress?.id)
+                originAddresses.onSelect = { [weak self] selectedAddress in
+                    self?.selectedOriginAddress = selectedAddress
+                }
             case .failure(let error):
                 DDLogError("⛔️ Error loading origin addresses for Woo Shipping labels: \(error)")
             }

diff --git a/...Label/WooShipping Create Shipping Labels/WooShippingOriginAddressListViewModelTests.swift b/...Label/WooShipping Create Shipping Labels/WooShippingOriginAddressListViewModelTests.swift
@@ -56,4 +56,22 @@ final class WooShippingOriginAddressListViewModelTests: XCTestCase {
         XCTAssertEqual(viewModel.selectedAddressID, addressToSelect.id)
     }
 
+    func test_select_calls_onSelect_closure() {
+        // Given
+        let addressToSelect = WooShippingOriginAddress.fake().copy(id: "1")
+        let addresses = [addressToSelect, WooShippingOriginAddress.fake().copy(id: "2")]
+        let viewModel = WooShippingOriginAddressListViewModel(addresses: addresses, selectedAddressID: nil)
+
+        // When
+        let selectedAddress = waitFor { promise in
+            viewModel.onSelect = { address in
+                promise(address)
+            }
+            viewModel.select(addressToSelect)
+        }
+
+        // Then
+        XCTAssertEqual(selectedAddress, addressToSelect)
+    }
+
 }