27 lines
4.0 KiB
TypeScript
27 lines
4.0 KiB
TypeScript
import { mkdtemp, readFile, symlink } from "node:fs/promises";
|
|
import os from "node:os";
|
|
import path from "node:path";
|
|
import { describe, expect, it } from "vitest";
|
|
import { SafeWorkspace } from "../src/safe-workspace.js";
|
|
import { MemoryAttachmentStore } from "../src/attachments.js";
|
|
import { calculateAdsMetrics, createAnalyzeUploadedReportTool, createDailyBriefTool, createMockAdsMetricsTool } from "../src/tools.js";
|
|
|
|
const projectRoot = path.resolve(import.meta.dirname, "../../..");
|
|
describe("SafeWorkspace", () => {
|
|
it("writes only fixed report.md", async () => { const root=await mkdtemp(path.join(os.tmpdir(),"asm-workspace-")); const ws=new SafeWorkspace(root); await ws.init(); await ws.writeReport("正文","报告"); expect(await readFile(path.join(root,"report.md"),"utf8")).toContain("# 报告"); await ws.remove(); });
|
|
it("rejects dot-dot and absolute escapes", async () => { const root=await mkdtemp(path.join(os.tmpdir(),"asm-workspace-")); const ws=new SafeWorkspace(root); await ws.init(); await expect(ws.resolve("../outside",true)).rejects.toThrow(/outside/); await expect(ws.resolve("/etc/passwd")).rejects.toThrow(/outside/); await ws.remove(); });
|
|
it("rejects symlink parents", async () => { const root=await mkdtemp(path.join(os.tmpdir(),"asm-workspace-")); const ws=new SafeWorkspace(root); await ws.init(); await symlink(os.tmpdir(),path.join(root,"link")); await expect(ws.resolve("link/file",true)).rejects.toThrow(/Symbolic/); await ws.remove(); });
|
|
});
|
|
describe("mock ads tool", () => {
|
|
const fixture=path.join(projectRoot,"fixtures/ads-account.json");
|
|
it("computes stable rounded metrics", async () => { const result=await calculateAdsMetrics(fixture,"demo-account",7); expect(result.impressions).toBe(256500); expect(result.ctr).toBeGreaterThan(2); expect(result.campaigns).toHaveLength(3); });
|
|
it("rejects unknown account and invalid days", async () => { await expect(calculateAdsMetrics(fixture,"unknown",7)).rejects.toThrow(/Unknown/); await expect(calculateAdsMetrics(fixture,"demo-account",31)).rejects.toThrow(/1 to 30/); });
|
|
it("uses a strict TypeBox schema", () => { const tool=createMockAdsMetricsTool(fixture); expect(tool.parameters).toMatchObject({ type:"object", additionalProperties:false }); });
|
|
});
|
|
describe("uploaded report tool", () => {
|
|
it("runs the bundled analyzer and returns compact evidence", async () => { const root=await mkdtemp(path.join(os.tmpdir(),"asm-analysis-")); const ws=new SafeWorkspace(root); await ws.init(); const store=new MemoryAttachmentStore(()=>ws); const source=await readFile(path.join(projectRoot,".agents/skills/daily-campaign-triage/tests/fixtures/sample-report.csv")); const attachment=await store.add("s",{filename:"sample-report.csv",mimetype:"text/csv",stream:{async *[Symbol.asyncIterator](){yield source}}}); const tool=createAnalyzeUploadedReportTool(projectRoot,"s",ws,store); const result=await tool.execute("call",{attachmentId:attachment.attachmentId,analysisDate:"2026-08-25",retentionLagDays:1}); expect(result.details).toMatchObject({contract_version:"1.0-compact",source:{row_count:11},dates:{analysis_date:"2026-08-25"},selection:{total_ads:2,included_ads:2}}); expect(JSON.stringify(result).length).toBeLessThan(30_000); await ws.remove(); });
|
|
});
|
|
describe("Skill Word report generation",()=>{
|
|
it("uses daily-campaign-triage build_brief.py and its dated four-section contract",async()=>{ const root=await mkdtemp(path.join(os.tmpdir(),"asm-word-")); const ws=new SafeWorkspace(root); await ws.init(); let created: unknown; const tool=createDailyBriefTool(projectRoot,"s",ws,(artifact)=>{created=artifact}); const brief=JSON.parse(await readFile(path.join(projectRoot,".agents/skills/daily-campaign-triage/examples/brief.example.json"),"utf8")); await tool.execute("call",brief); const name="2026-08-25-campaign-triage.docx"; const file=await readFile(path.join(root,name)); expect(created).toMatchObject({name,downloadUrl:`/api/sessions/s/artifacts/${name}`}); expect(file.subarray(0,2).toString()).toBe("PK"); expect(file.includes(Buffer.from("word/document.xml"))).toBe(true); await ws.remove(); });
|
|
});
|