chore: tests for insert-many tool

Author: himanshusinghs

src/tools/mongodb/create/insertMany.ts

@@ -3,6 +3,21 @@ import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
 import { DbOperationArgs, MongoDBToolBase } from "../mongodbTool.js";
 import { ToolArgs, OperationType } from "../../tool.js";
 
+export function insertManyResponse(collection: string, insertedCount: number, insertedIds: unknown[]): CallToolResult {
+    return {
+        content: [
+            {
+                text: `Inserted \`${insertedCount}\` document(s) into collection "${collection}"`,
+                type: "text",
+            },
+            {
+                text: `Inserted IDs: ${insertedIds.join(", ")}`,
+                type: "text",
+            },
+        ],
+    };
+}
+
 export class InsertManyTool extends MongoDBToolBase {
     protected name = "insert-many";
     protected description = "Insert an array of documents into a MongoDB collection";
@@ -24,17 +39,6 @@ export class InsertManyTool extends MongoDBToolBase {
         const provider = await this.ensureConnected();
         const result = await provider.insertMany(database, collection, documents);
 
-        return {
-            content: [
-                {
-                    text: `Inserted \`${result.insertedCount}\` document(s) into collection "${collection}"`,
-                    type: "text",
-                },
-                {
-                    text: `Inserted IDs: ${Object.values(result.insertedIds).join(", ")}`,
-                    type: "text",
-                },
-            ],
-        };
+        return insertManyResponse(collection, result.insertedCount, Object.values(result.insertedIds));
     }
 }

tests/accuracy/insert-many.test.ts

@@ -0,0 +1,72 @@
+import { describeAccuracyTests } from "./sdk/describe-accuracy-tests.js";
+import { getAvailableModels } from "./sdk/models.js";
+import { AccuracyTestConfig } from "./sdk/describe-accuracy-tests.js";
+import { insertManyResponse } from "../../src/tools/mongodb/create/insertMany.js";
+
+function callsInsertMany(prompt: string): AccuracyTestConfig {
+    return {
+        injectConnectedAssumption: true,
+        prompt: prompt,
+        mockedTools: {
+            "insert-many": function listDatabases() {
+                return insertManyResponse("coll1", 3, ["1FOO", "2BAR", "3BAZ"]);
+            },
+        },
+        expectedToolCalls: [
+            {
+                toolName: "insert-many",
+                parameters: {
+                    database: "db1",
+                    collection: "coll1",
+                    documents: [
+                        {
+                            id: 1,
+                            name: "name1",
+                        },
+                        {
+                            id: 2,
+                            name: "name2",
+                        },
+                        {
+                            id: 3,
+                            name: "name3",
+                        },
+                    ],
+                },
+            },
+        ],
+    };
+}
+
+function callsEmptyInsertMany(prompt: string) {
+    return {
+        injectConnectedAssumption: true,
+        prompt: prompt,
+        mockedTools: {
+            "insert-many": function listDatabases() {
+                return insertManyResponse("coll1", 3, ["1FOO", "2BAR", "3BAZ"]);
+            },
+        },
+        expectedToolCalls: [
+            {
+                toolName: "insert-many",
+                parameters: {
+                    database: "db1",
+                    collection: "coll1",
+                    documents: [{}, {}, {}],
+                },
+            },
+        ],
+    };
+}
+
+describeAccuracyTests("insert-many", getAvailableModels(), [
+    callsInsertMany(
+        [
+            "In my namespace 'db1.coll1', insert 3 documents each with the following fields:",
+            "- id: an incremental number starting from 1",
+            "- name: a string of format 'name<id>'",
+        ].join("\n")
+    ),
+    callsEmptyInsertMany("Add three empty documents in collection 'coll1' inside database 'db1'"),
+]);