internal/cache/kv_store_bbolt_test.go

Rama main · réplica de lectura del repositorio original.

package cache

import (
	"bytes"
	"context"
	"path/filepath"
	"testing"
	"time"

	"github.com/MoeForge/moego/ojson"
	"github.com/router-for-me/CLIProxyAPI/v8/internal/config"
	homekv "github.com/router-for-me/CLIProxyAPI/v8/internal/home"
	"github.com/router-for-me/CLIProxyAPI/v8/internal/kvstore"
)

func TestKVStoreBBoltPersistenceAndRouting(t *testing.T) {
	homekv.ClearCurrent()
	_ = kvstore.ShutdownLocal()
	ClearSignatureCache("")
	ClearCodexReasoningReplayCache()
	ClearClaudeWebSearchResultCache()
	t.Cleanup(func() {
		homekv.ClearCurrent()
		_ = kvstore.ShutdownLocal()
		ClearSignatureCache("")
		ClearCodexReasoningReplayCache()
		ClearClaudeWebSearchResultCache()
	})

	dbPath := filepath.Join(t.TempDir(), "kv.db")
	if errCfg := kvstore.ConfigureLocal(true, dbPath); errCfg != nil {
		t.Fatalf("kvstore.ConfigureLocal(true, %q) error = %v", dbPath, errCfg)
	}

	// 1. Signature cache persists to bbolt when in-memory cache is cleared.
	sigText := "thinking text persisted in bbolt"
	sigVal := "bboltValidSignature1234567890123456789012345678901234567890"
	CacheSignature(testModelName, sigText, sigVal)
	ClearSignatureCache("")
	if gotSig := GetCachedSignature(testModelName, sigText); gotSig != sigVal {
		t.Fatalf("GetCachedSignature() after ClearSignatureCache = %q, want %q", gotSig, sigVal)
	}

	// 2. Codex reasoning replay items persist to bbolt (including CAS append) when in-memory map is cleared.
	turn1 := [][]byte{
		[]byte(`{"type":"` + CodexReasoningReplayTurnType + `","id":"turn-1","assistant_fingerprint":"answer-1"}`),
		validCodexReasoningReplayItemForTest(21),
	}
	turn2 := [][]byte{
		[]byte(`{"type":"` + CodexReasoningReplayTurnType + `","id":"turn-2","call_ids":["call-2"]}`),
		validCodexReasoningReplayItemForTest(22),
	}
	if !CacheCodexReasoningReplayItems("gpt-5.4", "session-bbolt", turn1) {
		t.Fatalf("CacheCodexReasoningReplayItems() = false, want true")
	}
	if !AppendCodexReasoningReplayItemsBestEffort(context.Background(), "gpt-5.4", "session-bbolt", turn2) {
		t.Fatalf("AppendCodexReasoningReplayItemsBestEffort() = false, want true")
	}
	ClearCodexReasoningReplayCache()
	replayItems, foundReplay := GetCodexReasoningReplayItems("gpt-5.4", "session-bbolt")
	if !foundReplay || len(replayItems) != 4 {
		t.Fatalf("GetCodexReasoningReplayItems() after ClearCodexReasoningReplayCache = (len=%d, found=%v), want (len=4, found=true)", len(replayItems), foundReplay)
	}
	if gotID1 := ojson.GetBytes(replayItems[0], "id").String(); gotID1 != "turn-1" {
		t.Fatalf("replayItems[0] id = %q, want turn-1", gotID1)
	}
	if gotID2 := ojson.GetBytes(replayItems[2], "id").String(); gotID2 != "turn-2" {
		t.Fatalf("replayItems[2] id = %q, want turn-2", gotID2)
	}

	// 3. Claude web search result cache persists to bbolt when in-memory cache is cleared.
	webContent := []byte(`[{"type":"web_search_result","url":"https://example.com/bbolt"}]`)
	CacheClaudeWebSearchResult("session-bbolt", "srvtoolu_bbolt", webContent, 30*time.Minute, 16)
	ClearClaudeWebSearchResultCache()
	gotWeb, foundWeb := GetClaudeWebSearchResult("session-bbolt", "srvtoolu_bbolt", 30*time.Minute)
	if !foundWeb || !bytes.Equal(gotWeb, webContent) {
		t.Fatalf("GetClaudeWebSearchResult() after ClearClaudeWebSearchResultCache = (%q, %v), want (%q, true)", gotWeb, foundWeb, webContent)
	}

	// 4. When Home mode is active (home.SetCurrent), Home KV takes priority over local bbolt.
	homeClient := homekv.New(config.HomeConfig{Enabled: true, Host: "127.0.0.1", Port: 1})
	homekv.SetCurrent(homeClient)
	if gotHomeSig := GetCachedSignature(testModelName, sigText); gotHomeSig != "" {
		t.Fatalf("GetCachedSignature() in Home mode = %q, want empty (must not read local bbolt)", gotHomeSig)
	}
	if _, errHomeSig := GetCachedSignatureRequired(context.Background(), testModelName, sigText); errHomeSig == nil {
		t.Fatalf("GetCachedSignatureRequired() in unavailable Home mode = nil error, want Home unavailable error")
	}
	homeOnlyText := "thinking text attempted while Home active"
	CacheSignature(testModelName, homeOnlyText, sigVal)
	homekv.ClearCurrent()

	// After clearing Home client, bbolt is active again: original key is still readable, Home-only write did not touch bbolt.
	if gotAfterHome := GetCachedSignature(testModelName, sigText); gotAfterHome != sigVal {
		t.Fatalf("GetCachedSignature() after ClearCurrent = %q, want %q", gotAfterHome, sigVal)
	}
	if gotHomeOnly := GetCachedSignature(testModelName, homeOnlyText); gotHomeOnly != "" {
		t.Fatalf("GetCachedSignature(homeOnlyText) = %q, want empty", gotHomeOnly)
	}

	// 5. Disabling local bbolt KV restores standard in-memory cache behavior.
	if errDisable := kvstore.ConfigureLocal(false, ""); errDisable != nil {
		t.Fatalf("kvstore.ConfigureLocal(false, \"\") error = %v", errDisable)
	}
	if gotAfterDisable := GetCachedSignature(testModelName, sigText); gotAfterDisable != "" {
		t.Fatalf("GetCachedSignature() after disabling bbolt = %q, want empty in-memory cache", gotAfterDisable)
	}
	memText := "thinking text in memory only"
	CacheSignature(testModelName, memText, sigVal)
	if gotMem := GetCachedSignature(testModelName, memText); gotMem != sigVal {
		t.Fatalf("GetCachedSignature() in memory mode = %q, want %q", gotMem, sigVal)
	}
	ClearSignatureCache("")
	if gotCleared := GetCachedSignature(testModelName, memText); gotCleared != "" {
		t.Fatalf("GetCachedSignature() after ClearSignatureCache in memory mode = %q, want empty", gotCleared)
	}
}