xfy
2734b04d8f
- Delete unused files: tempfile subsystem, matcher variants, server/internal - Remove 200+ unused functions across proxy, ssl, lua, http2/3, stream, variable - Fix proxy test type errors (backgroundRefresh ctx→Request) - Move bench/tools mock backend into internal/testutil - Remove corresponding test functions for all deleted code
120 lines
2.7 KiB
Go
120 lines
2.7 KiB
Go
package handler
|
|
|
|
import (
|
|
"container/list"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestFileInfoCache(t *testing.T) {
|
|
// 创建临时文件
|
|
tmpDir := t.TempDir()
|
|
tmpFile := filepath.Join(tmpDir, "test.txt")
|
|
if err := os.WriteFile(tmpFile, []byte("hello"), 0o644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
cache := &FileInfoCache{
|
|
entries: make(map[string]*fileInfoEntry, fileInfoCacheMaxEntries),
|
|
lruList: list.New(),
|
|
}
|
|
|
|
t.Run("缓存未命中", func(t *testing.T) {
|
|
info, ok := cache.Get(tmpFile)
|
|
if ok {
|
|
t.Error("未命中的缓存应返回 false")
|
|
}
|
|
if info != nil {
|
|
t.Error("未命中时应返回 nil")
|
|
}
|
|
})
|
|
|
|
t.Run("缓存命中", func(t *testing.T) {
|
|
// 先获取真实 FileInfo
|
|
realInfo, err := os.Stat(tmpFile)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// 存入缓存
|
|
cache.Set(tmpFile, realInfo)
|
|
|
|
// 从缓存获取
|
|
cachedInfo, ok := cache.Get(tmpFile)
|
|
if !ok {
|
|
t.Error("缓存应命中")
|
|
}
|
|
if cachedInfo == nil {
|
|
t.Fatal("缓存命中时应返回非 nil")
|
|
}
|
|
if cachedInfo.Name() != realInfo.Name() {
|
|
t.Errorf("Name() = %q, want %q", cachedInfo.Name(), realInfo.Name())
|
|
}
|
|
if cachedInfo.Size() != realInfo.Size() {
|
|
t.Errorf("Size() = %d, want %d", cachedInfo.Size(), realInfo.Size())
|
|
}
|
|
})
|
|
}
|
|
|
|
func TestFileInfoCacheTTL(t *testing.T) {
|
|
// 创建临时文件
|
|
tmpDir := t.TempDir()
|
|
tmpFile := filepath.Join(tmpDir, "test.txt")
|
|
if err := os.WriteFile(tmpFile, []byte("hello"), 0o644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
cache := &FileInfoCache{
|
|
entries: make(map[string]*fileInfoEntry, fileInfoCacheMaxEntries),
|
|
lruList: list.New(),
|
|
}
|
|
realInfo, _ := os.Stat(tmpFile)
|
|
|
|
// 存入缓存
|
|
cache.Set(tmpFile, realInfo)
|
|
|
|
// 立即获取应命中
|
|
_, ok := cache.Get(tmpFile)
|
|
if !ok {
|
|
t.Error("立即获取应命中")
|
|
}
|
|
|
|
// 模拟过期：修改 cachedAt
|
|
cache.mu.Lock()
|
|
if entry, exists := cache.entries[tmpFile]; exists {
|
|
entry.cachedAt = time.Now().Add(-fileInfoCacheTTL - time.Second)
|
|
}
|
|
cache.mu.Unlock()
|
|
|
|
// 过期后应未命中
|
|
_, ok = cache.Get(tmpFile)
|
|
if ok {
|
|
t.Error("过期后应未命中")
|
|
}
|
|
}
|
|
|
|
func TestFileInfoCacheLRU(t *testing.T) {
|
|
cache := &FileInfoCache{
|
|
entries: make(map[string]*fileInfoEntry, fileInfoCacheMaxEntries),
|
|
lruList: list.New(),
|
|
}
|
|
|
|
// 创建测试文件信息
|
|
tmpDir := t.TempDir()
|
|
for i := range fileInfoCacheMaxEntries + 10 {
|
|
tmpFile := filepath.Join(tmpDir, "test"+string(rune('0'+i%10))+".txt")
|
|
os.WriteFile(tmpFile, []byte("hello"), 0o644)
|
|
info, _ := os.Stat(tmpFile)
|
|
cache.Set(tmpFile, info)
|
|
}
|
|
|
|
cache.mu.RLock()
|
|
entries := len(cache.entries)
|
|
cache.mu.RUnlock()
|
|
if entries > fileInfoCacheMaxEntries {
|
|
t.Errorf("Entries = %d, should not exceed %d", entries, fileInfoCacheMaxEntries)
|
|
}
|
|
}
|