Function TestCachedTool_Execute

pkg/tools/cache_test.go:375–429

(t *testing.T)

}

func TestCachedTool_Execute(t *testing.T) {
	config := &CacheConfig{
		Enabled:  true,
		Strategy: CacheStrategyMemory,
		TTL:      1 * time.Hour,
	}

	cache := NewToolCache(config)

	// 创建 mock 工具
	mockTool := &MockTool{
		name:        "test_tool",
		description: "Test tool",
	}

	// 创建带缓存的工具
	cachedTool := NewCachedTool(mockTool, cache)

	ctx := context.Background()
	input := map[string]any{"test": "value"}

	// 第一次执行
	result1, err := cachedTool.Execute(ctx, input, nil)
	if err != nil {
		t.Fatalf("Failed to execute: %v", err)
	}

	if mockTool.callCount != 1 {
		t.Errorf("Expected 1 call, got: %d", mockTool.callCount)
	}

	// 第二次执行（应该使用缓存）
	result2, err := cachedTool.Execute(ctx, input, nil)
	if err != nil {
		t.Fatalf("Failed to execute: %v", err)
	}

	if mockTool.callCount != 1 {
		t.Errorf("Expected 1 call (cached), got: %d", mockTool.callCount)
	}

	// 验证结果相同
	result1Map := result1.(map[string]any)
	result2Map := result2.(map[string]any)

	if result1Map["result"] != result2Map["result"] {
		t.Error("Expected same result from cache")
	}

	// 检查缓存统计
	stats := cache.GetStats()
	if stats.Hits != 1 {
		t.Errorf("Expected 1 cache hit, got: %d", stats.Hits)
	}
}

func TestCachedTool_DifferentInputs(t *testing.T) {
	config := &CacheConfig{