1 files changed, 34 insertions, 0 deletions
diff --git a/internal/handlers/handlers_test.go b/internal/handlers/handlers_test.go
index d16c7cf..bb383aa 100644
--- a/internal/handlers/handlers_test.go
+++ b/internal/handlers/handlers_test.go
@@ -2321,3 +2321,37 @@ func TestHandleCompleteAtom_DootShowsTitle(t *testing.T) {
 	}
 }
 
+func TestHandleTimeline_IncludesBudgetStatusWhenTrackedTaskExists(t *testing.T) {
+	h, cleanup := setupTestHandler(t)
+	defer cleanup()
+
+	project, err := h.store.CreateProject("Tracked", "#111111")
+	if err != nil {
+		t.Fatal(err)
+	}
+	if err := h.store.SetProjectBudgetTracked(project.ID, true); err != nil {
+		t.Fatal(err)
+	}
+	due := time.Now()
+	if err := h.store.CreateNativeTask(models.Task{ID: "t-tracked", Content: "Tracked", ProjectID: project.ID, DueDate: &due, EstimatedMinutes: 30}); err != nil {
+		t.Fatal(err)
+	}
+
+	req := httptest.NewRequest("GET", "/tabs/timeline", nil)
+	w := httptest.NewRecorder()
+	h.HandleTimeline(w, req)
+
+	mock := h.renderer.(*MockRenderer)
+	lastCall := mock.Calls[len(mock.Calls)-1]
+	data, ok := lastCall.Data.(TimelineData)
+	if !ok {
+		t.Fatalf("expected TimelineData, got %T", lastCall.Data)
+	}
+	if data.BudgetStatus == nil {
+		t.Fatal("expected non-nil BudgetStatus")
+	}
+	if data.BudgetStatus.Today.ScheduledMinutes != 30 {
+		t.Errorf("Today.ScheduledMinutes = %d, want 30", data.BudgetStatus.Today.ScheduledMinutes)
+	}
+}
+