fix: Finish the continuation and await the consumtion of the stream

Author: tom-ludwig

CodeEdit/Features/Documents/WorkspaceDocument/WorkspaceDocument+Index.swift

@@ -25,8 +25,11 @@ extension WorkspaceDocument.SearchState {
         }
 
         let (progressStream, continuation) = AsyncStream<Double>.makeStream()
+        
         // Dispatch this now, we want to continue after starting to monitor
-        Task { await self.monitorProgressStream(progressStream, activityId: activity.id) }
+        let monitorTask = Task {
+            await self.monitorProgressStream(progressStream, activityId: activity.id)
+        }
 
         Task.detached {
             let filePaths = self.getFileURLs(at: url)
@@ -53,6 +56,8 @@ extension WorkspaceDocument.SearchState {
                 }
             }
 
+            continuation.finish()
+            await monitorTask.value // Await monitor to finish draining the stream
             asyncController.index.flush()
 
             await MainActor.run {

CodeEditTests/Features/Documents/WorkspaceDocument+SearchState+FindAndReplaceTests.swift

@@ -99,9 +99,7 @@ final class FindAndReplaceTests: XCTestCase {
         // IMPORTANT:
         // This is only a temporary solution, in the feature a file watcher would track the file update
         // and trigger a index update.
-        Task {
-            await searchState.addProjectToIndex()
-        }
+        await searchState.addProjectToIndex()
         let startTime = Date()
         while searchState.indexStatus != .done {
             try? await Task.sleep(nanoseconds: 100_000_000)