@@ -12,6 +12,7 @@ struct POSTabEligibilityCheckerTests {
1212 private var siteSettings : MockSelectedSiteSettings !
1313 private var eligibilityService : MockPOSEligibilityService !
1414 private var mockSystemStatusService : MockPOSSystemStatusService !
15+ private var mockSiteSettingService : MockPOSSiteSettingService !
1516 private let siteID : Int64 = 2
1617
1718 init ( ) async throws {
@@ -21,6 +22,7 @@ struct POSTabEligibilityCheckerTests {
2122 eligibilityService = MockPOSEligibilityService ( )
2223 siteSettings = MockSelectedSiteSettings ( )
2324 mockSystemStatusService = MockPOSSystemStatusService ( )
25+ mockSiteSettingService = MockPOSSiteSettingService ( )
2426 setupWooCommerceVersion ( )
2527 }
2628
@@ -622,7 +624,8 @@ struct POSTabEligibilityCheckerTests {
622624 let checker = POSTabEligibilityChecker ( siteID: siteID,
623625 siteSettings: siteSettings,
624626 stores: stores,
625- systemStatusService: mockSystemStatusService)
627+ systemStatusService: mockSystemStatusService,
628+ siteSettingService: mockSiteSettingService)
626629
627630 // When
628631 let result = try await checker. refreshEligibility ( ineligibleReason: . featureSwitchDisabled)
@@ -873,3 +876,16 @@ private final class MockPOSSystemStatusService: POSSystemStatusServiceProtocol {
873876 }
874877 }
875878}
879+
880+ private final class MockPOSSiteSettingService : POSSiteSettingServiceProtocol {
881+ var setFeatureResult : Result < Bool , Error > = . success( true )
882+
883+ func setFeature( siteID: Int64 , feature: SiteSettingsFeature , enabled: Bool ) async throws -> Bool {
884+ switch setFeatureResult {
885+ case . success( let result) :
886+ return result
887+ case . failure( let error) :
888+ throw error
889+ }
890+ }
891+ }
0 commit comments