update see calss

apps/api/internal/aiaudit/aiaudit_test.go

@@ -0,0 +1,76 @@
+package aiaudit
+
+import (
+	"context"
+	"strings"
+	"testing"
+
+	"github.com/google/uuid"
+)
+
+// Call metadata is attached in layers: the job sets company/user/job once, then
+// each product narrows it. A later WithCall must not wipe what an outer scope set.
+func TestWithCall_mergesWithoutClearing(t *testing.T) {
+	t.Parallel()
+	company, user, job, product := uuid.New(), uuid.New(), uuid.New(), uuid.New()
+
+	ctx := WithCall(context.Background(), CallContext{
+		CompanyID: company, UserID: user, JobID: job, Role: RoleProcessing,
+	})
+	ctx = WithCall(ctx, CallContext{RawProductID: product})
+
+	got := FromContext(ctx)
+	if got.CompanyID != company || got.UserID != user || got.JobID != job {
+		t.Fatalf("outer scope lost: %+v", got)
+	}
+	if got.RawProductID != product {
+		t.Fatalf("product not attached: %+v", got)
+	}
+	if got.Role != RoleProcessing {
+		t.Fatalf("role lost: %q", got.Role)
+	}
+
+	// A narrower scope may override the role (categorize inside a processing job).
+	ctx = WithCall(ctx, CallContext{Role: RoleCategorize})
+	if got := FromContext(ctx); got.Role != RoleCategorize || got.JobID != job {
+		t.Fatalf("role override broke context: %+v", got)
+	}
+}
+
+func TestFromContext_zeroValueWhenAbsent(t *testing.T) {
+	t.Parallel()
+	if got := FromContext(context.Background()); got != (CallContext{}) {
+		t.Fatalf("expected zero CallContext, got %+v", got)
+	}
+	//nolint:staticcheck // explicitly asserting the nil-ctx guard
+	if got := FromContext(nil); got != (CallContext{}) {
+		t.Fatalf("nil ctx must be safe, got %+v", got)
+	}
+}
+
+// Capture must never be able to take down the call it observes.
+func TestRecorder_nilIsNoOp(t *testing.T) {
+	t.Parallel()
+	var r *Recorder
+	r.Record(context.Background(), Call{System: "x"}) // must not panic
+	if got := NewRecorder(nil); got != nil {
+		t.Fatal("nil pool must yield a nil (no-op) recorder")
+	}
+}
+
+// A runaway reply must not write an unbounded row.
+func TestClamp_boundsBody(t *testing.T) {
+	t.Parallel()
+	short := strings.Repeat("a", 10)
+	if clamp(short) != short {
+		t.Fatal("short bodies must pass through unchanged")
+	}
+	long := strings.Repeat("b", maxBodyRunes+500)
+	got := clamp(long)
+	if len([]rune(got)) <= maxBodyRunes {
+		t.Fatalf("clamped body should keep the cap plus a marker, got %d runes", len([]rune(got)))
+	}
+	if !strings.HasSuffix(got, "[truncated by aiaudit]") {
+		t.Fatal("clamped body must say it was truncated")
+	}
+}