SKYMirror
7f060dd0e4
Go 1.27.1 (Gin+GORM) + Vue 3 文件快传服务： - 安全审计全部修复（docs/security-audit-2026-09-05.md）： bcrypt 密码哈希与自动升级、presign 直传服务端大小/内容校验、 全局请求体上限、依赖升级（govulncheck 0 命中）、janitor 后台清理、 管理端审计动作落库、/admin CORS 收紧、通知内容白名单净化、 会话默认 7 天、限流缓存故障降级、robots.txt 端点等 - 前端：取件链接复制修复（不再重复拼接提取码）、markdown 净化器加固 - Redis 支持库号（FCB_REDIS_DB / redis://…/db URL） - 文档：docs/api/* 与 openapi.yaml 同步最新行为（robots.txt、 提码 5 位起、chunk 32MiB 上限、admin 审计动作等） 验证：gofmt/go vet/go test 全绿；二进制端到端冒烟通过
82 lines
1.8 KiB
Go
82 lines
1.8 KiB
Go
package cache
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestMemoryCacheSetGet(t *testing.T) {
|
|
c := NewMemory()
|
|
defer c.Close()
|
|
ctx := context.Background()
|
|
|
|
if err := c.Set(ctx, "k1", "v1", 0); err != nil {
|
|
t.Fatalf("Set 失败: %v", err)
|
|
}
|
|
v, err := c.Get(ctx, "k1")
|
|
if err != nil || v != "v1" {
|
|
t.Fatalf("Get = (%q, %v)", v, err)
|
|
}
|
|
if _, err := c.Get(ctx, "missing"); err != ErrNotFound {
|
|
t.Fatalf("缺失键应返回 ErrNotFound: %v", err)
|
|
}
|
|
_ = c.Delete(ctx, "k1")
|
|
if _, err := c.Get(ctx, "k1"); err != ErrNotFound {
|
|
t.Fatal("删除后应不存在")
|
|
}
|
|
}
|
|
|
|
func TestMemoryCacheTTL(t *testing.T) {
|
|
c := NewMemory()
|
|
defer c.Close()
|
|
ctx := context.Background()
|
|
_ = c.Set(ctx, "ttl", "x", 50*time.Millisecond)
|
|
if ok, _ := c.Exists(ctx, "ttl"); !ok {
|
|
t.Fatal("TTL 内应存在")
|
|
}
|
|
time.Sleep(80 * time.Millisecond)
|
|
if _, err := c.Get(ctx, "ttl"); err != ErrNotFound {
|
|
t.Fatal("过期后应不存在")
|
|
}
|
|
}
|
|
|
|
func TestMemoryCacheIncrWindow(t *testing.T) {
|
|
c := NewMemory()
|
|
defer c.Close()
|
|
ctx := context.Background()
|
|
for i := int64(1); i <= 3; i++ {
|
|
n, err := c.Incr(ctx, "rl", time.Minute)
|
|
if err != nil || n != i {
|
|
t.Fatalf("Incr = (%d, %v), want (%d, nil)", n, err, i)
|
|
}
|
|
}
|
|
// 窗口过期后重新计数
|
|
_ = c.Set(ctx, "short", "seed", time.Millisecond)
|
|
time.Sleep(5 * time.Millisecond)
|
|
n, err := c.Incr(ctx, "short", time.Millisecond)
|
|
if err != nil || n != 1 {
|
|
t.Fatalf("过期窗口重置失败: (%d, %v)", n, err)
|
|
}
|
|
}
|
|
|
|
func TestMemoryCacheConcurrentIncr(t *testing.T) {
|
|
c := NewMemory()
|
|
defer c.Close()
|
|
ctx := context.Background()
|
|
var wg sync.WaitGroup
|
|
for i := 0; i < 50; i++ {
|
|
wg.Add(1)
|
|
go func() {
|
|
defer wg.Done()
|
|
_, _ = c.Incr(ctx, "cnt", time.Minute)
|
|
}()
|
|
}
|
|
wg.Wait()
|
|
n, _ := c.Incr(ctx, "cnt", time.Minute)
|
|
if n != 51 {
|
|
t.Fatalf("并发计数丢失: %d != 51", n)
|
|
}
|
|
}
|