Add missing mock

duckduckgo · Dec 21, 2023 · 6290ed0 · 6290ed0
1 parent 64b8ad7
commit 6290ed0
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/DuckDuckGo/ContentBlocker/Mocks/MockPrivacyConfiguration.swift b/DuckDuckGo/ContentBlocker/Mocks/MockPrivacyConfiguration.swift
@@ -51,7 +51,18 @@ final class MockPrivacyConfiguration: PrivacyConfiguration {
     func userDisabledProtection(forDomain: String) {}
 }
 
+final class MockInternalUserStoring: InternalUserStoring {
+    var isInternalUser: Bool = false
+}
+
+extension DefaultInternalUserDecider {
+    convenience init(mockedStore: MockInternalUserStoring = MockInternalUserStoring()) {
+        self.init(store: mockedStore)
+    }
+}
+
 final class MockPrivacyConfigurationManager: NSObject, PrivacyConfigurationManaging {
+
     var embeddedConfigData: BrowserServicesKit.PrivacyConfigurationManager.ConfigurationData {
         fatalError("not implemented")
     }
@@ -70,6 +81,7 @@ final class MockPrivacyConfigurationManager: NSObject, PrivacyConfigurationManag
 
     var updatesPublisher: AnyPublisher<Void, Never> = Just(()).eraseToAnyPublisher()
     var privacyConfig: PrivacyConfiguration = MockPrivacyConfiguration()
+    var internalUserDecider: InternalUserDecider = DefaultInternalUserDecider()
 }
 
 #endif