Add flag to mock flag service and fix failing tests

Author: iamgabrielma

WooCommerce/WooCommerceTests/Mocks/MockFeatureFlagService.swift

@@ -24,6 +24,7 @@ final class MockFeatureFlagService: FeatureFlagService {
     var isProductGlobalUniqueIdentifierSupported: Bool
     var hideSitesInStorePicker: Bool
     var notificationSettings: Bool
+    var allowMerchantAIAPIKey: Bool
 
     init(isInboxOn: Bool = false,
          isShowInboxCTAEnabled: Bool = false,
@@ -46,7 +47,8 @@ final class MockFeatureFlagService: FeatureFlagService {
          favoriteProducts: Bool = false,
          isProductGlobalUniqueIdentifierSupported: Bool = false,
          hideSitesInStorePicker: Bool = false,
-         notificationSettings: Bool = false) {
+         notificationSettings: Bool = false,
+         allowMerchantAIAPIKey: Bool = false) {
         self.isInboxOn = isInboxOn
         self.isShowInboxCTAEnabled = isShowInboxCTAEnabled
         self.isUpdateOrderOptimisticallyOn = isUpdateOrderOptimisticallyOn
@@ -69,6 +71,7 @@ final class MockFeatureFlagService: FeatureFlagService {
         self.isProductGlobalUniqueIdentifierSupported = isProductGlobalUniqueIdentifierSupported
         self.hideSitesInStorePicker = hideSitesInStorePicker
         self.notificationSettings = notificationSettings
+        self.allowMerchantAIAPIKey = allowMerchantAIAPIKey
     }
 
     func isFeatureFlagEnabled(_ featureFlag: FeatureFlag) -> Bool {
@@ -117,6 +120,8 @@ final class MockFeatureFlagService: FeatureFlagService {
             return hideSitesInStorePicker
         case .notificationSettings:
             return notificationSettings
+        case .allowMerchantAIAPIKey:
+            return allowMerchantAIAPIKey
         default:
             return false
         }

WooCommerce/WooCommerceTests/ViewRelated/HubMenu/HubMenuViewModelTests.swift

@@ -206,11 +206,13 @@ final class HubMenuViewModelTests: XCTestCase {
         stores.updateDefaultStore(storeID: sampleSiteID)
         stores.updateDefaultStore(.fake().copy(siteID: sampleSiteID))
 
+        let featureFlagService = MockFeatureFlagService(allowMerchantAIAPIKey: false)
         let blazeEligibilityChecker = MockBlazeEligibilityChecker(isSiteEligible: true)
 
         // When
         let viewModel = HubMenuViewModel(siteID: sampleSiteID,
                                          tapToPayBadgePromotionChecker: TapToPayBadgePromotionChecker(),
+                                         featureFlagService: featureFlagService,
                                          stores: stores,
                                          blazeEligibilityChecker: blazeEligibilityChecker)
 
@@ -267,11 +269,13 @@ final class HubMenuViewModelTests: XCTestCase {
         stores.updateDefaultStore(storeID: sampleSiteID)
         stores.updateDefaultStore(.fake().copy(siteID: sampleSiteID))
 
+        let featureFlagService = MockFeatureFlagService(allowMerchantAIAPIKey: false)
         let checker = MockGoogleAdsEligibilityChecker(isEligible: true)
 
         // When
         let viewModel = HubMenuViewModel(siteID: sampleSiteID,
                                          tapToPayBadgePromotionChecker: TapToPayBadgePromotionChecker(),
+                                         featureFlagService: featureFlagService,
                                          stores: stores,
                                          googleAdsEligibilityChecker: checker)
         waitUntil {
@@ -568,6 +572,7 @@ final class HubMenuViewModelTests: XCTestCase {
         let blazeEligibilityChecker = MockBlazeEligibilityChecker(isSiteEligible: true)
         let googleAdsEligibilityChecker = MockGoogleAdsEligibilityChecker(isEligible: true)
         let inboxEligibilityChecker = MockInboxEligibilityChecker()
+        let featureFlagService = MockFeatureFlagService(allowMerchantAIAPIKey: false)
         inboxEligibilityChecker.isEligible = true
 
         let stores = MockStoresManager(sessionManager: .makeForTesting())
@@ -578,6 +583,7 @@ final class HubMenuViewModelTests: XCTestCase {
         let navigationPath = NavigationPath(["testPath1", "testPath2"])
         let viewModel = HubMenuViewModel(siteID: sampleSiteID,
                                          tapToPayBadgePromotionChecker: TapToPayBadgePromotionChecker(),
+                                         featureFlagService: featureFlagService,
                                          stores: stores,
                                          generalAppSettings: generalAppSettings,
                                          inboxEligibilityChecker: inboxEligibilityChecker,