Rewire current user email in AppEnvironment/AppDelegateViewModel

Kickstarter-iOS/AppDelegateViewModel.swift

@@ -214,9 +214,11 @@ public final class AppDelegateViewModel: AppDelegateViewModelType, AppDelegateVi
         AppEnvironment.current.apiService.fetchGraphUserEmail().wrapInOptional().materialize()
       }
 
-    _ = fetchUserEmailEvent.values()
+    self.fetchUserEmail = fetchUserEmailEvent.values()
       .map { user in
-        guard let email = user?.me.email else { return }
+        guard let email = user?.me.email else {
+          return
+        }
 
         AppEnvironment.updateCurrentUserEmail(email)
       }
@@ -904,6 +906,7 @@ public final class AppDelegateViewModel: AppDelegateViewModelType, AppDelegateVi
   public let emailVerificationCompleted: Signal<(String, Bool), Never>
   public let findRedirectUrl: Signal<URL, Never>
   public let forceLogout: Signal<(), Never>
+  private let fetchUserEmail: Signal<(), Never>
   public let goToActivity: Signal<(), Never>
   public let goToDiscovery: Signal<DiscoveryParams?, Never>
   public let goToLoginWithIntent: Signal<LoginIntent, Never>

Kickstarter-iOS/AppDelegateViewModelTests.swift

@@ -2434,6 +2434,37 @@ final class AppDelegateViewModelTests: TestCase {
       self.postNotificationName.assertValues([.ksr_remoteConfigClientConfigurationFailed])
     }
   }
+
+  func testUserSessionStarted_fetchesUserEmail_andClearsOnLogout() {
+    let fetchUserEmailQueryData = GraphAPI.FetchUserEmailQuery
+      .Data(unsafeResultMap: [
+        "me": [
+          "email": "nativesquad@ksr.com"
+        ]
+      ]
+      )
+
+    guard let envelope = UserEnvelope<GraphUserEmail>.userEnvelope(from: fetchUserEmailQueryData) else {
+      XCTFail()
+      return
+    }
+
+    let mockService = MockService(
+      fetchGraphUserEmailResult: .success(envelope)
+    )
+
+    withEnvironment(apiService: mockService) {
+      XCTAssertNil(AppEnvironment.current.currentUserEmail)
+
+      self.vm.inputs.userSessionStarted()
+
+      XCTAssertEqual(AppEnvironment.current.currentUserEmail, "nativesquad@ksr.com")
+
+      AppEnvironment.logout()
+
+      XCTAssertNil(AppEnvironment.current.currentUserEmail)
+    }
+  }
 }
 
 private let backingForCreatorPushData: [String: Any] = [

Library/AppEnvironment.swift

@@ -107,6 +107,7 @@ public struct AppEnvironment: AppEnvironmentType {
       apiService: AppEnvironment.current.apiService.logout(),
       cache: type(of: AppEnvironment.current.cache).init(),
       currentUser: nil,
+      currentUserEmail: nil,
       ksrAnalytics: self.current.ksrAnalytics |> KSRAnalytics.lens.loggedInUser .~ nil
     )
   }