mirror of
https://github.com/tiennm99/goclaw.git
synced 2026-07-26 18:19:05 +00:00
viettranx
28a70fd4f0
Add context helpers for audio config routing. Implement resolveVoiceAndModel with agent→user→global precedence. Add 5 precedence tests covering inheritance and override scenarios.
94 lines
2.4 KiB
Go
94 lines
2.4 KiB
Go
package store
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
func TestAgentAudioFromCtx_EmptyContext(t *testing.T) {
|
|
t.Parallel()
|
|
_, ok := AgentAudioFromCtx(context.Background())
|
|
if ok {
|
|
t.Error("expected ok=false on bare context.Background()")
|
|
}
|
|
}
|
|
|
|
func TestAgentAudioFromCtx_RoundTrip(t *testing.T) {
|
|
t.Parallel()
|
|
id := uuid.New()
|
|
raw := json.RawMessage(`{"tts_voice_id":"V1"}`)
|
|
snap := AgentAudioSnapshot{
|
|
AgentID: id,
|
|
OtherConfig: append([]byte(nil), raw...),
|
|
}
|
|
ctx := WithAgentAudio(context.Background(), snap)
|
|
got, ok := AgentAudioFromCtx(ctx)
|
|
if !ok {
|
|
t.Fatal("expected ok=true after WithAgentAudio")
|
|
}
|
|
if got.AgentID != id {
|
|
t.Errorf("AgentID: got %v, want %v", got.AgentID, id)
|
|
}
|
|
if !bytes.Equal(got.OtherConfig, raw) {
|
|
t.Errorf("OtherConfig: got %q, want %q", got.OtherConfig, raw)
|
|
}
|
|
}
|
|
|
|
func TestAgentAudioFromCtx_NestedOverride(t *testing.T) {
|
|
t.Parallel()
|
|
id1 := uuid.New()
|
|
id2 := uuid.New()
|
|
snap1 := AgentAudioSnapshot{AgentID: id1, OtherConfig: json.RawMessage(`{"v":"1"}`)}
|
|
snap2 := AgentAudioSnapshot{AgentID: id2, OtherConfig: json.RawMessage(`{"v":"2"}`)}
|
|
|
|
ctx := WithAgentAudio(context.Background(), snap1)
|
|
ctx = WithAgentAudio(ctx, snap2)
|
|
|
|
got, ok := AgentAudioFromCtx(ctx)
|
|
if !ok {
|
|
t.Fatal("expected ok=true")
|
|
}
|
|
if got.AgentID != id2 {
|
|
t.Errorf("expected child snap (id2=%v), got %v", id2, got.AgentID)
|
|
}
|
|
}
|
|
|
|
func TestAgentAudioFromCtx_ImmutabilityAfterInsertion(t *testing.T) {
|
|
t.Parallel()
|
|
srcBytes := []byte(`{"tts_voice_id":"V1"}`)
|
|
|
|
// Producer does a defensive copy before calling WithAgentAudio.
|
|
snap := AgentAudioSnapshot{
|
|
AgentID: uuid.New(),
|
|
OtherConfig: append([]byte(nil), srcBytes...),
|
|
}
|
|
ctx := WithAgentAudio(context.Background(), snap)
|
|
|
|
// Mutate the source slice AFTER insertion.
|
|
srcBytes[0] = 'X'
|
|
|
|
got, ok := AgentAudioFromCtx(ctx)
|
|
if !ok {
|
|
t.Fatal("expected ok=true")
|
|
}
|
|
// The stored snapshot must NOT reflect the mutation.
|
|
if got.OtherConfig[0] == 'X' {
|
|
t.Error("snapshot OtherConfig was mutated — defensive copy missing in WithAgentAudio or producer")
|
|
}
|
|
}
|
|
|
|
func TestAgentAudioFromCtx_NilUUIDReturnsFalse(t *testing.T) {
|
|
t.Parallel()
|
|
// Inserting a snapshot with uuid.Nil should cause AgentAudioFromCtx to return ok=false.
|
|
snap := AgentAudioSnapshot{AgentID: uuid.Nil}
|
|
ctx := WithAgentAudio(context.Background(), snap)
|
|
_, ok := AgentAudioFromCtx(ctx)
|
|
if ok {
|
|
t.Error("expected ok=false when AgentID is uuid.Nil")
|
|
}
|
|
}
|