Skip to content

Commit f1be87a

Browse files
Update frontend/src/lib/hooks/use-source-chat.test.tsx
Co-authored-by: cubic-dev-ai[bot] <191113872+cubic-dev-ai[bot]@users.noreply.github.com>
1 parent 56bde41 commit f1be87a

1 file changed

Lines changed: 14 additions & 2 deletions

File tree

frontend/src/lib/hooks/use-source-chat.test.tsx

Lines changed: 14 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -238,9 +238,21 @@ describe('useSourceChat sendMessage streaming', () => {
238238
vi.mocked(sourceChatApi.getSession).mockRejectedValue(new Error('network down'))
239239

240240
const consoleError = vi.spyOn(console, 'error').mockImplementation(() => {})
241-
const { result } = renderHook(() => useSourceChat('source:1'), { wrapper: makeWrapper() })
241+
try {
242+
const { result } = renderHook(() => useSourceChat('source:1'), { wrapper: makeWrapper() })
242243

243-
await waitFor(() => expect(result.current.currentSessionId).toBe('session:1'))
244+
await waitFor(() => expect(result.current.currentSessionId).toBe('session:1'))
245+
246+
await act(async () => {
247+
await result.current.sendMessage('hello')
248+
})
249+
250+
expect(sourceChatApi.sendMessage).not.toHaveBeenCalled()
251+
expect(result.current.isStreaming).toBe(false)
252+
expect(toast.error).toHaveBeenCalled()
253+
} finally {
254+
consoleError.mockRestore()
255+
}
244256

245257
try {
246258
await act(async () => {

0 commit comments

Comments
 (0)