Add inbox update reply done fail commands

2026-03-19 03:05:18 +08:00
parent e9792dee88
commit 11bee52ff4
6 changed files with 705 additions and 9 deletions
@@ -75,6 +75,81 @@ func TestInboxLifecycle(t *testing.T) {
 		t.Fatalf("expected claimed thread, got %q", claimedStatus)
 	}

+	updateOut := runInboxCommand(
+		t,
+		"--db", dbPath,
+		"--json",
+		"update",
+		"--agent", "worker-a",
+		"--thread", threadID,
+		"--status", "in_progress",
+		"--summary", "Implementation started",
+		"--body", "Scanning current HTTP client usage.",
+	)
+
+	var updateResp map[string]any
+	mustDecodeJSON(t, updateOut, &updateResp)
+	updatedStatus := nestedString(t, updateResp, "data", "thread", "status")
+	if updatedStatus != "in_progress" {
+		t.Fatalf("expected in_progress thread, got %q", updatedStatus)
+	}
+
+	blockedOut := runInboxCommand(
+		t,
+		"--db", dbPath,
+		"--json",
+		"update",
+		"--agent", "worker-a",
+		"--thread", threadID,
+		"--status", "blocked",
+		"--summary", "Need timeout decision",
+		"--payload-json", `{"question":"Should retries apply to read timeouts?"}`,
+	)
+
+	var blockedResp map[string]any
+	mustDecodeJSON(t, blockedOut, &blockedResp)
+	blockedStatus := nestedString(t, blockedResp, "data", "thread", "status")
+	if blockedStatus != "blocked" {
+		t.Fatalf("expected blocked thread, got %q", blockedStatus)
+	}
+
+	replyOut := runInboxCommand(
+		t,
+		"--db", dbPath,
+		"--json",
+		"reply",
+		"--from", "leader",
+		"--to", "worker-a",
+		"--thread", threadID,
+		"--summary", "Retry read timeouts",
+		"--body", "Yes, include read timeouts in the retry policy.",
+	)
+
+	var replyResp map[string]any
+	mustDecodeJSON(t, replyOut, &replyResp)
+	replyKind := nestedString(t, replyResp, "data", "message", "kind")
+	if replyKind != "answer" {
+		t.Fatalf("expected answer reply, got %q", replyKind)
+	}
+
+	doneOut := runInboxCommand(
+		t,
+		"--db", dbPath,
+		"--json",
+		"done",
+		"--agent", "worker-a",
+		"--thread", threadID,
+		"--summary", "Retry policy implemented",
+		"--body", "The HTTP client now retries the selected transient failures.",
+	)
+
+	var doneResp map[string]any
+	mustDecodeJSON(t, doneOut, &doneResp)
+	doneStatus := nestedString(t, doneResp, "data", "thread", "status")
+	if doneStatus != "done" {
+		t.Fatalf("expected done thread, got %q", doneStatus)
+	}
+
 	showOut := runInboxCommand(
 		t,
 		"--db", dbPath,
@@ -86,13 +161,80 @@ func TestInboxLifecycle(t *testing.T) {
 	var showResp map[string]any
 	mustDecodeJSON(t, showOut, &showResp)
 	showStatus := nestedString(t, showResp, "data", "thread", "status")
-	if showStatus != "claimed" {
-		t.Fatalf("expected show status claimed, got %q", showStatus)
+	if showStatus != "done" {
+		t.Fatalf("expected show status done, got %q", showStatus)
 	}
 	messagesValue := nestedValue(t, showResp, "data", "messages")
 	messages, ok := messagesValue.([]any)
-	if !ok || len(messages) != 2 {
-		t.Fatalf("expected two messages in thread history, got %#v", messagesValue)
+	if !ok || len(messages) != 6 {
+		t.Fatalf("expected six messages in thread history, got %#v", messagesValue)
+	}
+}
+
+func TestInboxFailLifecycle(t *testing.T) {
+	t.Parallel()
+
+	dbPath := filepath.Join(t.TempDir(), "coord.db")
+
+	runInboxCommand(t, "--db", dbPath, "--json", "init")
+
+	sendOut := runInboxCommand(
+		t,
+		"--db", dbPath,
+		"--json",
+		"send",
+		"--from", "leader",
+		"--to", "worker-b",
+		"--subject", "Investigate failing migration",
+		"--summary", "Check migration failure",
+		"--run", "run_blog_002",
+		"--task", "T2",
+	)
+
+	var sendResp map[string]any
+	mustDecodeJSON(t, sendOut, &sendResp)
+	threadID := nestedString(t, sendResp, "data", "thread", "thread_id")
+
+	runInboxCommand(
+		t,
+		"--db", dbPath,
+		"--json",
+		"claim",
+		"--agent", "worker-b",
+		"--thread", threadID,
+	)
+
+	failOut := runInboxCommand(
+		t,
+		"--db", dbPath,
+		"--json",
+		"fail",
+		"--agent", "worker-b",
+		"--thread", threadID,
+		"--summary", "Migration failed",
+		"--body", "The migration cannot proceed because the prior schema is inconsistent.",
+	)
+
+	var failResp map[string]any
+	mustDecodeJSON(t, failOut, &failResp)
+	failStatus := nestedString(t, failResp, "data", "thread", "status")
+	if failStatus != "failed" {
+		t.Fatalf("expected failed thread, got %q", failStatus)
+	}
+
+	showOut := runInboxCommand(
+		t,
+		"--db", dbPath,
+		"--json",
+		"show",
+		"--thread", threadID,
+	)
+
+	var showResp map[string]any
+	mustDecodeJSON(t, showOut, &showResp)
+	showStatus := nestedString(t, showResp, "data", "thread", "status")
+	if showStatus != "failed" {
+		t.Fatalf("expected show status failed, got %q", showStatus)
 	}
 }