Function TestEmitStartupInfo_WithSessionTokenData

pkg/runtime/runtime_test.go:1646–1686

(t *testing.T)

}

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

	// When restoring a session that already has token data,
	// EmitStartupInfo should emit a TokenUsageEvent with the context limit
	// looked up from the model store so the sidebar can display context %.
	prov := &mockProvider{id: "test/startup-model", stream: &mockStream{}}
	root := agent.New("startup-test-agent", "You are a startup test agent",
		agent.WithModel(prov),
		agent.WithDescription("Startup agent"),
	)
	tm := team.New(team.WithAgents(root))

	rt, err := NewLocalRuntime(t.Context(), tm, WithCurrentAgent("startup-test-agent"),
		WithModelStore(mockModelStoreWithLimit{limit: 200_000}))
	require.NoError(t, err)

	// Create a session with existing token data (simulating session restore)
	sess := session.New()
	sess.InputTokens = 5000
	sess.OutputTokens = 1000

	events := make(chan Event, 20)
	rt.EmitStartupInfo(t.Context(), sess, NewChannelSink(events))
	close(events)

	// Collect events and find the TokenUsageEvent
	var tokenEvent *TokenUsageEvent
	for event := range events {
		if te, ok := event.(*TokenUsageEvent); ok {
			tokenEvent = te
		}
	}

	require.NotNil(t, tokenEvent, "EmitStartupInfo should emit a TokenUsageEvent for a session with token data")
	assert.Equal(t, sess.ID, tokenEvent.SessionID)
	assert.Equal(t, int64(5000), tokenEvent.Usage.InputTokens)
	assert.Equal(t, int64(1000), tokenEvent.Usage.OutputTokens)
	assert.Equal(t, int64(6000), tokenEvent.Usage.ContextLength)
	assert.Equal(t, int64(200_000), tokenEvent.Usage.ContextLimit)
}

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