foliko 2.0.30 → 2.0.32

package/tests/core/p1-fixes.test.js

@@ -0,0 +1,76 @@
+import { describe, it, expect } from 'vitest';
+const fs = require('fs');
+const os = require('os');
+const path = require('path');
+
+const { ToolLoop } = require('../../src/agent/tool-loop.js');
+const { atomicWriteFileSync } = require('../../src/common/atomic-write.js');
+const { SessionManager } = require('../../src/session/session.js');
+
+function mockClient(responses) {
+  return { chat: { completions: { create: async () => { const r = responses.shift(); if (r instanceof Error) throw r; return r; } } } };
+}
+const toolCall = (name) => ({ choices: [{ message: { content: '', tool_calls: [{ id: 'c1', type: 'function', function: { name, arguments: '{}' } }] }, finish_reason: 'tool_calls' }], usage: {} });
+const answer = (t) => ({ choices: [{ message: { content: t }, finish_reason: 'stop' }], usage: {} });
+
+describe('P1#1 — maxSteps 用尽返回 stopReason', () => {
+  it('工具轮次耗尽 → stopReason=max_steps（不是成功）', async () => {
+    // 工具一直被调用，maxSteps=2 会在两轮后耗尽
+    const loop = new ToolLoop({ tools: { loopTool: { name: 'loopTool', execute: async () => 'again' } }, maxSteps: 2 });
+    const client = mockClient([toolCall('loopTool'), toolCall('loopTool'), toolCall('loopTool')]);
+    const messages = [{ role: 'user', content: 'go' }];
+    const result = await loop.generate({ messages, systemPrompt: 's', model: 'm', client });
+    expect(result.stopReason).toBe('max_steps');
+  });
+
+  it('正常给出答复 → stopReason=stop', async () => {
+    const loop = new ToolLoop({ tools: {}, maxSteps: 3 });
+    const client = mockClient([answer('done')]);
+    const messages = [{ role: 'user', content: 'go' }];
+    const result = await loop.generate({ messages, systemPrompt: 's', model: 'm', client });
+    expect(result.stopReason).toBe('stop');
+    expect(result.text).toBe('done');
+  });
+});
+
+describe('P1#6 — atomicWriteFileSync 原子写', () => {
+  it('写入内容可完整读回', () => {
+    const dir = fs.mkdtempSync(path.join(os.tmpdir(), 'foliko-atomic-'));
+    const fp = path.join(dir, 'x.jsonl');
+    atomicWriteFileSync(fp, 'line1\nline2\n');
+    expect(fs.readFileSync(fp, 'utf8')).toBe('line1\nline2\n');
+    // 覆盖写
+    atomicWriteFileSync(fp, 'new\n');
+    expect(fs.readFileSync(fp, 'utf8')).toBe('new\n');
+    // 不残留 tmp 文件
+    expect(fs.readdirSync(dir).filter(n => n.includes('.tmp-'))).toHaveLength(0);
+    fs.rmSync(dir, { recursive: true, force: true });
+  });
+});
+
+describe('P1#3 — buildSessionContext 走活跃分支', () => {
+  it('切换分支后，非活跃分支的消息不进入上下文', () => {
+    const dir = fs.mkdtempSync(path.join(os.tmpdir(), 'foliko-branch-'));
+    const sm = new SessionManager(dir, dir, path.join(dir, 's.jsonl'), true);
+    // 主线：user A -> assistant A
+    sm.appendMessage({ role: 'user', content: '问题A' });
+    const branchPointId = sm.appendMessage({ role: 'assistant', content: '回答A' }); // 从这里分叉
+
+    // 继续主线：user B -> assistant B
+    sm.appendMessage({ role: 'user', content: '问题B-主线' });
+    sm.appendMessage({ role: 'assistant', content: '回答B-主线' });
+
+    // 从 branchPoint 开新分支（leafId 回到分叉点，再追加）
+    sm.branch(branchPointId);
+    sm.appendMessage({ role: 'user', content: '问题B-新分支' });
+    sm.appendMessage({ role: 'assistant', content: '回答B-新分支' });
+
+    const texts = sm.getMessages().map(m => (typeof m.content === 'string' ? m.content : '')).join('|');
+    // 活跃分支应包含新分支内容，不含主线的 B
+    expect(texts).toContain('问题B-新分支');
+    expect(texts).not.toContain('问题B-主线');
+    // 共同前缀仍在
+    expect(texts).toContain('问题A');
+    fs.rmSync(dir, { recursive: true, force: true });
+  });
+});