test

1 files changed, 20 insertions, 0 deletions

diff --git a/internal/api/server_test.go b/internal/api/server_test.go
index 2209a69..ec927c0 100644
--- a/internal/api/server_test.go
+++ b/internal/api/server_test.go
@@ -896,6 +896,26 @@ func TestServer_CancelTask_Queued_TransitionsToCancelled(t *testing.T) {
 	}
 }
 
+func TestServer_CancelTask_Blocked_TransitionsToCancelled(t *testing.T) {
+	srv, store := testServer(t)
+	createTaskWithState(t, store, "cancel-blocked-1", task.StateBlocked)
+
+	req := httptest.NewRequest("POST", "/api/tasks/cancel-blocked-1/cancel", nil)
+	w := httptest.NewRecorder()
+	srv.Handler().ServeHTTP(w, req)
+
+	if w.Code != http.StatusOK {
+		t.Fatalf("status: want 200, got %d; body: %s", w.Code, w.Body.String())
+	}
+	updated, err := store.GetTask("cancel-blocked-1")
+	if err != nil {
+		t.Fatal(err)
+	}
+	if updated.State != task.StateCancelled {
+		t.Errorf("state: want CANCELLED, got %s", updated.State)
+	}
+}
+
 func TestServer_CancelTask_Completed_Returns409(t *testing.T) {
 	srv, store := testServer(t)
 	createTaskWithState(t, store, "cancel-completed-1", task.StateCompleted)