Function TestProcessToolCalls_UsesPinnedAgent

pkg/runtime/runtime_test.go:2823–2872

TestProcessToolCalls_UsesPinnedAgent verifies that tool-call events emitted by processToolCalls carry the pinned agent's name, not root's. Before the fix, processToolCalls called r.CurrentAgent() which always returned root for background sessions.

(t *testing.T)

// processToolCalls called r.CurrentAgent() which always returned root for
// background sessions.
func TestProcessToolCalls_UsesPinnedAgent(t *testing.T) {
	t.Parallel()

	var executed bool
	workerTool := tools.Tool{
		Name:       "worker_tool",
		Parameters: map[string]any{},
		Handler: func(_ context.Context, _ tools.ToolCall) (*tools.ToolCallResult, error) {
			executed = true
			return tools.ResultSuccess("ok"), nil
		},
	}

	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)
	rt.registerDefaultTools()
	assert.Equal(t, "root", rt.CurrentAgentName(t.Context()))

	// Simulate a background session pinned to "worker".
	sess := session.New(
		session.WithUserMessage("go"),
		session.WithToolsApproved(true),
		session.WithAgentName("worker"),
	)

	calls := []tools.ToolCall{{
		ID:       "call-1",
		Type:     "function",
		Function: tools.FunctionCall{Name: "worker_tool", Arguments: "{}"},
	}}

	events := make(chan Event, 32)
	rt.processToolCalls(t.Context(), sess, calls, []tools.Tool{workerTool}, NewChannelSink(events))
	close(events)

	assert.True(t, executed, "worker_tool handler should have been called")

	// Every event emitted must reference "worker", not "root".
	for ev := range events {
		if named, ok := ev.(interface{ GetAgentName() string }); ok {
			assert.Equal(t, "worker", named.GetAgentName(),
				"event %T should reference pinned agent \"worker\", not root", ev)
		}
	}
}

func TestFilterExcludedTools(t *testing.T) {
	t.Parallel()