1 files changed, 10 insertions, 2 deletions
diff --git a/internal/handlers/widget_test.go b/internal/handlers/widget_test.go
index 02df111..749af3a 100644
--- a/internal/handlers/widget_test.go
+++ b/internal/handlers/widget_test.go
@@ -721,18 +721,26 @@ func TestHandleWidgetAdd_CreatesTask(t *testing.T) {
 		t.Fatalf("expected 200, got %d", w.Code)
 	}
 
+	var resp widgetAddResponse
+	if err := json.NewDecoder(w.Body).Decode(&resp); err != nil {
+		t.Fatalf("failed to decode response: %v", err)
+	}
+	if resp.ID == "" {
+		t.Fatal("expected a non-empty id in the response")
+	}
+
 	tasks, err := s.GetUndatedNativeTasks()
 	if err != nil {
 		t.Fatalf("failed to read back tasks: %v", err)
 	}
 	found := false
 	for _, task := range tasks {
-		if task.Content == "Buy milk" {
+		if task.Content == "Buy milk" && task.ID == resp.ID {
 			found = true
 		}
 	}
 	if !found {
-		t.Error("expected a task with content 'Buy milk' to have been created")
+		t.Error("expected a task with content 'Buy milk' and matching id to have been created")
 	}
 }