Files
clawgo/pkg/agent/memory_log_test.go
2026-03-10 00:33:23 +08:00

67 lines
2.3 KiB
Go

package agent
import (
"os"
"path/filepath"
"strings"
"testing"
"time"
"github.com/YspCoder/clawgo/pkg/bus"
"github.com/YspCoder/clawgo/pkg/config"
"github.com/YspCoder/clawgo/pkg/runtimecfg"
)
func TestAppendDailySummaryLogUsesSubagentNamespaceAndTitle(t *testing.T) {
workspace := t.TempDir()
cfg := config.DefaultConfig()
cfg.Agents.Subagents["coder"] = config.SubagentConfig{
Enabled: true,
DisplayName: "Code Agent",
SystemPromptFile: "agents/coder/AGENT.md",
}
runtimecfg.Set(cfg)
t.Cleanup(func() { runtimecfg.Set(config.DefaultConfig()) })
loop := &AgentLoop{workspace: workspace}
loop.appendDailySummaryLog(bus.InboundMessage{
Channel: "cli",
SessionKey: "subagent:coder:subagent-1",
Content: "Role Profile Policy (agents/coder/AGENT.md):\n...\n\nTask:\n修复登录接口并补测试\nextra details",
Metadata: map[string]string{
"memory_ns": "coder",
},
}, "完成了登录接口修复、增加回归测试,并验证通过。")
entries, err := os.ReadFile(filepath.Join(workspace, "agents", "coder", "memory", currentDateFileName()))
if err != nil {
t.Fatalf("read namespaced daily note failed: %v", err)
}
content := string(entries)
if !strings.Contains(content, "Code Agent | 修复登录接口并补测试") {
t.Fatalf("expected agent title + task summary, got %s", content)
}
if !strings.Contains(content, "- Did: 完成了登录接口修复、增加回归测试,并验证通过。") {
t.Fatalf("expected did summary, got %s", content)
}
mainToday := filepath.Join(workspace, "memory", currentDateFileName())
mainEntries, err := os.ReadFile(mainToday)
if err != nil {
t.Fatalf("expected main memory summary to be written, got %v", err)
}
mainContent := string(mainEntries)
if !strings.Contains(mainContent, "Code Agent | 修复登录接口并补测试") {
t.Fatalf("expected main memory to include subagent title, got %s", mainContent)
}
if !strings.Contains(mainContent, "- Subagent: Code Agent") {
t.Fatalf("expected main memory to include subagent name, got %s", mainContent)
}
if !strings.Contains(mainContent, "- Did: 完成了登录接口修复、增加回归测试,并验证通过。") {
t.Fatalf("expected main memory to include summary, got %s", mainContent)
}
}
func currentDateFileName() string {
return time.Now().Format("2006-01-02") + ".md"
}