Fix prepareDependencies when using TestDependencyKey. (#315)

* Fix prepareDependencies when using TestDependencyKey. * wip * wip
pointfreeco · Dec 3, 2024 · 5526c8a · 5526c8a
1 parent 7d2eb4a
commit 5526c8a
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 1 deletion.
diff --git a/Sources/Dependencies/DependencyValues.swift b/Sources/Dependencies/DependencyValues.swift
@@ -477,7 +477,12 @@ public final class CachedValues: @unchecked Sendable {
     return withIssueContext(fileID: fileID, filePath: filePath, line: line, column: column) {
       let cacheKey = CacheKey(id: TypeIdentifier(key), context: context)
       #if DEBUG
-        if context == .live, !DependencyValues.isSetting, !(key is any DependencyKey.Type) {
+        if
+          context == .live,
+          !DependencyValues.isSetting,
+          !(cached[cacheKey] != nil && cached[cacheKey]?.preparationID != nil),
+          !(key is any DependencyKey.Type)
+        {
           reportIssue(
             {
               var dependencyDescription = ""

diff --git a/Tests/DependenciesTests/DependencyValuesTests.swift b/Tests/DependenciesTests/DependencyValuesTests.swift
@@ -736,6 +736,18 @@ final class DependencyValuesTests: XCTestCase {
     XCTAssertEqual(now, Date(timeIntervalSinceReferenceDate: 0))
   }
 
+  func testPrepareDependencies_setsDependency_LiveContext() {
+    withDependencies {
+      $0.context = .live
+    } operation: {
+      prepareDependencies {
+        $0[ClientWithEndpoint.self] = ClientWithEndpoint(get: { 1729 })
+      }
+      @Dependency(ClientWithEndpoint.self) var client
+      XCTAssertEqual(client.get(), 1729)
+    }
+  }
+
   #if DEBUG && !os(Linux) && !os(WASI) && !os(Windows)
     func testPrepareDependencies_MultiplePreparesWithNoAccessBetween() {
       prepareDependencies {