refactor(test-utils): update startLanguageServer to accept multiple w…

…orkspace folders Closes #229
volarjs · Aug 18, 2024 · 8af381d · 8af381d
1 parent 9b7d91e
commit 8af381d
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/packages/test-utils/index.ts b/packages/test-utils/index.ts
@@ -56,7 +56,7 @@ export function startLanguageServer(serverModule: string, cwd?: string | URL) {
 		process: childProcess,
 		connection,
 		async initialize(
-			rootUri: string,
+			rootUri: string | _.WorkspaceFolder[],
 			initializationOptions: _._InitializeParams['initializationOptions'],
 			capabilities: _.ClientCapabilities = {},
 			locale?: string
@@ -65,16 +65,14 @@ export function startLanguageServer(serverModule: string, cwd?: string | URL) {
 				_.InitializeRequest.type,
 				{
 					processId: childProcess.pid ?? null,
-					rootUri,
+					rootUri: typeof rootUri === 'string' ? rootUri : null,
+					workspaceFolders: Array.isArray(rootUri) ? rootUri : null,
 					initializationOptions,
 					capabilities,
 					locale,
 				} satisfies _.InitializeParams
 			);
-			await connection.sendNotification(
-				_.InitializedNotification.type,
-				{} satisfies _.InitializedParams
-			);
+			await connection.sendNotification(_.InitializedNotification.type, {} satisfies _.InitializedParams);
 			running = true;
 			return result;
 		},