mirror of
https://github.com/woodchen-ink/proxy-go.git
synced 2025-07-18 08:31:55 +08:00
wood chen
5f67325055
- Update cache expiration check to use LastAccess instead of CreatedAt - Reset expiration time on each cache item access - Enhance cache management with more accurate access tracking
65 lines
1.7 KiB
Go
65 lines
1.7 KiB
Go
package cache
|
||
|
||
import (
|
||
"net/http"
|
||
"os"
|
||
"testing"
|
||
"time"
|
||
)
|
||
|
||
func TestCacheExpiry(t *testing.T) {
|
||
// 创建临时目录用于测试
|
||
tempDir, err := os.MkdirTemp("", "cache-test-*")
|
||
if err != nil {
|
||
t.Fatal("Failed to create temp dir:", err)
|
||
}
|
||
defer os.RemoveAll(tempDir)
|
||
|
||
// 创建缓存管理器，设置较短的过期时间（5秒）用于测试
|
||
cm, err := NewCacheManager(tempDir)
|
||
if err != nil {
|
||
t.Fatal("Failed to create cache manager:", err)
|
||
}
|
||
cm.maxAge = 5 * time.Second
|
||
|
||
// 创建测试请求和响应
|
||
req, _ := http.NewRequest("GET", "http://example.com/test", nil)
|
||
resp := &http.Response{
|
||
StatusCode: http.StatusOK,
|
||
Header: make(http.Header),
|
||
}
|
||
testData := []byte("test data")
|
||
|
||
// 生成缓存键
|
||
key := cm.GenerateCacheKey(req)
|
||
|
||
// 1. 首先放入缓存
|
||
_, err = cm.Put(key, resp, testData)
|
||
if err != nil {
|
||
t.Fatal("Failed to put item in cache:", err)
|
||
}
|
||
|
||
// 2. 立即获取，应该能命中
|
||
if _, hit, _ := cm.Get(key, req); !hit {
|
||
t.Error("Cache should hit immediately after putting")
|
||
}
|
||
|
||
// 3. 等待3秒（未过期），再次访问
|
||
time.Sleep(3 * time.Second)
|
||
if _, hit, _ := cm.Get(key, req); !hit {
|
||
t.Error("Cache should hit after 3 seconds")
|
||
}
|
||
|
||
// 4. 再等待3秒（总共6秒，但因为上次访问重置了时间，所以应该还在有效期内）
|
||
time.Sleep(3 * time.Second)
|
||
if _, hit, _ := cm.Get(key, req); !hit {
|
||
t.Error("Cache should hit after 6 seconds because last access reset the timer")
|
||
}
|
||
|
||
// 5. 等待6秒（超过过期时间且无访问），这次应该过期
|
||
time.Sleep(6 * time.Second)
|
||
if _, hit, _ := cm.Get(key, req); hit {
|
||
t.Error("Cache should expire after 6 seconds of no access")
|
||
}
|
||
}
|