Tools/Notes
1
0
Fork 0
mirror of https://github.com/TriliumNext/Notes.git synced 2025-07-27 10:02:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

test(server): try to reduce number to avoid CI crashing

Browse Source
This commit is contained in:
Elian Doran 2025-06-15 11:58:03 +03:00
parent 2d5754c005
commit 1dce202d21
No known key found for this signature in database
1 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
apps/server/src/services/llm/providers/integration/streaming.spec.ts
View File

@ -70,7 +70,7 @@ describe('Provider Streaming Integration Tests', () => {
expect(result.completeText).toBe('Hello world!'); expect(result.completeText).toBe('Hello world!');
expect(result.chunkCount).toBe(4); expect(result.chunkCount).toBe(4);
expect(receivedChunks.length).toBeGreaterThan(0); expect(receivedChunks.length).toBeGreaterThan(0);
// Verify callback received content chunks // Verify callback received content chunks
const contentChunks = receivedChunks.filter(c => c.text); const contentChunks = receivedChunks.filter(c => c.text);
expect(contentChunks.length).toBe(3); expect(contentChunks.length).toBe(3);
@ -101,10 +101,10 @@ describe('Provider Streaming Integration Tests', () => {
choices: [{ delta: { content: 'The answer is 4' } }], choices: [{ delta: { content: 'The answer is 4' } }],
model: 'gpt-3.5-turbo' model: 'gpt-3.5-turbo'
}, },
{ {
choices: [{ finish_reason: 'stop' }], choices: [{ finish_reason: 'stop' }],
model: 'gpt-3.5-turbo', model: 'gpt-3.5-turbo',
done: true done: true
} }
]; ];
@ -174,7 +174,7 @@ describe('Provider Streaming Integration Tests', () => {
expect(result.completeText).toBe('The weather today is sunny.'); expect(result.completeText).toBe('The weather today is sunny.');
expect(result.chunkCount).toBe(4); expect(result.chunkCount).toBe(4);
// Verify final chunk has usage stats // Verify final chunk has usage stats
expect(result.finalChunk.prompt_eval_count).toBe(15); expect(result.finalChunk.prompt_eval_count).toBe(15);
expect(result.finalChunk.eval_count).toBe(8); expect(result.finalChunk.eval_count).toBe(8);
@ -319,7 +319,7 @@ describe('Provider Streaming Integration Tests', () => {
); );
expect(result.completeText).toBe('Based on my analysis, the answer is 42.'); expect(result.completeText).toBe('Based on my analysis, the answer is 42.');
// Verify thinking states were captured // Verify thinking states were captured
const thinkingChunks = receivedChunks.filter(c => c.chunk?.message?.thinking); const thinkingChunks = receivedChunks.filter(c => c.chunk?.message?.thinking);
expect(thinkingChunks.length).toBe(2); expect(thinkingChunks.length).toBe(2);
@ -332,7 +332,7 @@ describe('Provider Streaming Integration Tests', () => {
async *[Symbol.asyncIterator]() { async *[Symbol.asyncIterator]() {
yield { message: { content: 'Starting...' } }; yield { message: { content: 'Starting...' } };
// Simulate timeout // Simulate timeout
await new Promise((_, reject) => await new Promise((_, reject) =>
setTimeout(() => reject(new Error('Request timeout')), 100) setTimeout(() => reject(new Error('Request timeout')), 100)
); );
} }
@ -481,7 +481,7 @@ describe('Provider Streaming Integration Tests', () => {
describe('Memory Management', () => { describe('Memory Management', () => {
it('should not leak memory during long streaming sessions', async () => { it('should not leak memory during long streaming sessions', async () => {
// Reduced chunk count for CI stability - still tests memory management // Reduced chunk count for CI stability - still tests memory management
const chunkCount = 1000; // Reduced from 10000 const chunkCount = 500; // Reduced from 10000
const longSessionIterator = { const longSessionIterator = {
async *[Symbol.asyncIterator]() { async *[Symbol.asyncIterator]() {
for (let i = 0; i < chunkCount; i++) { for (let i = 0; i < chunkCount; i++) {
@ -489,7 +489,7 @@ describe('Provider Streaming Integration Tests', () => {
message: { content: `Chunk ${i} with some additional content to increase memory usage` }, message: { content: `Chunk ${i} with some additional content to increase memory usage` },
done: i === (chunkCount - 1) done: i === (chunkCount - 1)
}; };
// Periodic yield to event loop to prevent blocking // Periodic yield to event loop to prevent blocking
if (i % 50 === 0) { // More frequent yields for shorter test if (i % 50 === 0) { // More frequent yields for shorter test
await new Promise(resolve => setImmediate(resolve)); await new Promise(resolve => setImmediate(resolve));
@ -499,16 +499,16 @@ describe('Provider Streaming Integration Tests', () => {
}; };
const initialMemory = process.memoryUsage(); const initialMemory = process.memoryUsage();
const result = await processProviderStream( const result = await processProviderStream(
longSessionIterator, longSessionIterator,
mockProviderOptions mockProviderOptions
); );
const finalMemory = process.memoryUsage(); const finalMemory = process.memoryUsage();
expect(result.chunkCount).toBe(chunkCount); expect(result.chunkCount).toBe(chunkCount);
// Memory increase should be reasonable (less than 20MB for smaller test) // Memory increase should be reasonable (less than 20MB for smaller test)
const memoryIncrease = finalMemory.heapUsed - initialMemory.heapUsed; const memoryIncrease = finalMemory.heapUsed - initialMemory.heapUsed;
expect(memoryIncrease).toBeLessThan(20 * 1024 * 1024); expect(memoryIncrease).toBeLessThan(20 * 1024 * 1024);
@ -581,4 +581,4 @@ describe('Provider Streaming Integration Tests', () => {
)).rejects.toThrow('Invalid stream iterator'); )).rejects.toThrow('Invalid stream iterator');
}); });
}); });
}); });