18 lines
6.0 KiB
TypeScript
18 lines
6.0 KiB
TypeScript
import { afterEach, beforeEach, describe, expect, it } from "vitest";
|
|
import path from "node:path";
|
|
import os from "node:os";
|
|
import { mkdtemp, writeFile } from "node:fs/promises";
|
|
import { createApp, type AppContext } from "../src/app.js";
|
|
const root=path.resolve(import.meta.dirname,"../../..");
|
|
describe("Fastify API", () => { let context: AppContext; beforeEach(async()=>{context=await createApp(root)}); afterEach(async()=>{await context.app.close()});
|
|
it("returns health, requestId and only the built-in skill", async()=>{ const health=await context.app.inject({method:"GET",url:"/health"}); expect(health.statusCode).toBe(200); expect(health.json().requestId).toMatch(/^req_/); const skills=await context.app.inject({method:"GET",url:"/api/skills"}); expect(skills.json().skills.map((skill:{name:string})=>skill.name)).toEqual(["daily-campaign-triage"]); });
|
|
it("uses unified 404 and validation errors", async()=>{ const missing=await context.app.inject({method:"DELETE",url:"/api/sessions/nope"}); expect(missing.statusCode).toBe(404); expect(missing.json().error.code).toBe("SESSION_NOT_FOUND"); const created=(await context.app.inject({method:"POST",url:"/api/sessions"})).json(); const invalid=await context.app.inject({method:"POST",url:`/api/sessions/${created.sessionId}/messages`,payload:{text:"",attachmentIds:[]}}); expect(invalid.statusCode).toBe(400); expect(invalid.json().error.code).toBe("VALIDATION_ERROR"); });
|
|
it("preserves framework 4xx status codes",async()=>{ const response=await context.app.inject({method:"POST",url:"/api/sessions",headers:{"content-type":"application/json"},payload:""}); expect(response.statusCode).toBe(400); });
|
|
it("does not expose API keys in model config", async()=>{ const {sessionId}=(await context.app.inject({method:"POST",url:"/api/sessions"})).json(); const key="sk-training-super-secret"; const saved=await context.app.inject({method:"PUT",url:`/api/sessions/${sessionId}/model-config`,payload:{providerId:"openai",api:"openai-responses",apiKey:key,modelId:"gpt-5",supportsImages:true}}); expect(saved.statusCode).toBe(200); expect(saved.body).not.toContain(key); const get=await context.app.inject({method:"GET",url:`/api/sessions/${sessionId}/model-config`}); expect(get.body).not.toContain(key); expect(get.json().model.keyHint).toBe("••••cret"); });
|
|
it("loads a saved model and key after an app restart",async()=>{ const temp=await mkdtemp(path.join(os.tmpdir(),"asm-api-model-")); const file=path.join(temp,"model.json"); const first=await createApp(root,file); const {sessionId}=(await first.app.inject({method:"POST",url:"/api/sessions"})).json(); await first.app.inject({method:"PUT",url:`/api/sessions/${sessionId}/model-config`,payload:{providerId:"openai",api:"openai-responses",apiKey:"sk-persist-across-restart",modelId:"gpt-5",supportsImages:false}}); await first.app.close(); const second=await createApp(root,file); const created=(await second.app.inject({method:"POST",url:"/api/sessions"})).json(); const restored=await second.app.inject({method:"GET",url:`/api/sessions/${created.sessionId}/model-config`}); expect(restored.json().model).toMatchObject({source:"persistent",configured:true,keyHint:"••••tart"}); await second.app.close(); });
|
|
it("persists the confirmed account KPI profile across restarts",async()=>{ const temp=await mkdtemp(path.join(os.tmpdir(),"asm-api-profile-")); const file=path.join(temp,"account.json"); const profile={accountName:"蚂蚁",analysisDatePolicy:"yesterday",retentionLagDays:1,kpis:[{metric:"CPI",role:"primary",formula:"spend / installs",direction:"lower",target:"≤ 4",scope:"账户级",maturityRule:"当日完整"}],pacing:{enabled:true,targetValue:"600",currency:"USD",scope:"账户级",period:"每日",targetType:"cap"}}; const first=await createApp(root,undefined,file); const saved=await first.app.inject({method:"PUT",url:"/api/account-profile",payload:profile}); expect(saved.statusCode).toBe(200); await first.app.close(); const second=await createApp(root,undefined,file); const restored=await second.app.inject({method:"GET",url:"/api/account-profile"}); expect(restored.json().profile).toMatchObject({accountName:"蚂蚁",kpis:[{target:"≤ 4"}],pacing:{targetType:"cap"}}); await second.app.close(); });
|
|
it("returns 202 semantics only after run acceptance and no model gives 422", async()=>{ const {sessionId}=(await context.app.inject({method:"POST",url:"/api/sessions"})).json(); const response=await context.app.inject({method:"POST",url:`/api/sessions/${sessionId}/messages`,payload:{text:"hello",attachmentIds:[]}}); expect(response.statusCode).toBe(422); expect(response.json().error.code).toBe("MODEL_NOT_CONFIGURED"); });
|
|
it("downloads only a dated Skill Word artifact",async()=>{ const {sessionId}=(await context.app.inject({method:"POST",url:"/api/sessions"})).json(); const name="2026-09-15-campaign-triage.docx"; const missing=await context.app.inject({method:"GET",url:`/api/sessions/${sessionId}/artifacts/${name}`}); expect(missing.statusCode).toBe(404); const invalid=await context.app.inject({method:"GET",url:`/api/sessions/${sessionId}/artifacts/report.docx`}); expect(invalid.statusCode).toBe(400); const session=context.sessions.get(sessionId)!; await writeFile(await session.workspace.resolve(name,true),Buffer.from("PK-test")); const response=await context.app.inject({method:"GET",url:`/api/sessions/${sessionId}/artifacts/${name}`}); expect(response.statusCode).toBe(200); expect(response.headers["content-type"]).toContain("wordprocessingml.document"); expect(response.headers["content-disposition"]).toContain(name); });
|
|
it("opens an SSE stream with correct headers", async()=>{ const address=await context.app.listen({host:"127.0.0.1",port:0}); const created=await fetch(`${address}/api/sessions`,{method:"POST"}).then((r)=>r.json()) as {sessionId:string}; const controller=new AbortController(); const response=await fetch(`${address}/api/sessions/${created.sessionId}/events`,{signal:controller.signal}); expect(response.headers.get("content-type")).toContain("text/event-stream"); controller.abort(); });
|
|
});
|