1 files changed, 49 insertions, 0 deletions
diff --git a/internal/storage/db_test.go b/internal/storage/db_test.go
index 7eb81d2..4f9069a 100644
--- a/internal/storage/db_test.go
+++ b/internal/storage/db_test.go
@@ -340,6 +340,43 @@ func TestDB_UpdateTask(t *testing.T) {
 	})
 }
 
+func TestRejectTask(t *testing.T) {
+	db := testDB(t)
+	now := time.Now().UTC()
+	tk := &task.Task{
+		ID: "reject-1", Name: "R", Claude: task.ClaudeConfig{Instructions: "x"},
+		Priority: task.PriorityNormal, Retry: task.RetryConfig{MaxAttempts: 1, Backoff: "linear"},
+		Tags: []string{}, DependsOn: []string{},
+		State: task.StateReady, CreatedAt: now, UpdatedAt: now,
+	}
+	if err := db.CreateTask(tk); err != nil {
+		t.Fatal(err)
+	}
+
+	if err := db.RejectTask("reject-1", "needs more detail"); err != nil {
+		t.Fatalf("RejectTask: %v", err)
+	}
+
+	got, err := db.GetTask("reject-1")
+	if err != nil {
+		t.Fatalf("GetTask: %v", err)
+	}
+	if got.State != task.StatePending {
+		t.Errorf("state: want PENDING, got %v", got.State)
+	}
+	if got.RejectionComment != "needs more detail" {
+		t.Errorf("rejection_comment: want 'needs more detail', got %q", got.RejectionComment)
+	}
+}
+
+func TestRejectTask_NotFound(t *testing.T) {
+	db := testDB(t)
+	err := db.RejectTask("nonexistent", "comment")
+	if err == nil {
+		t.Fatal("expected error for nonexistent task")
+	}
+}
+
 func TestUpdateExecution(t *testing.T) {
 	db := testDB(t)
 	now := time.Now().UTC()
@@ -360,6 +397,9 @@ func TestUpdateExecution(t *testing.T) {
 	exec.ExitCode = 1
 	exec.ErrorMsg = "something broke"
 	exec.EndTime = now.Add(2 * time.Minute)
+	exec.StdoutPath = "/tmp/exec/stdout.log"
+	exec.StderrPath = "/tmp/exec/stderr.log"
+	exec.ArtifactDir = "/tmp/exec"
 	if err := db.UpdateExecution(exec); err != nil {
 		t.Fatal(err)
 	}
@@ -371,4 +411,13 @@ func TestUpdateExecution(t *testing.T) {
 	if got.ErrorMsg != "something broke" {
 		t.Errorf("error: want 'something broke', got %q", got.ErrorMsg)
 	}
+	if got.StdoutPath != "/tmp/exec/stdout.log" {
+		t.Errorf("stdout_path: want /tmp/exec/stdout.log, got %q", got.StdoutPath)
+	}
+	if got.StderrPath != "/tmp/exec/stderr.log" {
+		t.Errorf("stderr_path: want /tmp/exec/stderr.log, got %q", got.StderrPath)
+	}
+	if got.ArtifactDir != "/tmp/exec" {
+		t.Errorf("artifact_dir: want /tmp/exec, got %q", got.ArtifactDir)
+	}
 }