chore: add accuracy tests for drop-index tool

Author: himanshusinghs

tests/accuracy/dropIndex.test.ts

@@ -0,0 +1,74 @@
+import type { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
+import { describeAccuracyTests } from "./sdk/describeAccuracyTests.js";
+import { Matcher } from "./sdk/matcher.js";
+
+// We don't want to delete actual indexes
+const mockedTools = {
+    "drop-index": ({ indexName, database, collection }: Record<string, unknown>): CallToolResult => {
+        return {
+            content: [
+                {
+                    text: `Successfully dropped the index with name "${String(indexName)}" from the provided namespace "${String(database)}.${String(collection)}".`,
+                    type: "text",
+                },
+            ],
+        };
+    },
+} as const;
+
+describeAccuracyTests([
+    {
+        prompt: "Delete the index called year_1 from mflix.movies namespace",
+        expectedToolCalls: [
+            {
+                toolName: "drop-index",
+                parameters: {
+                    database: "mflix",
+                    collection: "movies",
+                    indexName: "year_1",
+                },
+            },
+        ],
+        mockedTools,
+    },
+    {
+        prompt: "First create a text index on field 'title' in 'mflix.movies' namespace and then drop all the indexes from 'mflix.movies' namespace",
+        expectedToolCalls: [
+            {
+                toolName: "create-index",
+                parameters: {
+                    database: "mflix",
+                    collection: "movies",
+                    name: Matcher.anyOf(Matcher.undefined, Matcher.string()),
+                    keys: {
+                        title: "text",
+                    },
+                },
+            },
+            {
+                toolName: "collection-indexes",
+                parameters: {
+                    database: "mflix",
+                    collection: "movies",
+                },
+            },
+            {
+                toolName: "drop-index",
+                parameters: {
+                    database: "mflix",
+                    collection: "movies",
+                    indexName: Matcher.string(),
+                },
+            },
+            {
+                toolName: "drop-index",
+                parameters: {
+                    database: "mflix",
+                    collection: "movies",
+                    indexName: Matcher.string(),
+                },
+            },
+        ],
+        mockedTools,
+    },
+]);

tests/accuracy/sdk/accuracyTestingClient.ts

@@ -7,7 +7,7 @@ import { MCP_SERVER_CLI_SCRIPT } from "./constants.js";
 import type { LLMToolCall } from "./accuracyResultStorage/resultStorage.js";
 import type { VercelMCPClient, VercelMCPClientTools } from "./agent.js";
 
-type ToolResultGeneratorFn = (...parameters: unknown[]) => CallToolResult | Promise<CallToolResult>;
+type ToolResultGeneratorFn = (parameters: Record<string, unknown>) => CallToolResult | Promise<CallToolResult>;
 export type MockedTools = Record<string, ToolResultGeneratorFn>;
 
 /**
@@ -44,7 +44,7 @@ export class AccuracyTestingClient {
                     try {
                         const toolResultGeneratorFn = this.mockedTools[toolName];
                         if (toolResultGeneratorFn) {
-                            return await toolResultGeneratorFn(args);
+                            return await toolResultGeneratorFn(args as Record<string, unknown>);
                         }
 
                         return await tool.execute(args, options);