Cyrene_v2/cmd/cyrene/persona_emotion_test.go

package main

import (
	"context"
	"testing"

	"git.yeij.top/AskaEth/Cyrene_v2/internal/persona/emotion"
	"git.yeij.top/AskaEth/Cyrene_v2/internal/persona/loader"
	personareg "git.yeij.top/AskaEth/Cyrene_v2/internal/persona/registry"
	"git.yeij.top/AskaEth/Cyrene_v2/internal/runtime/agent"
	"git.yeij.top/AskaEth/Cyrene_v2/internal/runtime/loop"
	"git.yeij.top/AskaEth/Cyrene_v2/internal/runtime/model"
)

// This file locks design/22 acceptance 4 at the composition-root level: with two
// roles registered, a turn of role Y must advance ONLY Y's emotion machine and
// affection (design/01 §2: 情绪/好感度 follow the persona), and the `mood` frame it
// broadcasts must name Y — never the default role. The turn's role reaches the hook
// through the loop's per-turn SessionEnv resolution (loop.TurnEnd.PersonaID), so a
// regression to a construction-time persona shows up as the wrong machine moving.

// testTwoPersonaApp registers two personas (each with its own emotion machine and
// affection curve) and wires the composition root's emotion observer exactly as
// wire.go does.
func testTwoPersonaApp(t *testing.T) (*app, *fakeMoodEmitter) {
	t.Helper()
	reg := personareg.New()
	for _, p := range []*loader.Persona{
		{
			ID:        "xilian",
			Name:      "昔涟",
			Emotion:   loader.EmotionParams{InitialState: emotion.Calm, DecayRate: 0.5},
			Affection: loader.AffectionParams{Initial: 50, Min: 0, Max: 700, Delta: 10},
		},
		{
			ID:        "yuki",
			Name:      "由纪",
			Emotion:   loader.EmotionParams{InitialState: emotion.Calm, DecayRate: 0.5},
			Affection: loader.AffectionParams{Initial: 20, Min: 0, Max: 700, Delta: 5},
		},
	} {
		if err := reg.RegisterPersona(p); err != nil {
			t.Fatalf("RegisterPersona(%s): %v", p.ID, err)
		}
	}
	if err := reg.RegisterCharacter(&personareg.Character{ID: "xilian", Name: "昔涟", Default: true, PersonaID: "xilian"}); err != nil {
		t.Fatalf("RegisterCharacter: %v", err)
	}
	if err := reg.RegisterCharacter(&personareg.Character{ID: "yuki", Name: "由纪", PersonaID: "yuki"}); err != nil {
		t.Fatalf("RegisterCharacter: %v", err)
	}

	sink := &fakeMoodEmitter{}
	a := &app{log: testLog(), personas: reg, moodSink: sink}
	a.emotion = a.buildEmotionObserver()
	return a, sink
}

// TestTurnEndMovesOnlyTheTurnPersona is acceptance 4: a Y turn moves Y's emotion and
// affection and leaves the default role's untouched, and the mood broadcast carries
// Y's id (so the frontend's 心情 badge is the active role's, not a global one).
func TestTurnEndMovesOnlyTheTurnPersona(t *testing.T) {
	t.Parallel()
	a, sink := testTwoPersonaApp(t)
	ag, err := agent.New("你是由纪。", nil, nil, agent.WithModel(moodAdapter{reply: "由纪在呢～"}, nil))
	if err != nil {
		t.Fatalf("agent.New: %v", err)
	}
	l := loop.New(loop.Options{
		Agent:     ag,
		Agents:    map[string]*agent.Agent{"yuki": ag},
		PersonaID: "xilian", // the configured default role is NOT the turn's role
		OnTurnEnd: a.onTurnEnd,
	})

	env := loop.OwnerEnv()
	env.PersonaID = "yuki"
	res, err := l.Run(loop.WithSessionEnv(context.Background(), env), "web", []model.Message{model.TextMessage(model.RoleUser, "好开心呀，最喜欢你了")})
	if err != nil {
		t.Fatalf("Run: %v", err)
	}
	if res.Mood == "" {
		t.Fatal("Result.Mood is empty, want Y's mood for the turn")
	}

	yukiMachine, yukiAffection, ok := a.personas.Emotions().Get("yuki")
	if !ok {
		t.Fatal("persona yuki has no emotion machine")
	}
	if yukiMachine.Current() == emotion.Calm {
		t.Fatalf("yuki machine state = %s, want it moved by her own turn", yukiMachine.Current())
	}
	if yukiAffection.Value() != 25 {
		t.Fatalf("yuki affection = %d, want 25 (20 + her own step 5)", yukiAffection.Value())
	}

	xilianMachine, xilianAffection, ok := a.personas.Emotions().Get("xilian")
	if !ok {
		t.Fatal("persona xilian has no emotion machine")
	}
	if xilianMachine.Current() != emotion.Calm {
		t.Fatalf("default role machine = %s, want it untouched (calm)", xilianMachine.Current())
	}
	if xilianAffection.Value() != 50 {
		t.Fatalf("default role affection = %d, want it untouched (50)", xilianAffection.Value())
	}

	if sink.calls != 1 || sink.personaID != "yuki" {
		t.Fatalf("mood broadcast = %+v, want exactly one frame naming yuki", sink)
	}
}