Function TestCachedTool_DifferentInputs

pkg/tools/cache_test.go:431–476

(t *testing.T)

}

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

	cache := NewToolCache(config)

	mockTool := &MockTool{
		name:        "test_tool",
		description: "Test tool",
	}

	cachedTool := NewCachedTool(mockTool, cache)

	ctx := context.Background()

	// 不同的输入应该执行不同的调用
	input1 := map[string]any{"test": "value1"}
	input2 := map[string]any{"test": "value2"}

	_, err := cachedTool.Execute(ctx, input1, nil)
	if err != nil {
		t.Fatalf("Failed to execute: %v", err)
	}

	_, err = cachedTool.Execute(ctx, input2, nil)
	if err != nil {
		t.Fatalf("Failed to execute: %v", err)
	}

	if mockTool.callCount != 2 {
		t.Errorf("Expected 2 calls for different inputs, got: %d", mockTool.callCount)
	}

	// 相同输入应该使用缓存
	_, err = cachedTool.Execute(ctx, input1, nil)
	if err != nil {
		t.Fatalf("Failed to execute: %v", err)
	}

	if mockTool.callCount != 2 {
		t.Errorf("Expected 2 calls (third was cached), got: %d", mockTool.callCount)
	}
}

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