Test store dependency binding (#1620)

* Bind dependencies when constructing test store and when asserting.

* clean up

* clean up;

* feedback

(cherry picked from commit 9a22c5a8a9d0d31fb0f2e61481cb026d5e83282a)

# Conflicts:
#	Tests/ComposableArchitectureTests/TestStoreTests.swift

Author: mbrandonw

Sources/ComposableArchitecture/TestStore.swift

@@ -568,8 +568,9 @@ open class TestStore<State, Action, ScopedState, ScopedAction, Environment> {
   ///   - initialState: The state the feature starts in.
   ///   - reducer: The reducer that powers the runtime of the feature.
   public init<Reducer: ReducerProtocol>(
-    initialState: State,
+    initialState: @autoclosure () -> State,
     reducer: Reducer,
+    prepareDependencies: (inout DependencyValues) -> Void = { _ in },
     file: StaticString = #file,
     line: UInt = #line
   )
@@ -580,6 +581,11 @@ open class TestStore<State, Action, ScopedState, ScopedAction, Environment> {
     Action == ScopedAction,
     Environment == Void
   {
+    var dependencies = DependencyValues()
+    dependencies.context = .test
+    prepareDependencies(&dependencies)
+    let initialState = DependencyValues.$_current.withValue(dependencies) { initialState() }
+
     let reducer = TestReducer(Reduce(reducer), initialState: initialState)
     self._environment = .init(wrappedValue: ())
     self.file = file
@@ -589,6 +595,7 @@ open class TestStore<State, Action, ScopedState, ScopedAction, Environment> {
     self.store = Store(initialState: initialState, reducer: reducer)
     self.timeout = 100 * NSEC_PER_MSEC
     self.toScopedState = { $0 }
+    self.dependencies = dependencies
   }
 
   @available(
@@ -1037,7 +1044,9 @@ extension TestStore where ScopedState: Equatable {
     case .on:
       var expectedWhenGivenPreviousState = expected
       if let updateStateToExpectedResult = updateStateToExpectedResult {
-        try updateStateToExpectedResult(&expectedWhenGivenPreviousState)
+        try DependencyValues.$_current.withValue(self.dependencies) {
+          try updateStateToExpectedResult(&expectedWhenGivenPreviousState)
+        }
       }
       expected = expectedWhenGivenPreviousState
 
@@ -1050,7 +1059,9 @@ extension TestStore where ScopedState: Equatable {
     case .off:
       var expectedWhenGivenActualState = actual
       if let updateStateToExpectedResult = updateStateToExpectedResult {
-        try updateStateToExpectedResult(&expectedWhenGivenActualState)
+        try DependencyValues.$_current.withValue(self.dependencies) {
+          try updateStateToExpectedResult(&expectedWhenGivenActualState)
+        }
       }
       expected = expectedWhenGivenActualState
 
@@ -1062,10 +1073,12 @@ extension TestStore where ScopedState: Equatable {
         && expectedWhenGivenActualState == actual
       {
         var expectedWhenGivenPreviousState = current
-        if let modify = updateStateToExpectedResult {
+        if let updateStateToExpectedResult = updateStateToExpectedResult {
           _XCTExpectFailure(strict: false) {
             do {
-              try modify(&expectedWhenGivenPreviousState)
+              try DependencyValues.$_current.withValue(self.dependencies) {
+                try updateStateToExpectedResult(&expectedWhenGivenPreviousState)
+              }
             } catch {
               XCTFail(
                 """
@@ -2063,11 +2076,7 @@ public struct TestStoreTask: Hashable, Sendable {
 
 class TestReducer<State, Action>: ReducerProtocol {
   let base: Reduce<State, Action>
-  var dependencies = { () -> DependencyValues in
-    var dependencies = DependencyValues()
-    dependencies.context = .test
-    return dependencies
-  }()
+  var dependencies = DependencyValues()
   let effectDidSubscribe = AsyncStream<Void>.streamWithContinuation()
   var inFlightEffects: Set<LongLivingEffect> = []
   var receivedActions: [(action: Action, state: State)] = []