Change subscription manager to use environment variable token

duckduckgo · Nov 1, 2024 · 4b4305c · 4b4305c
1 parent 92a6d5e
commit 4b4305c
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/...ources/DataBrokerProtection/Authentication/DataBrokerProtectionSubscriptionManaging.swift b/...ources/DataBrokerProtection/Authentication/DataBrokerProtectionSubscriptionManaging.swift
@@ -19,6 +19,7 @@
 import Foundation
 import Subscription
 import Common
+import AppKitExtensions
 
 public protocol DataBrokerProtectionSubscriptionManaging {
     var isUserAuthenticated: Bool { get }
@@ -31,20 +32,23 @@ public final class DataBrokerProtectionSubscriptionManager: DataBrokerProtection
     let subscriptionManager: SubscriptionManager
 
     public var isUserAuthenticated: Bool {
-        return true
-        subscriptionManager.accountManager.accessToken != nil
+        accessToken != nil
     }
 
     public var accessToken: String? {
-        subscriptionManager.accountManager.accessToken
+        // We use a staging token for privacy pro supplied through a github secret/action
+        // for PIR end to end tests
+        if NSApplication.runType == .integrationTests, let token = ProcessInfo.processInfo.environment["PRIVACYPRO_STAGING_TOKEN"] {
+            return token
+        }
+        return subscriptionManager.accountManager.accessToken
     }
 
     public init(subscriptionManager: SubscriptionManager) {
         self.subscriptionManager = subscriptionManager
     }
 
     public func hasValidEntitlement() async throws -> Bool {
-        return true
         switch await subscriptionManager.accountManager.hasEntitlement(forProductName: .dataBrokerProtection,
                                                                        cachePolicy: .reloadIgnoringLocalCacheData) {
         case let .success(result):