1 files changed, 70 insertions, 0 deletions
diff --git a/internal/storage/db_test.go b/internal/storage/db_test.go
index 0426bd0..0e67e02 100644
--- a/internal/storage/db_test.go
+++ b/internal/storage/db_test.go
@@ -1332,3 +1332,73 @@ func TestListTasksByStory(t *testing.T) {
 	}
 }
 
+func TestUpdateTaskCheckerReport(t *testing.T) {
+	db := testDB(t)
+	tk := &task.Task{
+		ID: "cr-1", Name: "orig", RepositoryURL: "https://github.com/x/y",
+		Agent:    task.AgentConfig{Type: "claude", Instructions: "x"},
+		Priority: task.PriorityNormal,
+		Retry:    task.RetryConfig{MaxAttempts: 1, Backoff: "linear"},
+		Tags: []string{}, DependsOn: []string{},
+		State: task.StatePending, CreatedAt: time.Now().UTC(), UpdatedAt: time.Now().UTC(),
+	}
+	if err := db.CreateTask(tk); err != nil {
+		t.Fatalf("CreateTask: %v", err)
+	}
+	if err := db.UpdateTaskCheckerReport("cr-1", "Tests failed: missing endpoint"); err != nil {
+		t.Fatalf("UpdateTaskCheckerReport: %v", err)
+	}
+	got, err := db.GetTask("cr-1")
+	if err != nil {
+		t.Fatalf("GetTask: %v", err)
+	}
+	if got.CheckerReport != "Tests failed: missing endpoint" {
+		t.Errorf("expected checker report, got %q", got.CheckerReport)
+	}
+}
+
+func TestGetCheckerTask(t *testing.T) {
+	db := testDB(t)
+	checked := &task.Task{
+		ID: "chk-orig", Name: "orig", RepositoryURL: "https://github.com/x/y",
+		Agent:    task.AgentConfig{Type: "claude", Instructions: "x"},
+		Priority: task.PriorityNormal,
+		Retry:    task.RetryConfig{MaxAttempts: 1, Backoff: "linear"},
+		Tags: []string{}, DependsOn: []string{},
+		State: task.StatePending, CreatedAt: time.Now().UTC(), UpdatedAt: time.Now().UTC(),
+	}
+	if err := db.CreateTask(checked); err != nil {
+		t.Fatalf("CreateTask checked: %v", err)
+	}
+	checker := &task.Task{
+		ID: "chk-checker", Name: "Check: orig", CheckerForTaskID: "chk-orig",
+		RepositoryURL: "https://github.com/x/y",
+		Agent:    task.AgentConfig{Type: "claude", Instructions: "validate"},
+		Priority: task.PriorityNormal,
+		Retry:    task.RetryConfig{MaxAttempts: 1, Backoff: "linear"},
+		Tags: []string{}, DependsOn: []string{},
+		State: task.StatePending, CreatedAt: time.Now().UTC(), UpdatedAt: time.Now().UTC(),
+	}
+	if err := db.CreateTask(checker); err != nil {
+		t.Fatalf("CreateTask checker: %v", err)
+	}
+
+	// Should find the checker task.
+	got, err := db.GetCheckerTask("chk-orig")
+	if err != nil {
+		t.Fatalf("GetCheckerTask: %v", err)
+	}
+	if got == nil || got.ID != "chk-checker" {
+		t.Errorf("expected checker task ID chk-checker, got %v", got)
+	}
+
+	// Should return nil when no checker exists.
+	none, err := db.GetCheckerTask("nonexistent")
+	if err != nil {
+		t.Fatalf("GetCheckerTask nonexistent: %v", err)
+	}
+	if none != nil {
+		t.Errorf("expected nil for task with no checker, got %v", none)
+	}
+}
+