chore: tests for delete-many tool

Author: himanshusinghs

src/tools/mongodb/delete/deleteMany.ts

@@ -4,6 +4,17 @@ import { DbOperationArgs, MongoDBToolBase } from "../mongodbTool.js";
 import { ToolArgs, OperationType } from "../../tool.js";
 import { checkIndexUsage } from "../../../helpers/indexCheck.js";
 
+export function deleteManyResponse(collection: string, delectedCount: number): CallToolResult {
+    return {
+        content: [
+            {
+                text: `Deleted \`${delectedCount}\` document(s) from collection "${collection}"`,
+                type: "text",
+            },
+        ],
+    };
+}
+
 export class DeleteManyTool extends MongoDBToolBase {
     protected name = "delete-many";
     protected description = "Removes all documents that match the filter from a MongoDB collection";
@@ -45,13 +56,6 @@ export class DeleteManyTool extends MongoDBToolBase {
 
         const result = await provider.deleteMany(database, collection, filter);
 
-        return {
-            content: [
-                {
-                    text: `Deleted \`${result.deletedCount}\` document(s) from collection "${collection}"`,
-                    type: "text",
-                },
-            ],
-        };
+        return deleteManyResponse(collection, result.deletedCount);
     }
 }

tests/accuracy/delete-many.test.ts

@@ -0,0 +1,53 @@
+import { describeAccuracyTests } from "./sdk/describe-accuracy-tests.js";
+import { getAvailableModels } from "./sdk/models.js";
+import { AccuracyTestConfig } from "./sdk/describe-accuracy-tests.js";
+import { deleteManyResponse } from "../../src/tools/mongodb/delete/deleteMany.js";
+
+function callsDeleteManyWithEmptyFilters(prompt: string): AccuracyTestConfig {
+    return {
+        injectConnectedAssumption: true,
+        prompt: prompt,
+        mockedTools: {
+            "delete-many": function listDatabases() {
+                return deleteManyResponse("coll1", 10);
+            },
+        },
+        expectedToolCalls: [
+            {
+                toolName: "delete-many",
+                parameters: {
+                    database: "db1",
+                    collection: "coll1",
+                },
+            },
+        ],
+    };
+}
+
+function callsDeleteManyWithFilters(prompt: string): AccuracyTestConfig {
+    return {
+        injectConnectedAssumption: true,
+        prompt: prompt,
+        mockedTools: {
+            "delete-many": function listDatabases() {
+                return deleteManyResponse("coll1", 10);
+            },
+        },
+        expectedToolCalls: [
+            {
+                toolName: "delete-many",
+                parameters: {
+                    database: "db1",
+                    collection: "coll1",
+                    filters: { provider: "BongoDB" },
+                },
+            },
+        ],
+    };
+}
+
+describeAccuracyTests("delete-many", getAvailableModels(), [
+    callsDeleteManyWithEmptyFilters("Delete all the documents from 'db1.coll1' namespace"),
+    callsDeleteManyWithEmptyFilters("Purge the collection 'coll1' in database 'db1'"),
+    callsDeleteManyWithFilters("Remove all the documents from namespace 'db1.coll1' where provider is 'BongoDB'"),
+]);