Function TestResolveSessionAgent_PinnedAgent

pkg/runtime/runtime_test.go:2763–2780

TestResolveSessionAgent_PinnedAgent verifies that resolveSessionAgent returns the session-pinned agent when AgentName is set, even though the runtime's currentAgent points elsewhere (root). Before the fix, the shared currentAgent field was always used, so background sub-agent tasks ran with root's c

(t *testing.T)

// currentAgent points elsewhere (root). Before the fix, the shared currentAgent
// field was always used, so background sub-agent tasks ran with root's config.
func TestResolveSessionAgent_PinnedAgent(t *testing.T) {
	t.Parallel()

	prov := &mockProvider{id: "test/mock-model", stream: &mockStream{}}
	worker := agent.New("worker", "Worker agent", agent.WithModel(prov))
	root := agent.New("root", "Root agent", agent.WithModel(prov))
	tm := team.New(team.WithAgents(root, worker))

	rt, err := NewLocalRuntime(t.Context(), tm, WithSessionCompaction(false), WithModelStore(mockModelStore{}))
	require.NoError(t, err)
	assert.Equal(t, "root", rt.CurrentAgentName(t.Context()), "default agent should be root")

	// Session pinned to worker (as run_background_agent does).
	sess := session.New(session.WithAgentName("worker"))

	resolved := rt.resolveSessionAgent(sess)
	assert.Equal(t, "worker", resolved.Name(), "resolveSessionAgent should return pinned agent")
}

// TestResolveSessionAgent_FallsBackToCurrentAgent verifies that when no
// AgentName is set on the session, resolveSessionAgent falls back to the