Merge pull request #10 from SwiftfulThinking/mock_auth_status

Add mock status support
SwiftfulThinking · May 27, 2024 · 683a61b · 683a61b
2 parents 1c98161 + 2da0a2e
commit 683a61b
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 6 deletions.
diff --git a/Sources/SwiftfulFirebaseAuth/Core/AuthManager.swift b/Sources/SwiftfulFirebaseAuth/Core/AuthManager.swift
@@ -19,15 +19,19 @@ public struct AuthInfo {
     }
 }
 
+public enum MockAuthConfiguration {
+    case signInAndOut, signedIn, signedOut
+}
+
 public enum Configuration {
-    case mock, firebase
+    case mock(_ configuration: MockAuthConfiguration), firebase
 
     var provider: AuthProvider {
         switch self {
         case .firebase:
             return FirebaseAuthProvider()
-        case .mock:
-            return MockAuthProvider()
+        case .mock(let configuration):
+            return MockAuthProvider(configuration: configuration)
         }
     }
 }

diff --git a/Sources/SwiftfulFirebaseAuth/Core/AuthManagerEnvironmentKey.swift b/Sources/SwiftfulFirebaseAuth/Core/AuthManagerEnvironmentKey.swift
@@ -10,7 +10,7 @@ import SwiftUI
 
 public struct AuthManagerEnvironmentKey: EnvironmentKey {
     @MainActor
-    public static let defaultValue: AuthManager = AuthManager(configuration: .mock)
+    public static let defaultValue: AuthManager = AuthManager(configuration: .mock(.signInAndOut))
 }
 
 public extension EnvironmentValues {

diff --git a/Sources/SwiftfulFirebaseAuth/Providers/MockAuthProvider.swift b/Sources/SwiftfulFirebaseAuth/Providers/MockAuthProvider.swift
@@ -28,8 +28,15 @@ final class MockAuthProvider: AuthProvider {
         }
     }
 
-    init() {
-        self.authenticatedUser = UserDefaults.userIsSignedIn ? MockAuthProvider.mockUser : nil
+    init(configuration: MockAuthConfiguration) {
+        switch configuration {
+        case .signInAndOut:
+            self.authenticatedUser = UserDefaults.userIsSignedIn ? MockAuthProvider.mockUser : nil
+        case .signedIn:
+            self.authenticatedUser = MockAuthProvider.mockUser
+        case .signedOut:
+            self.authenticatedUser = nil
+        }
     }
 
     private var continuation: AsyncStream<UserAuthInfo?>.Continuation? = nil