1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
package api
import (
"encoding/json"
"errors"
"net/http"
"net/http/httptest"
"testing"
"github.com/thepeterstone/claudomator/internal/budget"
)
type fakeBudget struct {
hs []budget.Headroom
err error
}
func (f fakeBudget) All() ([]budget.Headroom, error) { return f.hs, f.err }
func TestHandleGetBudget_NoBudgetConfigured_ReturnsEmpty(t *testing.T) {
srv, _ := testServer(t)
req := httptest.NewRequest("GET", "/api/budget", nil)
w := httptest.NewRecorder()
srv.Handler().ServeHTTP(w, req)
if w.Code != http.StatusOK {
t.Fatalf("status: want 200, got %d", w.Code)
}
var hs []budget.Headroom
if err := json.Unmarshal(w.Body.Bytes(), &hs); err != nil {
t.Fatalf("unmarshal: %v", err)
}
if len(hs) != 0 {
t.Errorf("want empty list, got %+v", hs)
}
}
func TestHandleGetBudget_ReportsHeadroom(t *testing.T) {
srv, _ := testServer(t)
srv.SetBudget(fakeBudget{hs: []budget.Headroom{
{Provider: "claude", Limited: true, Limit: 10, Spent: 4, Remaining: 6, Fraction: 0.6},
}})
req := httptest.NewRequest("GET", "/api/budget", nil)
w := httptest.NewRecorder()
srv.Handler().ServeHTTP(w, req)
if w.Code != http.StatusOK {
t.Fatalf("status: want 200, got %d", w.Code)
}
var hs []budget.Headroom
if err := json.Unmarshal(w.Body.Bytes(), &hs); err != nil {
t.Fatalf("unmarshal: %v", err)
}
if len(hs) != 1 || hs[0].Provider != "claude" || hs[0].Remaining != 6 {
t.Errorf("unexpected headroom: %+v", hs)
}
}
func TestHandleGetBudget_ErrorReturns500(t *testing.T) {
srv, _ := testServer(t)
srv.SetBudget(fakeBudget{err: errors.New("db down")})
req := httptest.NewRequest("GET", "/api/budget", nil)
w := httptest.NewRecorder()
srv.Handler().ServeHTTP(w, req)
if w.Code != http.StatusInternalServerError {
t.Errorf("want 500, got %d", w.Code)
}
}
|
