Add tests for product type selector in CIAB environment

Author: RafaelKayumov

WooCommerce/Classes/ViewRelated/Products/Edit Product/BottomSheetListSelector/ProductTypeBottomSheetListSelectorCommand.swift

@@ -39,7 +39,7 @@ final class ProductTypeBottomSheetListSelectorCommand: BottomSheetListSelectorCo
 
     init(source: Source,
          subscriptionProductsEligibilityChecker: WooSubscriptionProductsEligibilityCheckerProtocol,
-         siteCIABEligibilityChecker: CIABEligibilityCheckerProtocol,
+         siteCIABEligibilityChecker: CIABEligibilityCheckerProtocol = CIABEligibilityChecker(),
          onSelection: @escaping (BottomSheetProductType) -> Void) {
         self.source = source
         self.onSelection = onSelection

WooCommerce/WooCommerceTests/Mocks/MockCIABEligibilityChecker.swift

@@ -22,7 +22,7 @@ final class MockCIABEligibilityChecker: CIABEligibilityCheckerProtocol {
     }
 
     func isFeatureSupportedForCurrentSite(_ feature: CIABAffectedFeature) -> Bool {
-        return !mockedCIABDisabledFeatures.contains(feature)
+        return !mockedCIABDisabledFeatures.contains(feature) || !isCurrentSiteCIAB
     }
 
     func isFeatureSupported(_ feature: CIABAffectedFeature, for site: Site) -> Bool {

WooCommerce/WooCommerceTests/ViewRelated/Products/BottomSheetListSelector/ProductTypeBottomSheetListSelectorCommandTests.swift

@@ -120,4 +120,72 @@ final class ProductTypeBottomSheetListSelectorCommandTests: XCTestCase {
         ]
         XCTAssertEqual(selectedActions, expectedActions)
     }
+
+    func test_creation_form_data_does_not_contain_grouped_and_variable_types_if_site_is_ciab() {
+        // Given
+        let subscriptionEligibilityChecker = MockWooSubscriptionProductsEligibilityChecker(isEligible: true)
+        let siteCIABEligibilityChecker = MockCIABEligibilityChecker(mockedIsCurrentSiteCIAB: true)
+
+        // When
+        let command = ProductTypeBottomSheetListSelectorCommand(
+            source: .creationForm,
+            subscriptionProductsEligibilityChecker: subscriptionEligibilityChecker,
+            siteCIABEligibilityChecker: siteCIABEligibilityChecker
+        ) { _ in }
+
+        // Then
+        XCTAssertFalse(command.data.contains(.grouped), "'command.data' should not contain grouped product type")
+        XCTAssertFalse(command.data.contains(.variable), "'command.data' should not contain variable product type")
+    }
+
+    func test_creation_form_data_contains_grouped_and_variable_types_if_site_is_non_ciab_and_other_requirements_met() {
+        // Given
+        let subscriptionEligibilityChecker = MockWooSubscriptionProductsEligibilityChecker(isEligible: true)
+        let siteCIABEligibilityChecker = MockCIABEligibilityChecker(mockedIsCurrentSiteCIAB: false)
+
+        // When
+        let command = ProductTypeBottomSheetListSelectorCommand(
+            source: .creationForm,
+            subscriptionProductsEligibilityChecker: subscriptionEligibilityChecker,
+            siteCIABEligibilityChecker: siteCIABEligibilityChecker
+        ) { _ in }
+
+        // Then
+        XCTAssertTrue(command.data.contains(.grouped), "'command.data' should contain grouped product type")
+        XCTAssertTrue(command.data.contains(.variable), "'command.data' should contain variable product type")
+    }
+
+    func test_edit_form_data_does_not_contain_grouped_and_variable_types_if_site_is_ciab() {
+        // Given
+        let subscriptionEligibilityChecker = MockWooSubscriptionProductsEligibilityChecker(isEligible: true)
+        let siteCIABEligibilityChecker = MockCIABEligibilityChecker(mockedIsCurrentSiteCIAB: true)
+
+        // When
+        let command = ProductTypeBottomSheetListSelectorCommand(
+            source: .editForm(selected: .simple(isVirtual: false)),
+            subscriptionProductsEligibilityChecker: subscriptionEligibilityChecker,
+            siteCIABEligibilityChecker: siteCIABEligibilityChecker
+        ) { _ in }
+
+        // Then
+        XCTAssertFalse(command.data.contains(.grouped), "'command.data' should not contain grouped product type")
+        XCTAssertFalse(command.data.contains(.variable), "'command.data' should not contain variable product type")
+    }
+
+    func test_edit_form_data_contains_grouped_and_variable_types_if_site_is_non_ciab_and_other_requirements_met() {
+        // Given
+        let subscriptionEligibilityChecker = MockWooSubscriptionProductsEligibilityChecker(isEligible: true)
+        let siteCIABEligibilityChecker = MockCIABEligibilityChecker(mockedIsCurrentSiteCIAB: false)
+
+        // When
+        let command = ProductTypeBottomSheetListSelectorCommand(
+            source: .editForm(selected: .simple(isVirtual: false)),
+            subscriptionProductsEligibilityChecker: subscriptionEligibilityChecker,
+            siteCIABEligibilityChecker: siteCIABEligibilityChecker
+        ) { _ in }
+
+        // Then
+        XCTAssertTrue(command.data.contains(.grouped), "'command.data' should contain grouped product type")
+        XCTAssertTrue(command.data.contains(.variable), "'command.data' should contain variable product type")
+    }
 }