mirror of
https://github.com/YspCoder/clawgo.git
synced 2026-04-18 12:07:29 +08:00
82 lines
2.2 KiB
Go
82 lines
2.2 KiB
Go
package tools
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
)
|
|
|
|
func TestAgentMailboxStoreReloadsInboxIndexes(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
workspace := t.TempDir()
|
|
store := NewAgentMailboxStore(workspace)
|
|
|
|
thread, err := store.EnsureThread(AgentThread{
|
|
Owner: "planner",
|
|
Topic: "handoff",
|
|
CreatedAt: 10,
|
|
UpdatedAt: 10,
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("ensure thread: %v", err)
|
|
}
|
|
msg, err := store.AppendMessage(AgentMessage{
|
|
ThreadID: thread.ThreadID,
|
|
FromAgent: "planner",
|
|
ToAgent: "worker-a",
|
|
Type: "task",
|
|
Content: "check deploy logs",
|
|
RequiresReply: true,
|
|
Status: "queued",
|
|
CreatedAt: 20,
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("append message: %v", err)
|
|
}
|
|
|
|
inbox, err := store.Inbox("worker-a", 10)
|
|
if err != nil {
|
|
t.Fatalf("inbox: %v", err)
|
|
}
|
|
if len(inbox) != 1 || inbox[0].MessageID != msg.MessageID {
|
|
t.Fatalf("unexpected inbox: %#v", inbox)
|
|
}
|
|
|
|
reloaded := NewAgentMailboxStore(workspace)
|
|
threadInbox, err := reloaded.ThreadInbox(thread.ThreadID, "worker-a", 10)
|
|
if err != nil {
|
|
t.Fatalf("thread inbox: %v", err)
|
|
}
|
|
if len(threadInbox) != 1 || threadInbox[0].Content != "check deploy logs" {
|
|
t.Fatalf("unexpected thread inbox: %#v", threadInbox)
|
|
}
|
|
|
|
if _, err := reloaded.UpdateMessageStatus(msg.MessageID, "processed", 30); err != nil {
|
|
t.Fatalf("update status: %v", err)
|
|
}
|
|
inbox, err = reloaded.Inbox("worker-a", 10)
|
|
if err != nil {
|
|
t.Fatalf("inbox after status update: %v", err)
|
|
}
|
|
if len(inbox) != 0 {
|
|
t.Fatalf("expected empty inbox after status update, got %#v", inbox)
|
|
}
|
|
|
|
runtimeDir := filepath.Join(workspace, "agents", "runtime")
|
|
if err := os.Remove(filepath.Join(runtimeDir, "threads.meta.json")); err != nil {
|
|
t.Fatalf("remove thread meta: %v", err)
|
|
}
|
|
if err := os.Remove(filepath.Join(runtimeDir, "agent_messages.meta.json")); err != nil {
|
|
t.Fatalf("remove messages meta: %v", err)
|
|
}
|
|
rebuilt := NewAgentMailboxStore(workspace)
|
|
messages, err := rebuilt.MessagesByThread(thread.ThreadID, 10)
|
|
if err != nil {
|
|
t.Fatalf("messages by thread: %v", err)
|
|
}
|
|
if len(messages) != 1 || messages[0].Status != "processed" {
|
|
t.Fatalf("unexpected rebuilt messages: %#v", messages)
|
|
}
|
|
}
|