add unit tests

Author: nirinchev

src/common/config.ts

@@ -11,7 +11,7 @@ import { z } from "zod";
 const levenshtein = levenshteinModule.default;
 
 const previewFeatures = z.enum(["vectorSearch"]);
-export type PreviewFeatures = z.infer<typeof previewFeatures>;
+export type PreviewFeature = z.infer<typeof previewFeatures>;
 
 // From: https://github.com/mongodb-js/mongosh/blob/main/packages/cli-repl/src/arg-parser.ts
 export const OPTIONS = {

src/tools/tool.ts

@@ -6,7 +6,7 @@ import type { Session } from "../common/session.js";
 import { LogId } from "../common/logger.js";
 import type { Telemetry } from "../telemetry/telemetry.js";
 import { type ToolEvent } from "../telemetry/types.js";
-import type { PreviewFeatures, UserConfig } from "../common/config.js";
+import type { PreviewFeature, UserConfig } from "../common/config.js";
 import type { Server } from "../server.js";
 import type { Elicitation } from "../elicitation.js";
 
@@ -321,8 +321,8 @@ export abstract class ToolBase {
         this.telemetry.emitEvents([event]);
     }
 
-    protected isFeatureEnabled(flag: PreviewFeatures): boolean {
-        return this.config.previewFeatures.includes(flag);
+    protected isFeatureEnabled(feature: PreviewFeature): boolean {
+        return this.config.previewFeatures.includes(feature);
     }
 }
 

tests/unit/toolBase.test.ts

@@ -3,7 +3,7 @@ import { z } from "zod";
 import { ToolBase, type OperationType, type ToolCategory, type ToolConstructorParams } from "../../src/tools/tool.js";
 import type { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
 import type { Session } from "../../src/common/session.js";
-import type { UserConfig } from "../../src/common/config.js";
+import type { PreviewFeature, UserConfig } from "../../src/common/config.js";
 import type { Telemetry } from "../../src/telemetry/telemetry.js";
 import type { Elicitation } from "../../src/elicitation.js";
 import type { CompositeLogger } from "../../src/common/logger.js";
@@ -32,6 +32,7 @@ describe("ToolBase", () => {
 
         mockConfig = {
             confirmationRequiredTools: [],
+            previewFeatures: [],
         } as unknown as UserConfig;
 
         mockTelemetry = {} as Telemetry;
@@ -100,6 +101,21 @@ describe("ToolBase", () => {
             expect(mockRequestConfirmation).toHaveBeenCalledTimes(1);
         });
     });
+
+    describe("isFeatureEnabled", () => {
+        it("should return false for any feature by default", () => {
+            expect(testTool["isFeatureEnabled"]("vectorSearch")).to.equal(false);
+            expect(testTool["isFeatureEnabled"]("someOtherFeature" as PreviewFeature)).to.equal(false);
+        });
+
+        it("should return true for enabled features", () => {
+            mockConfig.previewFeatures = ["vectorSearch", "someOtherFeature" as PreviewFeature];
+            expect(testTool["isFeatureEnabled"]("vectorSearch")).to.equal(true);
+            expect(testTool["isFeatureEnabled"]("someOtherFeature" as PreviewFeature)).to.equal(true);
+
+            expect(testTool["isFeatureEnabled"]("anotherFeature" as PreviewFeature)).to.equal(false);
+        });
+    });
 });
 
 class TestTool extends ToolBase {