Function TestContextCancellation

pkg/runtime/runtime_test.go:605–639

(t *testing.T)

}

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

	stream := newStreamBuilder().
		AddContent("This should not complete").
		AddStopWithUsage(10, 5).
		Build()

	prov := &mockProvider{id: "test/mock-model", stream: stream}
	root := agent.New("root", "You are a test agent", agent.WithModel(prov))
	tm := team.New(team.WithAgents(root))

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

	sess := session.New(session.WithUserMessage("Hi"))
	sess.Title = "Unit Test"

	ctx, cancel := context.WithCancel(t.Context())
	evCh := rt.RunStream(ctx, sess)

	cancel()

	var events []Event
	for ev := range evCh {
		events = append(events, ev)
	}

	require.GreaterOrEqual(t, len(events), 4)
	require.IsType(t, &TeamInfoEvent{}, events[0])
	require.IsType(t, &ToolsetInfoEvent{}, events[1])
	require.IsType(t, &UserMessageEvent{}, events[2])
	require.IsType(t, &StreamStartedEvent{}, events[3])
	require.IsType(t, &StreamStoppedEvent{}, events[len(events)-1])
}

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