lsp/testing: Ensure test locations sent at boot

This was missed in #1888

This makes test locations more similar to diagnostics.

Signed-off-by: Charlie Egan <charlie_egan@apple.com>

Author: charlieegan3

internal/lsp/server.go

@@ -2150,12 +2150,23 @@ func (l *LanguageServer) loadWorkspaceContents(ctx context.Context, newOnly bool
 			return nil
 		}
 
-		if _, err = updateParse(ctx, l.parseOpts(fileURI, l.builtinsForCurrentCapabilities())); err != nil {
+		parseSuccess, err := updateParse(ctx, l.parseOpts(fileURI, l.builtinsForCurrentCapabilities()))
+		if err != nil {
 			failed = append(failed, fileLoadFailure{URI: fileURI, Error: fmt.Errorf("failed to update parse: %w", err)})
 
 			return nil // continue processing other files
 		}
 
+		if parseSuccess {
+			l.testLocationJobs <- lintFileJob{Reason: "server initialized", URI: fileURI}
+		} else {
+			// this is covering the case where the client starts and the state
+			// is different (the client remembers test locations and state).
+			if err := l.sendTestLocations(ctx, fileURI, []any{}); err != nil {
+				l.log.Message("failed to send empty test locations after parse failure: %s", err)
+			}
+		}
+
 		changedOrNewURIs = append(changedOrNewURIs, fileURI)
 
 		return nil
@@ -2221,6 +2232,12 @@ func (l *LanguageServer) handleWorkspaceDidChangeWatchedFiles(
 }
 
 func (l *LanguageServer) sendFileDiagnostics(ctx context.Context, fileURI string) {
+	if l.conn == nil {
+		l.log.Debug("sendFileDiagnostics called with no connection: %s", fileURI)
+
+		return
+	}
+
 	// first, set the diagnostics for the file to the current parse errors
 	fileDiags, _ := l.cache.GetParseErrors(fileURI)
 	if len(fileDiags) == 0 {

internal/lsp/testing.go

@@ -78,6 +78,12 @@ func (l *LanguageServer) processTestLocationsUpdate(ctx context.Context, fileURI
 }
 
 func (l *LanguageServer) sendTestLocations(ctx context.Context, fileURI string, locations any) error {
+	if l.conn == nil {
+		l.log.Debug("sendTestLocations called with no connection: %s", fileURI)
+
+		return nil
+	}
+
 	params := map[string]any{
 		"uri":       fileURI,
 		"locations": locations,