@@ -4,6 +4,7 @@ import { mock, spyOn } from 'bun:test'
44import { describe , test , expect , beforeEach , afterEach } from 'bun:test'
55
66import { useSendMessage } from '../use-send-message'
7+ import type { ElapsedTimeTracker } from '../use-elapsed-time'
78import * as codebuffClient from '../../utils/codebuff-client'
89import * as loadAgentDefs from '../../utils/load-agent-definitions'
910import * as localAgentRegistry from '../../utils/local-agent-registry'
@@ -67,6 +68,7 @@ timerDescribe('useSendMessage timer', () => {
6768 let isChainInProgressRef : React . MutableRefObject < boolean >
6869 let abortControllerRef : React . MutableRefObject < AbortController | null >
6970 let onBeforeMessageSend : ReturnType < typeof mock >
71+ let mainAgentTimer : ElapsedTimeTracker
7072
7173 beforeEach ( ( ) => {
7274 // Setup state setter mocks
@@ -105,6 +107,12 @@ timerDescribe('useSendMessage timer', () => {
105107 isChainInProgressRef = { current : false }
106108 abortControllerRef = { current : null }
107109 onBeforeMessageSend = mock ( async ( ) => ( { success : true , errors : [ ] } ) )
110+ mainAgentTimer = {
111+ start : ( ) => { } ,
112+ stop : ( ) => { } ,
113+ elapsedSeconds : 0 ,
114+ startTime : null ,
115+ }
108116
109117 // Spy on external module functions
110118 spyOn ( codebuffClient , 'getCodebuffClient' ) . mockReturnValue ( {
@@ -146,6 +154,7 @@ timerDescribe('useSendMessage timer', () => {
146154 setCanProcessQueue : mockSetCanProcessQueue ,
147155 abortControllerRef,
148156 onBeforeMessageSend,
157+ mainAgentTimer,
149158 scrollToLatest : mockScrollToLatest ,
150159 availableWidth : 80 ,
151160 } ) ,
@@ -208,6 +217,7 @@ timerDescribe('useSendMessage timer', () => {
208217 setCanProcessQueue : mockSetCanProcessQueue ,
209218 abortControllerRef,
210219 onBeforeMessageSend,
220+ mainAgentTimer,
211221 scrollToLatest : mockScrollToLatest ,
212222 availableWidth : 80 ,
213223 } ) ,
@@ -262,6 +272,7 @@ timerDescribe('useSendMessage timer', () => {
262272 setCanProcessQueue : mockSetCanProcessQueue ,
263273 abortControllerRef,
264274 onBeforeMessageSend,
275+ mainAgentTimer,
265276 scrollToLatest : mockScrollToLatest ,
266277 availableWidth : 80 ,
267278 } ) ,
0 commit comments