| import { ChatService } from '$lib/services/chat.service'; |
| import type { ApiStreamSession } from '$lib/types'; |
| import { describe, expect, it } from 'vitest'; |
|
|
| function makeSession(overrides: Partial<ApiStreamSession>): ApiStreamSession { |
| return { |
| completed_at: 0, |
| conversation_id: 'conv', |
| is_done: true, |
| started_at: 0, |
| total_bytes: 0, |
| ...overrides |
| }; |
| } |
|
|
| describe('selectActiveStream', () => { |
| it('returns null on empty input', () => { |
| expect(ChatService.selectActiveStream([])).toBeNull(); |
| }); |
|
|
| it('returns null on null or undefined input', () => { |
| expect(ChatService.selectActiveStream(null)).toBeNull(); |
| expect(ChatService.selectActiveStream(undefined)).toBeNull(); |
| }); |
|
|
| it('returns the single session when it is running', () => { |
| const s = makeSession({ conversation_id: 'only', is_done: false, started_at: 42 }); |
|
|
| expect(ChatService.selectActiveStream([s])).toBe(s); |
| }); |
|
|
| it('returns null when the single session is finalized', () => { |
| const s = makeSession({ conversation_id: 'only', is_done: true, started_at: 42 }); |
|
|
| expect(ChatService.selectActiveStream([s])).toBeNull(); |
| }); |
|
|
| it('prefers a still running session over a finalized one regardless of started_at', () => { |
| const finalized = makeSession({ conversation_id: 'old', is_done: true, started_at: 1000 }); |
| const running = makeSession({ conversation_id: 'new', is_done: false, started_at: 10 }); |
|
|
| expect(ChatService.selectActiveStream([finalized, running])?.conversation_id).toBe('new'); |
| expect(ChatService.selectActiveStream([running, finalized])?.conversation_id).toBe('new'); |
| }); |
|
|
| it('among running sessions, picks the most recently started one', () => { |
| const a = makeSession({ conversation_id: 'a', is_done: false, started_at: 100 }); |
| const b = makeSession({ conversation_id: 'b', is_done: false, started_at: 200 }); |
| const c = makeSession({ conversation_id: 'c', is_done: false, started_at: 150 }); |
|
|
| expect(ChatService.selectActiveStream([a, b, c])?.conversation_id).toBe('b'); |
| expect(ChatService.selectActiveStream([c, a, b])?.conversation_id).toBe('b'); |
| }); |
|
|
| it('returns null when all sessions are finalized, the DB already holds the content', () => { |
| const a = makeSession({ conversation_id: 'a', is_done: true, started_at: 10 }); |
| const b = makeSession({ conversation_id: 'b', is_done: true, started_at: 30 }); |
| const c = makeSession({ conversation_id: 'c', is_done: true, started_at: 20 }); |
|
|
| expect(ChatService.selectActiveStream([a, b, c])).toBeNull(); |
| }); |
|
|
| it('keeps the first match on ties when both are running with identical started_at', () => { |
| |
| const a = makeSession({ conversation_id: 'first', is_done: false, started_at: 50 }); |
| const b = makeSession({ conversation_id: 'second', is_done: false, started_at: 50 }); |
|
|
| expect(ChatService.selectActiveStream([a, b])?.conversation_id).toBe('first'); |
| }); |
|
|
| it('handles a typical realistic mix: two finalized old, one freshly running, one freshly finalized', () => { |
| const old1 = makeSession({ conversation_id: 'old1', is_done: true, started_at: 100 }); |
| const old2 = makeSession({ conversation_id: 'old2', is_done: true, started_at: 200 }); |
| const freshFin = makeSession({ conversation_id: 'freshFin', is_done: true, started_at: 500 }); |
| const running = makeSession({ conversation_id: 'running', is_done: false, started_at: 400 }); |
|
|
| expect(ChatService.selectActiveStream([old1, old2, freshFin, running])?.conversation_id).toBe( |
| 'running' |
| ); |
| }); |
| }); |
|
|
