Fix TestStore.init with prepareDependencies. (pointfreeco#1955)

* Fix TestStore.init with prepareDependencies.

* Update Tests/ComposableArchitectureTests/TestStoreTests.swift

Co-authored-by: Oron Ben Zvi <[email protected]>

* Update Tests/ComposableArchitectureTests/TestStoreTests.swift

Co-authored-by: Oron Ben Zvi <[email protected]>

---------

Co-authored-by: Oron Ben Zvi <[email protected]>

Author: mbrandonw

Sources/ComposableArchitecture/TestStore.swift

@@ -666,8 +666,8 @@ public final class TestStore<State, Action, ScopedState, ScopedAction, Environme
     Environment == Void
   {
     var dependencies = DependencyValues._current
-    prepareDependencies(&dependencies)
     let initialState = withDependencies {
+      prepareDependencies(&dependencies)
       $0 = dependencies
     } operation: {
       initialState()

Tests/ComposableArchitectureTests/TestStoreTests.swift

@@ -328,12 +328,14 @@ final class TestStoreTests: XCTestCase {
   func testOverrideDependenciesOnTestStore_Init() {
     struct Counter: ReducerProtocol {
       @Dependency(\.calendar) var calendar
+      @Dependency(\.client.fetch) var fetch
       @Dependency(\.locale) var locale
       @Dependency(\.timeZone) var timeZone
       @Dependency(\.urlSession) var urlSession
 
       func reduce(into state: inout Int, action: Bool) -> EffectTask<Bool> {
         _ = self.calendar
+        _ = self.fetch()
         _ = self.locale
         _ = self.timeZone
         _ = self.urlSession
@@ -347,6 +349,7 @@ final class TestStoreTests: XCTestCase {
       reducer: Counter()
     ) {
       $0.calendar = Calendar(identifier: .gregorian)
+      $0.client.fetch = { 1 }
       $0.locale = Locale(identifier: "en_US")
       $0.timeZone = TimeZone(secondsFromGMT: 0)!
       $0.urlSession = URLSession(configuration: .ephemeral)
@@ -402,3 +405,14 @@ final class TestStoreTests: XCTestCase {
     }
   }
 }
+
+private struct Client: DependencyKey {
+  var fetch: () -> Int
+  static let liveValue = Client(fetch: { 42 })
+}
+extension DependencyValues {
+  fileprivate var client: Client {
+    get { self[Client.self] }
+    set { self[Client.self] = newValue }
+  }
+}