diff options
Diffstat (limited to 'internal/handlers/widget_test.go')
| -rw-r--r-- | internal/handlers/widget_test.go | 159 |
1 files changed, 159 insertions, 0 deletions
diff --git a/internal/handlers/widget_test.go b/internal/handlers/widget_test.go index a638eca..10a5c02 100644 --- a/internal/handlers/widget_test.go +++ b/internal/handlers/widget_test.go @@ -1268,3 +1268,162 @@ func TestHandleWidgetGet_BudgetTrackedTaskDueToday_IncludesBudgetStatus(t *testi } } + +func TestHandleWidgetAvailabilityGet_ReturnsBlocks(t *testing.T) { + db, cleanup := setupTestDB(t) + defer cleanup() + h := &Handler{store: db} + if _, err := h.store.CreateAvailabilityBlock(1, "18:00", "20:00", "evening"); err != nil { + t.Fatal(err) + } + + req := httptest.NewRequest("GET", "/api/widget/availability", nil) + w := httptest.NewRecorder() + h.HandleWidgetAvailabilityGet(w, req) + + if w.Code != http.StatusOK { + t.Fatalf("status = %d, want 200", w.Code) + } + var blocks []models.AvailabilityBlock + if err := json.NewDecoder(w.Body).Decode(&blocks); err != nil { + t.Fatal(err) + } + if len(blocks) != 1 || blocks[0].Label != "evening" { + t.Errorf("blocks = %+v", blocks) + } +} + +func TestHandleWidgetAvailabilityCreate_CreatesBlock(t *testing.T) { + db, cleanup := setupTestDB(t) + defer cleanup() + h := &Handler{store: db} + + body := `{"weekday":2,"start_time":"07:00","end_time":"08:00","label":"morning walk"}` + req := httptest.NewRequest("POST", "/api/widget/availability", strings.NewReader(body)) + w := httptest.NewRecorder() + h.HandleWidgetAvailabilityCreate(w, req) + + if w.Code != http.StatusOK { + t.Fatalf("status = %d, want 200", w.Code) + } + blocks, err := h.store.GetAvailabilityBlocks() + if err != nil { + t.Fatal(err) + } + if len(blocks) != 1 || blocks[0].Weekday != 2 { + t.Errorf("blocks = %+v", blocks) + } +} + +func TestHandleWidgetAvailabilityDelete_RemovesBlock(t *testing.T) { + db, cleanup := setupTestDB(t) + defer cleanup() + h := &Handler{store: db} + block, err := h.store.CreateAvailabilityBlock(3, "09:00", "10:00", "") + if err != nil { + t.Fatal(err) + } + + body := `{"id":"` + block.ID + `"}` + req := httptest.NewRequest("POST", "/api/widget/availability/delete", strings.NewReader(body)) + w := httptest.NewRecorder() + h.HandleWidgetAvailabilityDelete(w, req) + + if w.Code != http.StatusOK { + t.Fatalf("status = %d, want 200", w.Code) + } + blocks, err := h.store.GetAvailabilityBlocks() + if err != nil { + t.Fatal(err) + } + if len(blocks) != 0 { + t.Errorf("expected block deleted, got %+v", blocks) + } +} + +func TestHandleWidgetAvailabilityDelete_UnknownID_Returns404(t *testing.T) { + db, cleanup := setupTestDB(t) + defer cleanup() + h := &Handler{store: db} + + req := httptest.NewRequest("POST", "/api/widget/availability/delete", strings.NewReader(`{"id":"nope"}`)) + w := httptest.NewRecorder() + h.HandleWidgetAvailabilityDelete(w, req) + + if w.Code != http.StatusNotFound { + t.Errorf("status = %d, want 404", w.Code) + } +} + +func TestHandleWidgetTaskEstimate_SetsEstimate(t *testing.T) { + db, cleanup := setupTestDB(t) + defer cleanup() + h := &Handler{store: db} + if err := h.store.CreateNativeTask(models.Task{ID: "t-1", Content: "task"}); err != nil { + t.Fatal(err) + } + + body := `{"id":"t-1","estimated_minutes":25}` + req := httptest.NewRequest("POST", "/api/widget/task/estimate", strings.NewReader(body)) + w := httptest.NewRecorder() + h.HandleWidgetTaskEstimate(w, req) + + if w.Code != http.StatusOK { + t.Fatalf("status = %d, want 200", w.Code) + } + task, err := h.store.GetNativeTaskByID("t-1") + if err != nil { + t.Fatal(err) + } + if task.EstimatedMinutes != 25 { + t.Errorf("EstimatedMinutes = %d, want 25", task.EstimatedMinutes) + } +} + +func TestHandleWidgetProjectsBudgetTracked_SetsFlag(t *testing.T) { + db, cleanup := setupTestDB(t) + defer cleanup() + h := &Handler{store: db} + project, err := h.store.CreateProject("P", "#111111") + if err != nil { + t.Fatal(err) + } + + body := `{"id":"` + project.ID + `","tracked":true}` + req := httptest.NewRequest("POST", "/api/widget/projects/budget-tracked", strings.NewReader(body)) + w := httptest.NewRecorder() + h.HandleWidgetProjectsBudgetTracked(w, req) + + if w.Code != http.StatusOK { + t.Fatalf("status = %d, want 200", w.Code) + } + got, err := h.store.GetProjectByID(project.ID) + if err != nil { + t.Fatal(err) + } + if !got.BudgetTracked { + t.Error("expected BudgetTracked = true") + } +} + +func TestHandleWidgetLabelsBudgetTracked_SetsFlag(t *testing.T) { + db, cleanup := setupTestDB(t) + defer cleanup() + h := &Handler{store: db} + + body := `{"name":"errands","tracked":true}` + req := httptest.NewRequest("POST", "/api/widget/labels/budget-tracked", strings.NewReader(body)) + w := httptest.NewRecorder() + h.HandleWidgetLabelsBudgetTracked(w, req) + + if w.Code != http.StatusOK { + t.Fatalf("status = %d, want 200", w.Code) + } + names, err := h.store.GetBudgetTrackedLabelNames() + if err != nil { + t.Fatal(err) + } + if !names["errands"] { + t.Error("expected 'errands' tracked") + } +} |