1 files changed, 15 insertions, 0 deletions
diff --git a/internal/handlers/widget_test.go b/internal/handlers/widget_test.go
index 35c21e2..95f5264 100644
--- a/internal/handlers/widget_test.go
+++ b/internal/handlers/widget_test.go
@@ -924,6 +924,21 @@ func TestHandleWidgetTaskUpdate_UpdatesTitleAndDescription(t *testing.T) {
 	}
 }
 
+func TestHandleWidgetTaskUpdate_UnknownID_Returns404(t *testing.T) {
+	s, cleanup := setupTestDB(t)
+	defer cleanup()
+	h := &Handler{store: s}
+
+	body := `{"id":"does-not-exist","title":"New title","description":""}`
+	req := httptest.NewRequest("POST", "/api/widget/task/update", strings.NewReader(body))
+	w := httptest.NewRecorder()
+	http.HandlerFunc(h.HandleWidgetTaskUpdate).ServeHTTP(w, req)
+
+	if w.Code != http.StatusNotFound {
+		t.Errorf("expected 404 for an unknown task id, got %d", w.Code)
+	}
+}
+
 func TestHandleWidgetTaskRecurrence_SetsPattern(t *testing.T) {
 	s, cleanup := setupTestDB(t)
 	defer cleanup()