Comfy-Org/ComfyUI_frontend
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI_frontend.git synced 2026-02-05 15:40:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

[tests] Update useServerLogs test to handle task-started events

Browse Source 
Update test to simulate cm-task-started events before logs events to match the actual behavior of the composable.
This commit is contained in:
bymyself
2025-06-21 21:08:52 -07:00
committed by Jin Yi
parent 645e235dd6
commit b7c8180eb3
1 changed files with 36 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
tests-ui/tests/composables/useServerLogs.test.ts
View File

@@ -74,12 +74,26 @@ describe('useServerLogs', () => {
await startListening()
// Get the callback that was registered with useEventListener
const eventCallback = vi.mocked(useEventListener).mock.calls[0][2] as (
// Get the callbacks that were registered with useEventListener
const mockCalls = vi.mocked(useEventListener).mock.calls
const logsCallback = mockCalls.find((call) => call[1] === 'logs')?.[2] as (
event: CustomEvent<LogsWsMessage>
) => void
const taskStartedCallback = mockCalls.find(
(call) => call[1] === 'cm-task-started'
)?.[2] as (event: CustomEvent<any>) => void
// Simulate receiving a log event
// First, simulate task started event
const taskStartedEvent = new CustomEvent('cm-task-started', {
detail: {
type: 'cm-task-started',
ui_id: 'test-ui-id'
}
})
taskStartedCallback(taskStartedEvent)
await nextTick()
// Now simulate receiving a log event
const mockEvent = new CustomEvent('logs', {
detail: {
type: 'logs',
@@ -87,7 +101,7 @@ describe('useServerLogs', () => {
} as unknown as LogsWsMessage
}) as CustomEvent<LogsWsMessage>
eventCallback(mockEvent)
logsCallback(mockEvent)
await nextTick()
expect(logs.value).toEqual(['Log message 1', 'Log message 2'])
@@ -101,10 +115,26 @@ describe('useServerLogs', () => {
await startListening()
const eventCallback = vi.mocked(useEventListener).mock.calls[0][2] as (
// Get the callbacks that were registered with useEventListener
const mockCalls = vi.mocked(useEventListener).mock.calls
const logsCallback = mockCalls.find((call) => call[1] === 'logs')?.[2] as (
event: CustomEvent<LogsWsMessage>
) => void
const taskStartedCallback = mockCalls.find(
(call) => call[1] === 'cm-task-started'
)?.[2] as (event: CustomEvent<any>) => void
// First, simulate task started event
const taskStartedEvent = new CustomEvent('cm-task-started', {
detail: {
type: 'cm-task-started',
ui_id: 'test-ui-id'
}
})
taskStartedCallback(taskStartedEvent)
await nextTick()
// Now simulate receiving a log event
const mockEvent = new CustomEvent('logs', {
detail: {
type: 'logs',
@@ -116,7 +146,7 @@ describe('useServerLogs', () => {
} as unknown as LogsWsMessage
}) as CustomEvent<LogsWsMessage>
eventCallback(mockEvent)
logsCallback(mockEvent)
await nextTick()
expect(logs.value).toEqual(['Log message 1 dont remove me', ''])