Adapts unit tests

Author: Ecarrion

WooCommerce/WooCommerceTests/ViewRelated/Products/Variations/ProductVariationsViewModelTests.swift

@@ -137,8 +137,8 @@ final class ProductVariationsViewModelTests: XCTestCase {
 
         // When
         let error = waitFor { promise in
-            viewModel.generateAllVariations(for: product) { result in
-                if case let .failure(error) = result {
+            viewModel.generateAllVariations(for: product) { state in
+                if case let .error(error) = state {
                     promise(error)
                 }
             }
@@ -148,7 +148,7 @@ final class ProductVariationsViewModelTests: XCTestCase {
         XCTAssertEqual(error, .tooManyVariations(variationCount: 125))
     }
 
-    func test_generating_less_than_100_variations_invokes_create_action() {
+    func test_generating_less_than_100_variations_ask_for_confirmation_and_creates_variations() {
         // Given
         let product = Product.fake().copy(attributes: [
             ProductAttribute.fake().copy(attributeID: 1, name: "Size", options: ["XS", "S", "M", "L", "XL"]),
@@ -171,8 +171,11 @@ final class ProductVariationsViewModelTests: XCTestCase {
 
         // When
         let succeeded = waitFor { promise in
-            viewModel.generateAllVariations(for: product) { result in
-                if case .success = result {
+            viewModel.generateAllVariations(for: product) { state in
+                if case let .confirmation(_, onCompletion) = state {
+                    onCompletion(true)
+                }
+                if case .finished = state {
                     promise(true)
                 }
             }
@@ -181,4 +184,41 @@ final class ProductVariationsViewModelTests: XCTestCase {
         // Then
         XCTAssertTrue(succeeded)
     }
+
+    func test_generating_less_than_100_variations_ask_for_confirmation_and_sends_cancel_state() {
+        // Given
+        let product = Product.fake().copy(attributes: [
+            ProductAttribute.fake().copy(attributeID: 1, name: "Size", options: ["XS", "S", "M", "L", "XL"]),
+            ProductAttribute.fake().copy(attributeID: 2, name: "Color", options: ["Red", "Green", "Blue", "White", "Black"]),
+        ])
+
+        let stores = MockStoresManager(sessionManager: SessionManager.makeForTesting())
+        stores.whenReceivingAction(ofType: ProductVariationAction.self) { action in
+            switch action {
+            case .synchronizeAllProductVariations(_, _, let onCompletion):
+                onCompletion(.success([]))
+            case .createProductVariations(_, _, _, let onCompletion):
+                onCompletion(.success([]))
+            default:
+                break
+            }
+        }
+
+        let viewModel = ProductVariationsViewModel(stores: stores, formType: .edit)
+
+        // When
+        let canceled = waitFor { promise in
+            viewModel.generateAllVariations(for: product) { state in
+                if case let .confirmation(_, onCompletion) = state {
+                    onCompletion(false)
+                }
+                if case .canceled = state {
+                    promise(true)
+                }
+            }
+        }
+
+        // Then
+        XCTAssertTrue(canceled)
+    }
 }