Update unit tests for doneButtonEnabled and completion block

Author: itsmeichigo

WooCommerce/WooCommerceTests/ViewRelated/Shipping Label/Create Shipping Label/Multi-package/ShippingLabelPackagesFormViewModelTests.swift

@@ -78,6 +78,59 @@ class ShippingLabelPackagesFormViewModelTests: XCTestCase {
         XCTAssertEqual(viewModel.itemViewModels.last?.selectedPackageID, package2.packageID)
     }
 
+    func test_doneButtonEnabled_returns_true_when_all_packages_are_valid() {
+        // Given
+        let order = MockOrders().empty().copy(siteID: sampleSiteID)
+        let package1 = ShippingLabelPackageAttributes(packageID: "Box 1", totalWeight: "12", productIDs: [1, 2, 3])
+        let package2 = ShippingLabelPackageAttributes(packageID: "Box 2", totalWeight: "5.5", productIDs: [1, 2, 3])
+
+        // When
+        let viewModel = ShippingLabelPackagesFormViewModel(order: order,
+                                                           packagesResponse: nil,
+                                                           selectedPackages: [package1, package2]) { _ in }
+
+        // Then
+        XCTAssertTrue(viewModel.doneButtonEnabled)
+    }
+
+    func test_doneButtonEnabled_returns_false_when_not_all_packages_are_valid() {
+        // Given
+        let order = MockOrders().empty().copy(siteID: sampleSiteID)
+        let package1 = ShippingLabelPackageAttributes(packageID: "Box 1", totalWeight: "12", productIDs: [1, 2, 3])
+        let package2 = ShippingLabelPackageAttributes(packageID: "Box 2", totalWeight: "5.5", productIDs: [1, 2, 3])
+
+        // When
+        let viewModel = ShippingLabelPackagesFormViewModel(order: order,
+                                                           packagesResponse: nil,
+                                                           selectedPackages: [package1, package2]) { _ in }
+        viewModel.itemViewModels.first?.totalWeight = "0"
+
+        // Then
+        XCTAssertFalse(viewModel.doneButtonEnabled)
+    }
+
+    func test_onCompletion_returns_correctly() {
+        // Given
+        let order = MockOrders().empty().copy(siteID: sampleSiteID)
+        let package1 = ShippingLabelPackageAttributes(packageID: "Box 1", totalWeight: "12", productIDs: [1, 2, 3])
+        let package2 = ShippingLabelPackageAttributes(packageID: "Box 2", totalWeight: "5.5", productIDs: [1, 2, 3])
+
+        var result: [ShippingLabelPackageAttributes] = []
+        let completionHandler: ShippingLabelPackagesFormViewModel.Completion = { packages in
+            result = packages
+        }
+        // When
+        let viewModel = ShippingLabelPackagesFormViewModel(order: order,
+                                                           packagesResponse: nil,
+                                                           selectedPackages: [package1, package2],
+                                                           onCompletion: completionHandler)
+        viewModel.confirmPackageSelection()
+
+        // Then
+        XCTAssertEqual(result.count, 2)
+        XCTAssertEqual(result.first?.packageID, package1.packageID)
+        XCTAssertEqual(result.last?.packageID, package2.packageID)
+    }
 }
 
 // MARK: - Utils