Add unit tests

Author: Ecarrion

WooCommerce/Classes/ViewRelated/Products/Variations/ProductVariationsViewModel.swift

@@ -40,7 +40,7 @@ final class ProductVariationsViewModel {
             print("Variations to Generate: \(variationsToGenerate.count)")
 
             // Guard for 100 variation limit
-            guard variationsToGenerate.count < 100 else {
+            guard variationsToGenerate.count <= 100 else {
                 return onCompletion(.failure(.tooManyVariations(variationCount: variationsToGenerate.count)))
             }
 
@@ -82,7 +82,7 @@ extension ProductVariationsViewModel {
 extension ProductVariationsViewModel {
     /// Type to represent known generation errors
     ///
-    enum GenerationError: LocalizedError {
+    enum GenerationError: LocalizedError, Equatable {
         case tooManyVariations(variationCount: Int)
 
         var errorTitle: String {

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

@@ -114,4 +114,27 @@ final class ProductVariationsViewModelTests: XCTestCase {
         XCTAssertTrue(product.existsRemotely)
         XCTAssertEqual(viewModel.formType, .readonly)
     }
+
+    func test_trying_to_generate_more_than_100_variations_will_return_error() {
+        // 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"]),
+            ProductAttribute.fake().copy(attributeID: 3, name: "Fabric", options: ["Cotton", "Nylon", "Polyester", "Silk", "Linen"]),
+        ])
+        let viewModel = ProductVariationsViewModel(formType: .edit)
+
+        // When
+        let error = waitFor { promise in
+            viewModel.generateAllVariations(for: product) { result in
+                if case let .failure(error) = result {
+                    promise(error)
+                }
+            }
+        }
+
+        // Then
+        XCTAssertEqual(error, .tooManyVariations(variationCount: 125))
+
+    }
 }