paliad/internal/services/paliadin_test.go

package services

import (
	"strings"
	"testing"
)

// Tests for the PoC paliadin trailer parser. The parser is load-bearing:
// it's how the dashboard learns which tools Claude used, how many rows
// each returned, and how Claude classified the question. Wrong parsing
// = silently broken monitoring.

func TestSplitTrailer_HappyPath(t *testing.T) {
	body := strings.TrimSpace(`
Diese Woche stehen 3 Fristen an:

- 16.05. Klageerwiderung [#deadline-OPEN:c47bd2-1]
- 17.05. Replik [#deadline-OPEN:e92a01-3]
- 20.05. Wiedereinsetzung [#deadline-OPEN:f31b09-7]

---
[paliadin-meta]
used_tools: search_my_deadlines, lookup_court
rows_seen: 3, 1
classifier_tag: data
[/paliadin-meta]
`)
	clean, meta := splitTrailer(body)

	if strings.Contains(clean, "[paliadin-meta]") {
		t.Fatalf("trailer not stripped from body:\n%s", clean)
	}
	if !strings.HasPrefix(clean, "Diese Woche") {
		t.Errorf("body lost prefix: %q", clean[:50])
	}
	wantTools := []string{"search_my_deadlines", "lookup_court"}
	if len(meta.UsedTools) != len(wantTools) {
		t.Fatalf("UsedTools len = %d; want %d", len(meta.UsedTools), len(wantTools))
	}
	for i, want := range wantTools {
		if meta.UsedTools[i] != want {
			t.Errorf("UsedTools[%d] = %q; want %q", i, meta.UsedTools[i], want)
		}
	}
	wantRows := []int{3, 1}
	if len(meta.RowsSeen) != len(wantRows) {
		t.Fatalf("RowsSeen len = %d; want %d", len(meta.RowsSeen), len(wantRows))
	}
	for i, want := range wantRows {
		if meta.RowsSeen[i] != want {
			t.Errorf("RowsSeen[%d] = %d; want %d", i, meta.RowsSeen[i], want)
		}
	}
	if meta.ClassifierTag != "data" {
		t.Errorf("ClassifierTag = %q; want %q", meta.ClassifierTag, "data")
	}
}

func TestSplitTrailer_NoTrailer(t *testing.T) {
	body := "Just a response, no trailer."
	clean, meta := splitTrailer(body)
	if clean != body {
		t.Errorf("body changed: %q vs %q", clean, body)
	}
	if len(meta.UsedTools) != 0 || len(meta.RowsSeen) != 0 || meta.ClassifierTag != "" {
		t.Errorf("meta should be zero: %+v", meta)
	}
}

func TestSplitTrailer_EmptyToolsList(t *testing.T) {
	body := strings.TrimSpace(`
Klageerwiderung ist die Erwiderung auf die Klage.

---
[paliadin-meta]
used_tools:
rows_seen:
classifier_tag: concept
[/paliadin-meta]
`)
	clean, meta := splitTrailer(body)
	if strings.Contains(clean, "[paliadin-meta]") {
		t.Errorf("trailer not stripped")
	}
	if len(meta.UsedTools) != 0 {
		t.Errorf("UsedTools should be empty: %v", meta.UsedTools)
	}
	if meta.ClassifierTag != "concept" {
		t.Errorf("ClassifierTag = %q; want concept", meta.ClassifierTag)
	}
}

func TestCountChips(t *testing.T) {
	cases := []struct {
		body string
		want int
	}{
		{"plain text", 0},
		{"see [#deadline-OPEN:abc-123]", 1},
		{"two [#deadline-OPEN:abc] and [#projekt-OPEN:slug]", 2},
		{"chip nav [chip:nav:/projects/123]", 1},
		{"chip filter [chip:filter:status=pending]", 1},
		{"mixed [#frist-OPEN:x] and [chip:nav:/y]", 2},
		// Hallucinated / malformed markers don't count.
		{"[#deadline-OPEN:]", 0},
		{"[#deadline-OPEN]", 0},
		{"[chip:invalid]", 0},
	}
	for _, c := range cases {
		got := countChips(c.body)
		if got != c.want {
			t.Errorf("countChips(%q) = %d; want %d", c.body, got, c.want)
		}
	}
}

func TestApproxTokenCount(t *testing.T) {
	cases := []struct {
		s    string
		want int
	}{
		{"", 0},
		{"hello", 1},                              // 1 word × 1.3 → 1
		{"hello world", 2},                        // 2 × 1.3 = 2.6 → 2
		{"one two three four five six seven", 9}, // 7 × 1.3 = 9.1 → 9
	}
	for _, c := range cases {
		got := approxTokenCount(c.s)
		if got != c.want {
			t.Errorf("approxTokenCount(%q) = %d; want %d", c.s, got, c.want)
		}
	}
}

func TestSanitiseForTmux(t *testing.T) {
	in := "first line\nsecond line\rthird"
	got := sanitiseForTmux(in)
	if strings.ContainsAny(got, "\n\r") {
		t.Errorf("sanitiseForTmux did not strip newlines: %q", got)
	}
}

func TestSanitiseForTmux_TruncatesLong(t *testing.T) {
	long := strings.Repeat("x", 10_000)
	got := sanitiseForTmux(long)
	if !strings.HasSuffix(got, "[…truncated]") {
		t.Errorf("expected truncation marker, got tail: %q", got[len(got)-20:])
	}
}