Add keychain prompt mode and read strategy overrides for testing in UsageStore

Author: ratulsarna

Sources/CodexBar/UsageStore.swift

@@ -1301,6 +1301,8 @@ extension UsageStore {
         let credentialsURLOverride: URL?
         let testingOverrides: ClaudeOAuthCredentialsStore.TestingOverridesSnapshot
         let keychainDeniedUntilStoreOverride: ClaudeOAuthKeychainAccessGate.DeniedUntilStore?
+        let keychainPromptModeOverride: ClaudeOAuthKeychainPromptMode?
+        let keychainReadStrategyOverride: ClaudeOAuthKeychainReadStrategy?
         let cliPathOverride: String?
         let statusFetchOverride: ClaudeStatusProbe.FetchOverride?
         #endif
@@ -1318,13 +1320,23 @@ extension UsageStore {
                                             .withDeniedUntilStoreOverrideForTesting(self
                                                 .keychainDeniedUntilStoreOverride)
                                             {
-                                                await ClaudeCLIResolver
-                                                    .withResolvedBinaryPathOverrideForTesting(self
-                                                        .cliPathOverride)
-                                                    {
-                                                        await ClaudeStatusProbe
-                                                            .withFetchOverrideForTesting(self.statusFetchOverride) {
-                                                                await operation()
+                                                await ClaudeOAuthKeychainPromptPreference
+                                                    .withTaskOverrideForTesting(self.keychainPromptModeOverride) {
+                                                        await ClaudeOAuthKeychainReadStrategyPreference
+                                                            .withTaskOverrideForTesting(self
+                                                                .keychainReadStrategyOverride)
+                                                            {
+                                                                await ClaudeCLIResolver
+                                                                    .withResolvedBinaryPathOverrideForTesting(self
+                                                                        .cliPathOverride)
+                                                                    {
+                                                                        await ClaudeStatusProbe
+                                                                            .withFetchOverrideForTesting(self
+                                                                                .statusFetchOverride)
+                                                                            {
+                                                                                await operation()
+                                                                            }
+                                                                    }
                                                             }
                                                     }
                                             }
@@ -1348,6 +1360,8 @@ extension UsageStore {
             credentialsURLOverride: ClaudeOAuthCredentialsStore.currentCredentialsURLOverrideForTesting,
             testingOverrides: ClaudeOAuthCredentialsStore.currentTestingOverridesSnapshotForTask,
             keychainDeniedUntilStoreOverride: ClaudeOAuthKeychainAccessGate.currentDeniedUntilStoreOverrideForTesting,
+            keychainPromptModeOverride: ClaudeOAuthKeychainPromptPreference.currentTaskOverrideForTesting,
+            keychainReadStrategyOverride: ClaudeOAuthKeychainReadStrategyPreference.currentTaskOverrideForTesting,
             cliPathOverride: ClaudeCLIResolver.currentResolvedBinaryPathOverrideForTesting,
             statusFetchOverride: ClaudeStatusProbe.currentFetchOverrideForTesting)
         #else

Sources/CodexBarCore/Providers/Claude/ClaudeOAuth/ClaudeOAuthKeychainPromptMode.swift

@@ -58,7 +58,7 @@ public enum ClaudeOAuthKeychainPromptPreference {
     }
 
     #if DEBUG
-    static func withTaskOverrideForTesting<T>(
+    public static func withTaskOverrideForTesting<T>(
         _ mode: ClaudeOAuthKeychainPromptMode?,
         operation: () throws -> T) rethrows -> T
     {
@@ -67,13 +67,17 @@ public enum ClaudeOAuthKeychainPromptPreference {
         }
     }
 
-    static func withTaskOverrideForTesting<T>(
+    public static func withTaskOverrideForTesting<T>(
         _ mode: ClaudeOAuthKeychainPromptMode?,
         operation: () async throws -> T) async rethrows -> T
     {
         try await self.$taskOverride.withValue(mode) {
             try await operation()
         }
     }
+
+    public static var currentTaskOverrideForTesting: ClaudeOAuthKeychainPromptMode? {
+        self.taskOverride
+    }
     #endif
 }

Sources/CodexBarCore/Providers/Claude/ClaudeOAuth/ClaudeOAuthKeychainReadStrategy.swift

@@ -25,7 +25,7 @@ public enum ClaudeOAuthKeychainReadStrategyPreference {
     }
 
     #if DEBUG
-    static func withTaskOverrideForTesting<T>(
+    public static func withTaskOverrideForTesting<T>(
         _ strategy: ClaudeOAuthKeychainReadStrategy?,
         operation: () throws -> T) rethrows -> T
     {
@@ -34,13 +34,17 @@ public enum ClaudeOAuthKeychainReadStrategyPreference {
         }
     }
 
-    static func withTaskOverrideForTesting<T>(
+    public static func withTaskOverrideForTesting<T>(
         _ strategy: ClaudeOAuthKeychainReadStrategy?,
         operation: () async throws -> T) async rethrows -> T
     {
         try await self.$taskOverride.withValue(strategy) {
             try await operation()
         }
     }
+
+    public static var currentTaskOverrideForTesting: ClaudeOAuthKeychainReadStrategy? {
+        self.taskOverride
+    }
     #endif
 }