Function TestErrorEvent

pkg/runtime/runtime_test.go:571–603

(t *testing.T)

}

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

	prov := &mockProviderWithError{id: "test/error-model"}
	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"

	evCh := rt.RunStream(t.Context(), sess)

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

	require.Len(t, events, 8)
	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, &ToolsetInfoEvent{}, events[4])
	require.IsType(t, &AgentInfoEvent{}, events[5])
	require.IsType(t, &ErrorEvent{}, events[6])
	require.IsType(t, &StreamStoppedEvent{}, events[7])

	errorEvent := events[6].(*ErrorEvent)
	require.Contains(t, errorEvent.Error, "simulated error")
}

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