Function TestCache_StorageIsAStopHookBuiltin

pkg/runtime/cache_test.go:160–187

TestCache_DisabledHasNoEffect verifies that when the agent has no cache attached, every call goes through to the model. TestCache_StorageIsAStopHookBuiltin asserts the architectural contract that the cache_response storage is wired through the new hook system as a stop-hook builtin (not as a hard-co

(t *testing.T)

// a stop-hook builtin (not as a hard-coded runtime call). It is the
// regression test for the migration to the hooks mechanism.
func TestCache_StorageIsAStopHookBuiltin(t *testing.T) {
	t.Parallel()

	c, err := cache.New(cache.Config{Enabled: true})
	require.NoError(t, err)

	root := agent.New("root", "You are a test agent",
		agent.WithModel(&messageRecordingProvider{id: "test/mock"}),
		agent.WithCache(c),
	)
	tm := team.New(team.WithAgents(root))

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

	// The runtime should have auto-injected a stop hook of type=builtin
	// pointing at BuiltinCacheResponse for an agent that has a cache.
	exec := rt.hooksExec(root)
	require.NotNil(t, exec, "a cache-enabled agent must have a hooks executor")
	require.True(t, exec.Has(hooks.EventStop), "cache-enabled agent must have a stop hook")

	// The cache_response builtin must be registered on the runtime's
	// private hooks registry; that's the seam that lets the closure
	// reach back to a.Cache() through Input.AgentName.
	fn, ok := rt.hooksRegistry.LookupBuiltin(BuiltinCacheResponse)
	require.True(t, ok, "cache_response builtin must be registered")
	require.NotNil(t, fn)
}

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