fix tests

Author: kentcdodds

exercises/02.tools/01.problem.simple/src/index.test.ts

@@ -6,14 +6,18 @@ import {
 import { StdioClientTransport } from '@modelcontextprotocol/sdk/client/stdio.js'
 import { test, expect } from 'vitest'
 
-async function setupClient() {
-	const client = new Client({
-		name: 'EpicMeTester',
-		version: '1.0.0',
-	})
+async function setupClient({ capabilities }: ClientOptions = {}) {
+	const client = new Client(
+		{
+			name: 'EpicMeTester',
+			version: '1.0.0',
+		},
+		{ capabilities },
+	)
 	const transport = new StdioClientTransport({
 		command: 'tsx',
 		args: ['src/index.ts'],
+		stderr: 'ignore',
 	})
 	await client.connect(transport)
 	return {

exercises/03.resources/05.problem.linked/src/index.test.ts

@@ -26,19 +26,19 @@ async function setupClient({ capabilities }: ClientOptions = {}) {
 	const transport = new StdioClientTransport({
 		command: 'tsx',
 		args: ['src/index.ts'],
+		stderr: 'ignore',
 		env: {
 			...process.env,
 			EPIC_ME_DB_PATH,
 		},
-		stderr: 'ignore',
 	})
 	await client.connect(transport)
 	return {
 		client,
 		EPIC_ME_DB_PATH,
 		async [Symbol.asyncDispose]() {
 			await client.transport?.close()
-			await fs.unlink(EPIC_ME_DB_PATH)
+			await fs.unlink(EPIC_ME_DB_PATH).catch(() => {})
 		},
 	}
 }

exercises/04.prompts/03.problem.completion/src/index.test.ts

@@ -162,7 +162,7 @@ test('Prompts List', async () => {
 })
 
 test('Prompt Argument Completion', async () => {
-	await using setup = await setupClient()
+	await using setup = await setupClient({ capabilities: { completion: true } })
 	const { client } = setup
 	// First create some entries to have data for completion
 	await client.callTool({
@@ -200,9 +200,9 @@ test('Prompt Argument Completion', async () => {
 		invariant(firstArg, '🚨 First prompt argument should exist')
 
 		// Test completion functionality using the proper MCP SDK method
-		const completionResult = await (client as any).completePrompt({
+		const completionResult = await client.complete({
 			ref: {
-				type: 'prompt',
+				type: 'ref/prompt',
 				name: firstPrompt.name,
 			},
 			argument: {