Skip to content

Commit 8975982

Browse files
committed
fix tabs
1 parent 2223eb7 commit 8975982

File tree

2 files changed

+74
-74
lines changed

2 files changed

+74
-74
lines changed

src/server/mcp.test.ts

Lines changed: 62 additions & 62 deletions
Original file line numberDiff line numberDiff line change
@@ -1966,74 +1966,74 @@ describe.each(zodTestMatrix)('$zodVersionLabel', (entry: ZodMatrixEntry) => {
19661966
* Test: Tool disable, enable, and remove via tool instance
19671967
*/
19681968
test("should manage tool when using tool instance", async () => {
1969-
const mcpServer = new McpServer({
1970-
name: "test server",
1971-
version: "1.0",
1972-
});
1973-
1974-
// Register initial tool
1975-
const tool = mcpServer.tool("test", async () => ({
1976-
content: [
1977-
{
1978-
type: "text",
1979-
text: "Test response",
1980-
},
1981-
],
1982-
}));
1983-
1984-
expect(mcpServer['_registeredTools'].test).toBeDefined();
1985-
1986-
// Now disable the tool
1987-
tool.disable();
1988-
1989-
expect(mcpServer['_registeredTools'].test.enabled).toBe(false);
1990-
1991-
// Now enable the tool
1992-
tool.enable();
1993-
1994-
expect(mcpServer['_registeredTools'].test.enabled).toBe(true);
1995-
1996-
// Now delete the tool
1997-
tool.remove();
1998-
1999-
expect(mcpServer['_registeredTools'].test).toBeUndefined();
1969+
const mcpServer = new McpServer({
1970+
name: "test server",
1971+
version: "1.0",
1972+
});
1973+
1974+
// Register initial tool
1975+
const tool = mcpServer.tool('test', async () => ({
1976+
content: [
1977+
{
1978+
type: 'text',
1979+
text: 'Test response'
1980+
}
1981+
]
1982+
}));
1983+
1984+
expect(mcpServer['_registeredTools'].test).toBeDefined();
1985+
1986+
// Now disable the tool
1987+
tool.disable();
1988+
1989+
expect(mcpServer['_registeredTools'].test.enabled).toBe(false);
1990+
1991+
// Now enable the tool
1992+
tool.enable();
1993+
1994+
expect(mcpServer['_registeredTools'].test.enabled).toBe(true);
1995+
1996+
// Now delete the tool
1997+
tool.remove();
1998+
1999+
expect(mcpServer['_registeredTools'].test).toBeUndefined();
20002000
});
20012001

20022002
/***
20032003
* Test: Tool disable, enable, and remove via server instance
20042004
*/
20052005
test("should manage tool when using server instance", async () => {
2006-
const mcpServer = new McpServer({
2007-
name: "test server",
2008-
version: "1.0",
2009-
});
2010-
2011-
// Register initial tool
2012-
mcpServer.tool("test", async () => ({
2013-
content: [
2014-
{
2015-
type: "text",
2016-
text: "Test response",
2017-
},
2018-
],
2019-
}));
2020-
2021-
expect(mcpServer['_registeredTools'].test).toBeDefined();
2022-
2023-
// Now disable the tool
2024-
mcpServer.disableTool("test");
2025-
2026-
expect(mcpServer['_registeredTools'].test.enabled).toBe(false);
2027-
2028-
// Now enable the tool
2029-
mcpServer.enableTool("test");
2030-
2031-
expect(mcpServer['_registeredTools'].test.enabled).toBe(true);
2032-
2033-
// Now delete the tool
2034-
mcpServer.removeTool("test");
2035-
2036-
expect(mcpServer['_registeredTools'].test).toBeUndefined();
2006+
const mcpServer = new McpServer({
2007+
name: "test server",
2008+
version: "1.0",
2009+
});
2010+
2011+
// Register initial tool
2012+
mcpServer.tool('test', async () => ({
2013+
content: [
2014+
{
2015+
type: 'text',
2016+
text: 'Test response'
2017+
}
2018+
]
2019+
}));
2020+
2021+
expect(mcpServer['_registeredTools'].test).toBeDefined();
2022+
2023+
// Now disable the tool
2024+
mcpServer.disableTool("test");
2025+
2026+
expect(mcpServer['_registeredTools'].test.enabled).toBe(false);
2027+
2028+
// Now enable the tool
2029+
mcpServer.enableTool("test");
2030+
2031+
expect(mcpServer['_registeredTools'].test.enabled).toBe(true);
2032+
2033+
// Now delete the tool
2034+
mcpServer.removeTool("test");
2035+
2036+
expect(mcpServer['_registeredTools'].test).toBeUndefined();
20372037
});
20382038
});
20392039

src/server/mcp.ts

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1066,32 +1066,32 @@ export class McpServer {
10661066
* Does nothing if the tool is not registered.
10671067
*/
10681068
enableTool(name: string) {
1069-
const tool = this._registeredTools[name];
1070-
if (tool) {
1071-
tool.enable();
1072-
}
1069+
const tool = this._registeredTools[name];
1070+
if (tool) {
1071+
tool.enable();
1072+
}
10731073
};
10741074

10751075
/**
10761076
* Disables a tool from the server by name.
10771077
* Does nothing if the tool is not registered.
10781078
*/
10791079
disableTool(name: string) {
1080-
const tool = this._registeredTools[name];
1081-
if (tool) {
1082-
tool.disable();
1083-
}
1080+
const tool = this._registeredTools[name];
1081+
if (tool) {
1082+
tool.disable();
1083+
}
10841084
};
10851085

10861086
/**
10871087
* Updates a tool from the server by name.
10881088
* Does nothing if the tool is not registered.
10891089
*/
10901090
updateTool<InputArgs extends ZodRawShapeCompat, OutputArgs extends ZodRawShapeCompat>(name: string, updates: ToolUpdates<InputArgs, OutputArgs>) {
1091-
const tool = this._registeredTools[name];
1092-
if (tool) {
1093-
tool.update(updates);
1094-
}
1091+
const tool = this._registeredTools[name];
1092+
if (tool) {
1093+
tool.update(updates);
1094+
}
10951095
};
10961096

10971097
/**

0 commit comments

Comments
 (0)