Fix flakey test (#1364)

* Fix flakey test

* Try something more general

(cherry picked from commit d4834b56e92f347ac6c8b9908d415fd51c3d18f4)

# Conflicts:
#	Sources/ComposableArchitecture/TestStore.swift

Author: stephencelis

Sources/ComposableArchitecture/TestStore.swift

@@ -272,9 +272,12 @@
             effects
             .producer
             .on(
-              starting: { [weak self] in
+              starting: { [effectDidSubscribe = self.effectDidSubscribe, weak self] in
                 self?.inFlightEffects.insert(effect)
-                self?.effectDidSubscribe.continuation.yield()
+                Task {
+                  await Task.megaYield()
+                  effectDidSubscribe.continuation.yield()
+                }
               },
               completed: { [weak self] in self?.inFlightEffects.remove(effect) },
               disposed: { [weak self] in self?.inFlightEffects.remove(effect) }

Tests/ComposableArchitectureTests/ComposableArchitectureTests.swift

@@ -114,11 +114,7 @@ final class ComposableArchitectureTests: XCTestCase {
       case response(Int)
     }
 
-    struct Environment {
-      let fetch: (Int) async -> Int
-    }
-
-    let reducer = Reducer<Int, Action, Environment> { state, action, environment in
+    let reducer = Reducer<Int, Action, Void> { state, action, _ in
       enum CancelID {}
 
       switch action {
@@ -129,7 +125,7 @@ final class ComposableArchitectureTests: XCTestCase {
         state += 1
         return .task { [state] in
           try await mainQueue.sleep(for: .seconds(1))
-          return .response(await environment.fetch(state))
+          return .response(state * state)
         }
         .cancellable(id: CancelID.self)
 
@@ -142,9 +138,7 @@ final class ComposableArchitectureTests: XCTestCase {
     let store = TestStore(
       initialState: 0,
       reducer: reducer,
-      environment: Environment(
-        fetch: { value in value * value }
-      )
+      environment: ()
     )
 
     await store.send(.incr) { $0 = 1 }