@@ -23,16 +23,6 @@ vi.mock("@roo-code/core", () => ({
2323 } ,
2424} ) )
2525
26- vi . mock ( "@roo-code/telemetry" , ( ) => ( {
27- TelemetryService : {
28- instance : {
29- captureToolUsage : vi . fn ( ) ,
30- captureConsecutiveMistakeError : vi . fn ( ) ,
31- } ,
32- } ,
33- } ) )
34-
35- import { TelemetryService } from "@roo-code/telemetry"
3626import { customToolRegistry } from "@roo-code/core"
3727
3828describe ( "presentAssistantMessage - Custom Tool Recording" , ( ) => {
@@ -118,7 +108,6 @@ describe("presentAssistantMessage - Custom Tool Recording", () => {
118108
119109 // Should record as "custom_tool", not "my_custom_tool"
120110 expect ( mockTask . recordToolUsage ) . toHaveBeenCalledWith ( "custom_tool" )
121- expect ( TelemetryService . instance . captureToolUsage ) . toHaveBeenCalledWith ( mockTask . taskId , "custom_tool" )
122111 } )
123112 } )
124113
@@ -171,7 +160,6 @@ describe("presentAssistantMessage - Custom Tool Recording", () => {
171160
172161 // Should record as "read_file", not "custom_tool"
173162 expect ( mockTask . recordToolUsage ) . toHaveBeenCalledWith ( "read_file" )
174- expect ( TelemetryService . instance . captureToolUsage ) . toHaveBeenCalledWith ( mockTask . taskId , "read_file" )
175163 } )
176164
177165 it ( "should record MCP tool usage as 'use_mcp_tool' (not custom_tool)" , async ( ) => {
@@ -213,7 +201,6 @@ describe("presentAssistantMessage - Custom Tool Recording", () => {
213201
214202 // Should record as "use_mcp_tool", not "custom_tool"
215203 expect ( mockTask . recordToolUsage ) . toHaveBeenCalledWith ( "use_mcp_tool" )
216- expect ( TelemetryService . instance . captureToolUsage ) . toHaveBeenCalledWith ( mockTask . taskId , "use_mcp_tool" )
217204 } )
218205 } )
219206
@@ -355,7 +342,6 @@ describe("presentAssistantMessage - Custom Tool Recording", () => {
355342
356343 // Should not record usage for partial blocks
357344 expect ( mockTask . recordToolUsage ) . not . toHaveBeenCalled ( )
358- expect ( TelemetryService . instance . captureToolUsage ) . not . toHaveBeenCalled ( )
359345 } )
360346 } )
361347} )
0 commit comments