09a3d3ab17
Copilot Setup Steps / copilot-setup-steps (push) Failing after 2s
Python check requirements.txt / check-requirements (push) Has been cancelled
Python Type-Check / python type-check (push) Has been cancelled
Update Operations Documentation / update-ops-docs (push) Has been cancelled
Check Pre-Tokenizer Hashes / pre-tokenizer-hashes (push) Has been cancelled
75 lines
3.5 KiB
TypeScript
75 lines
3.5 KiB
TypeScript
import { describe, expect, it } from 'vitest';
|
|
import { ChatService } from '$lib/services/chat.service';
|
|
import type { ApiStreamSession } from '$lib/types';
|
|
|
|
function makeSession(overrides: Partial<ApiStreamSession>): ApiStreamSession {
|
|
return {
|
|
conversation_id: 'conv',
|
|
is_done: true,
|
|
total_bytes: 0,
|
|
started_at: 0,
|
|
completed_at: 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', () => {
|
|
// reduce visits left to right, the initial accumulator stays unless a strictly greater value appears
|
|
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'
|
|
);
|
|
});
|
|
});
|