Function TestDoCompactAfterHookFires

pkg/runtime/session_compaction_test.go:232–286

TestDoCompactAfterHookFires verifies that after_compaction fires when a summary was applied (LLM-path or hook-path), and that the hook receives the produced summary text together with the *pre-compaction* token counts (so observability handlers can express "compacted from X to Y").

(t *testing.T)

// *pre-compaction* token counts (so observability handlers can
// express "compacted from X to Y").
func TestDoCompactAfterHookFires(t *testing.T) {
	t.Parallel()

	dir := t.TempDir()
	logFile := dir + "/after.log"

	const customSummary = "summary from the before hook"
	beforeJSON := `{"hook_specific_output":{"hook_event_name":"before_compaction","summary":"` + customSummary + `"}}`

	hookCfg := &latest.HooksConfig{
		BeforeCompaction: []latest.HookDefinition{
			{Type: "command", Command: "echo '" + beforeJSON + "'", Timeout: 5},
		},
		AfterCompaction: []latest.HookDefinition{
			// Capture summary plus pre-compaction tokens; if the runtime
			// regresses to passing post-compaction values we'll see
			// input_tokens == EstimateMessageTokens(summary) instead of
			// the pre-compaction 1234.
			{Type: "command", Command: "cat | jq -r '\"\\(.summary)|\\(.input_tokens)|\\(.output_tokens)\"' > " + logFile, Timeout: 5},
		},
	}

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

	rt, err := NewLocalRuntime(t.Context(), tm,
		WithSessionCompaction(false),
		WithModelStore(mockModelStoreWithLimit{limit: 100_000}),
	)
	require.NoError(t, err)

	sess := session.New(session.WithMessages([]session.Item{
		session.NewMessageItem(&session.Message{Message: chat.Message{Role: chat.MessageRoleUser, Content: "hi"}}),
	}))
	// Seed pre-compaction token counts so we can verify the hook
	// receives them rather than the post-compaction values (which
	// would be approximately EstimateMessageTokens(summary) and 0).
	sess.InputTokens = 1234
	sess.OutputTokens = 567

	events := make(chan Event, 32)
	rt.compactWithReason(t.Context(), sess, "", compactionReasonThreshold, NewChannelSink(events))
	close(events)
	for range events {
	}

	logged, readErr := os.ReadFile(logFile)
	require.NoError(t, readErr, "after_compaction hook must have run and produced the log file")
	assert.Equal(t, customSummary+"|1234|567\n", string(logged),
		"after_compaction must receive the produced summary and the *pre-compaction* token counts")
}

// TestDoCompactNoHooksMatchesPriorBehavior is a regression guard: with
// no compaction-related hooks configured, compactWithReason must still