52 lines
2.0 KiB
Go
52 lines
2.0 KiB
Go
|
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)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|