chore: checkpoint 第一版

internal/orchestration/viking_context_integration_test.go

@@ -0,0 +1,51 @@
+package orchestration
+
+import (
+	"context"
+	"data-terminal-cloud/internal/database"
+	"data-terminal-cloud/internal/deviceauth"
+	"os"
+	"strings"
+	"testing"
+	"time"
+)
+
+func TestVikingStartupContextRequiresMatchingSubject(t *testing.T) {
+	db := os.Getenv("TEST_DATABASE_URL")
+	if db == "" {
+		t.Skip("TEST_DATABASE_URL required")
+	}
+	ctx := context.Background()
+	pool, err := database.Open(ctx, db, "viking-context-test")
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer pool.Close()
+	for _, matches := range []bool{true, false} {
+		tx, err := pool.Begin(ctx)
+		if err != nil {
+			t.Fatal(err)
+		}
+		identity := deviceauth.Identity{ID: "10000000-0000-0000-0000-000000000501", PublicID: "10000000-0000-0000-0000-000000000511", TenantID: "10000000-0000-0000-0000-000000000001", Status: "registered"}
+		subject := "10000000-0000-0000-0000-000000000401"
+		if !matches {
+			subject = "10000000-0000-0000-0000-000000000402"
+		}
+		_, err = tx.Exec(ctx, `UPDATE consent_snapshot SET effective_consents='{"voice_chat":true,"transcript_storage":true,"story_extraction":true,"personalized_memory":true}' WHERE tenant_id=$1::uuid`, identity.TenantID)
+		if err != nil {
+			t.Fatal(err)
+		}
+		now := time.Now().UTC()
+		_, internalID, err := NewRepository().Create(ctx, tx, identity, CreateInput{RequestedGoal: "open_conversation"}, AllocationOptions{Now: now, AllocationExpires: now.Add(time.Minute), TokenExpires: now.Add(5 * time.Minute), RTCAppID: "test-app", EngineMode: "s2s_hybrid", HeartbeatInterval: 15, VikingSubjectID: subject, VikingContext: "独立记忆测试：喜欢茉莉花茶"})
+		if err != nil {
+			tx.Rollback(ctx)
+			t.Fatal(err)
+		}
+		var summary string
+		err = tx.QueryRow(ctx, `SELECT cx.sanitized_summary FROM conversation c JOIN context_snapshot cx ON cx.id=c.context_snapshot_id WHERE c.id=$1::uuid`, internalID).Scan(&summary)
+		tx.Rollback(ctx)
+		if err != nil || strings.Contains(summary, "独立记忆测试") != matches {
+			t.Fatalf("matches=%v summary=%s err=%v", matches, summary, err)
+		}
+	}
+}