Fix flakey test

app/modules/services/CodeCompletionService/Tests/FileWatcherTests.swift

@@ -34,6 +34,7 @@ struct FileWatcherTests {
       directories: [workspace1])
 
     let setupExpectation = expectation(description: "setUp called")
+    let didOpenExpectation = expectation(description: "didOpen called")
     let didSaveExpectation = expectation(description: "didSave called")
 
     let didSaveCalls = Atomic<[(workspace: any Workspace, file: URL, content: String, version: Int)]>([])
@@ -42,6 +43,10 @@ struct FileWatcherTests {
       setupExpectation.fulfill()
     }
 
+    mockCodeCompletionProvider.onDidOpen = { _, _, _, _ in
+      didOpenExpectation.fulfill()
+    }
+
     mockCodeCompletionProvider.onDidSave = { workspace, file, content, version in
       didSaveCalls.mutate { $0.append((workspace, file, content, version)) }
       didSaveExpectation.fulfill()
@@ -66,8 +71,7 @@ struct FileWatcherTests {
     let state1 = createXcodeState(workspace: workspace1, file: file1.path, content: "let x = 1")
     mockXcodeObserver.mutableStatePublisher.send(state1)
 
-    // Wait for initialization
-    try await fulfillment(of: setupExpectation)
+    try await fulfillment(of: [setupExpectation, didOpenExpectation])
 
     // Simulate file save by changing content on disk and triggering watcher
     try mockFileManager.write(string: "let x = 2", to: file1, options: [])

local-server/build.sha256

@@ -1 +1 @@
-cd7571cf4efa03f16b8e1437fd3495904110c26524e2826854b4ae969a2a3d2c
+d70ed262e28022f81a0e676f6a9a394a0eb67835cbd7824da548fd14a0fd33fc