Function TestReprobe_NoChangeMeansNoExtraEvents

pkg/runtime/runtime_test.go:3268–3319

TestReprobe_NoChangeMeansNoExtraEvents verifies that reprobe is a no-op (no extra ToolsetInfo events, no panics) when no new tools appear after a tool call.

(t *testing.T)

// (no extra ToolsetInfo events, no panics) when no new tools appear after
// a tool call.
func TestReprobe_NoChangeMeansNoExtraEvents(t *testing.T) {
	t.Parallel()

	staticTool := tools.Tool{
		Name:       "do_thing",
		Parameters: map[string]any{},
		Handler: func(_ context.Context, _ tools.ToolCall) (*tools.ToolCallResult, error) {
			return tools.ResultSuccess("done"), nil
		},
	}

	stream1 := newStreamBuilder().
		AddToolCallName("c1", "do_thing").
		AddToolCallArguments("c1", `{}`).
		AddStopWithUsage(5, 5).
		Build()

	prov := &recordingProvider{
		id:      "test/mock-model",
		streams: []*mockStream{stream1},
	}

	ts := newStubToolSet(nil, []tools.Tool{staticTool}, nil)
	root := agent.New("root", "test", agent.WithModel(prov), agent.WithToolSets(ts))
	tm := team.New(team.WithAgents(root))

	rt, err := NewLocalRuntime(t.Context(), tm, WithSessionCompaction(false), WithModelStore(mockModelStore{}))
	require.NoError(t, err)
	rt.registerDefaultTools()

	sess := session.New(session.WithUserMessage("Do the thing"))
	sess.Title = "no-change reprobe test"
	sess.ToolsApproved = true

	evCh := rt.RunStream(t.Context(), sess)
	var events []Event
	for ev := range evCh {
		events = append(events, ev)
	}

	// Count ToolsetInfo events — reprobe should NOT emit an extra one.
	var counts []int
	for _, ev := range events {
		if ti, ok := ev.(*ToolsetInfoEvent); ok {
			counts = append(counts, ti.AvailableTools)
		}
	}
	// All counts should be 1 (the static tool).
	for _, c := range counts {
		assert.Equal(t, 1, c, "unexpected ToolsetInfo count — reprobe emitted extra event when tools unchanged")
	}
}

func toolNames(ts []tools.Tool) []string {
	names := make([]string, len(ts))