feat(ai): AmbientDrift — frame-grounded drift judge

2026-06-05 21:30:37 -04:00
parent 2690fdd513
commit b37ee40dd2
2 changed files with 147 additions and 0 deletions
@@ -0,0 +1,68 @@
+// internal/ai/ambient_test.go
+package ai
+
+import (
+	"context"
+	"errors"
+	"strings"
+	"testing"
+)
+
+func TestServiceAmbientDriftSuccess(t *testing.T) {
+	fb := &fakeBackend{out: `{"drifting": true, "message": "deep in YouTube, none of today's goals"}`}
+	line, err := NewService(fb).AmbientDrift(context.Background(), "## Goals\nship keel", []string{"YouTube - Brave"})
+	if err != nil {
+		t.Fatalf("ambient drift: %v", err)
+	}
+	if line == "" {
+		t.Fatal("expected a non-empty drift line")
+	}
+	if !strings.Contains(fb.gotPrompt, "ship keel") {
+		t.Fatalf("prompt should embed the frame, got: %s", fb.gotPrompt)
+	}
+	if !strings.Contains(fb.gotPrompt, "YouTube - Brave") {
+		t.Fatalf("prompt should embed the recent titles, got: %s", fb.gotPrompt)
+	}
+}
+
+func TestServiceAmbientDriftBackendError(t *testing.T) {
+	fb := &fakeBackend{err: errors.New("boom")}
+	if _, err := NewService(fb).AmbientDrift(context.Background(), "frame", []string{"x"}); err == nil {
+		t.Fatal("want backend error")
+	}
+}
+
+func TestParseAmbientDrift(t *testing.T) {
+	tests := []struct {
+		name    string
+		in      string
+		want    string
+		wantErr error
+	}{
+		{"on track yields empty", `{"drifting": false, "message": ""}`, "", nil},
+		{"drift returns message", `{"drifting": true, "message": "watching unrelated videos"}`, "watching unrelated videos", nil},
+		{"drift is trimmed", `{"drifting": true, "message": "  slid  "}`, "slid", nil},
+		{"drift without message degrades to silence", `{"drifting": true, "message": ""}`, "", nil},
+		{"json embedded in prose", `ok: {"drifting": true, "message": "off track"} done`, "off track", nil},
+		{"empty response", "   ", "", ErrEmptyResponse},
+		{"no json", "no braces here", "", ErrNoJSON},
+		{"malformed json", `{"drifting": true, "message":`, "", ErrInvalidAmbientDrift},
+	}
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			got, err := parseAmbientDrift(tt.in)
+			if tt.wantErr != nil {
+				if !errors.Is(err, tt.wantErr) {
+					t.Fatalf("err = %v, want %v", err, tt.wantErr)
+				}
+				return
+			}
+			if err != nil {
+				t.Fatalf("unexpected err: %v", err)
+			}
+			if got != tt.want {
+				t.Fatalf("got %q, want %q", got, tt.want)
+			}
+		})
+	}
+}