feat: harden jsonl runtime reliability

pkg/tools/subagent_mailbox_test.go

@@ -0,0 +1,81 @@
+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)
+	}
+}