1 files changed, 97 insertions, 0 deletions
diff --git a/packages/lsp/src/tool.test.ts b/packages/lsp/src/tool.test.ts
index 03787ae..efd1514 100644
--- a/packages/lsp/src/tool.test.ts
+++ b/packages/lsp/src/tool.test.ts
@@ -176,4 +176,101 @@ describe("tool", () => {
 		expect(result.isError).toBe(true);
 		expect(result.content).toContain("requires both");
 	});
+
+	it("diagnostics op: a server that times out is skipped with a raise-to-user notice", async () => {
+		const mockClient = {
+			getState: () => "connected" as const,
+			getStateError: () => undefined,
+			request: async () => null,
+			waitForDiagnostics: async () => ({ formatted: "", slow: false, timedOut: true }),
+		};
+
+		const tool = createLspTool(
+			stubManager({
+				status: async () => [
+					{
+						id: "steep",
+						name: "Steep",
+						root: "/project",
+						extensions: [".rb"],
+						state: "connected",
+					},
+				],
+				getClient: () => mockClient as never,
+			}),
+		);
+
+		const result = await tool.execute(
+			{ operation: "diagnostics", path: "game.rb" },
+			{
+				toolCallId: "test",
+				onOutput: () => {},
+				signal: AbortSignal.timeout(5000),
+				log: {
+					debug: () => {},
+					info: () => {},
+					warn: () => {},
+					error: () => {},
+					child: () => ({}) as never,
+					span: () => ({}) as never,
+				},
+				cwd: "/project",
+			},
+		);
+
+		expect(result.isError).not.toBe(true);
+		expect(result.content).toContain("[Steep]");
+		expect(result.content).toContain("took too long");
+		expect(result.content).toContain(">10s");
+		expect(result.content).toContain("raise this to the user");
+	});
+
+	it("diagnostics op: responding servers' diagnostics are merged, tagged by source", async () => {
+		const mockClient = {
+			getState: () => "connected" as const,
+			getStateError: () => undefined,
+			request: async () => null,
+			waitForDiagnostics: async () => ({
+				formatted: "ERROR L1:1: boom",
+				slow: false,
+				timedOut: false,
+			}),
+		};
+
+		const tool = createLspTool(
+			stubManager({
+				status: async () => [
+					{
+						id: "steep",
+						name: "Steep",
+						root: "/project",
+						extensions: [".rb"],
+						state: "connected",
+					},
+				],
+				getClient: () => mockClient as never,
+			}),
+		);
+
+		const result = await tool.execute(
+			{ operation: "diagnostics", path: "game.rb" },
+			{
+				toolCallId: "test",
+				onOutput: () => {},
+				signal: AbortSignal.timeout(5000),
+				log: {
+					debug: () => {},
+					info: () => {},
+					warn: () => {},
+					error: () => {},
+					child: () => ({}) as never,
+					span: () => ({}) as never,
+				},
+				cwd: "/project",
+			},
+		);
+
+		expect(result.content).toContain("[Steep]");
+		expect(result.content).toContain("boom");
+	});
 });