Update mocks

duckduckgo · Nov 13, 2024 · 0b7c142 · 0b7c142
1 parent 05e9407
commit 0b7c142
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 3 deletions.
diff --git a/IntegrationTests/PhishingDetection/PhishingDetectionIntegrationTests.swift b/IntegrationTests/PhishingDetection/PhishingDetectionIntegrationTests.swift
@@ -148,7 +148,10 @@ class PhishingDetectionIntegrationTests: XCTestCase {
 }
 
 class MockFeatureFlagger: FeatureFlagger {
-    func isFeatureOn<F>(forProvider: F) -> Bool where F: BrowserServicesKit.FeatureFlagSourceProviding {
+    var internalUserDecider: InternalUserDecider = DefaultInternalUserDecider(store: MockInternalUserStoring())
+    var localOverrides: FeatureFlagLocalOverriding?
+
+    func isFeatureOn<Flag: FeatureFlagProtocol>(for featureFlag: Flag, allowOverride: Bool) -> Bool {
         return true
     }
 }
diff --git a/UnitTests/Menus/MainMenuTests.swift b/UnitTests/Menus/MainMenuTests.swift
@@ -194,7 +194,10 @@ class MainMenuTests: XCTestCase {
 }
 
 private class DummyFeatureFlagger: FeatureFlagger {
-    func isFeatureOn<F: BrowserServicesKit.FeatureFlagSourceProviding>(forProvider: F) -> Bool {
+    var internalUserDecider: InternalUserDecider = DefaultInternalUserDecider(store: MockInternalUserStoring())
+    var localOverrides: FeatureFlagLocalOverriding?
+
+    func isFeatureOn<Flag: FeatureFlagProtocol>(for: Flag, allowOverride: Bool) -> Bool {
         false
     }
 }

diff --git a/UnitTests/TabExtensionsTests/ErrorPageTabExtensionTest.swift b/UnitTests/TabExtensionsTests/ErrorPageTabExtensionTest.swift
@@ -480,8 +480,11 @@ class ChallangeSender: URLAuthenticationChallengeSender {
 }
 
 class MockFeatureFlagger: FeatureFlagger {
+    var internalUserDecider: InternalUserDecider = DefaultInternalUserDecider(store: MockInternalUserStoring())
+    var localOverrides: FeatureFlagLocalOverriding?
+
     var isFeatureOn = true
-    func isFeatureOn<F>(forProvider: F) -> Bool where F: BrowserServicesKit.FeatureFlagSourceProviding {
+    func isFeatureOn<Flag: FeatureFlagProtocol>(for featureFlag: Flag, allowOverride: Bool) -> Bool {
         return isFeatureOn
     }
 }