跳转到主内容
极星编程网:以代码为星,赴技术山海!

如何在Golang中测试缓存操作_验证存取和过期逻辑

Go中测试缓存需接口抽象、可控时间与内存实现:定义Cache接口,注入nowFunc控制过期逻辑,用MemCache+sync.RWMutex实现可测内存缓存,覆盖存取一致性、TTL刷新、并发安全等场景。 在 Go 中测试缓存操作,核心是隔离外部依赖、控制时间、验证行为是否符合预期——尤其是存取一致性与过期逻辑。不需要真实 Redis 或内存缓存实例,用可 mock 的接口 + 可控时钟就能覆盖关键路径。 用接口抽象缓存,便于替换实现 定义统一缓存接口,让业务代码不绑定具体实现:
type Cache interface { Set(key string, value interface{}, ttl time.Duration) error Get(key string, dst interface{}) error Delete(key string) error }
这样测试时可用内存缓存(如
map
+
sync.RWMutex
)或专为测试设计的
FakeCache
,甚至注入带时间戳记录的调试版。 手动控制时间验证过期逻辑 Go 标准库 的
time.Now
无法直接 mock,推荐用依赖注入方式传入
time.Now
函数: 立即学习 “ go语言免费学习笔记(深入) ”; Python3.6标准库参考手册(PDF版) Python3.6标准库参考手册(PDF版) 下载 缓存实现中接收
nowFunc func() time.Time
参数 测试时传入固定或可进阶的时间函数,例如:
now := time.Date(2024, 1, 1, 12, 0, 0, 0, time.UTC)
nowFunc := func() time.Time { return now }
再调用
Set("k", "v", 5*time.Second)
,然后把
now
推进到
+6s
后调用
Get
,应返回缓存未命中错误 覆盖典型场景的测试用例 每个测试聚焦一个行为,命名体现意图(如
TestCache_GetReturnsValueWhenNotExpired
): 存后即取能命中 :Set → Get,检查值相等、无错误 过期后 Get 失败 :Set(ttl=1s) → sleep 1.1s → Get,检查返回
cache.ErrNotFound
或类似语义错误 Set 同 key 覆盖旧值和 TTL :Set(k,v1,1s) → wait 0.5s → Set(k,v2,1s) → wait 0.6s → Get,应得 v2(而非因首次过期失败) 并发读写安全 :用
sync.WaitGroup
启多个 goroutine 同时 Set/Get 同 key,不 panic、结果一致 避免真实依赖,用内存实现做单元测试 写一个轻量
MemCache
用于测试(非生产用),自带时间感知:
type MemCache struct { mu sync.RWMutex items map[string]cacheItem nowFn func() time.Time } type cacheItem struct { value interface{} exp time.Time } func (m *MemCache) Set(key string, value interface{}, ttl time.Duration) error { m.mu.Lock() defer m.mu.Unlock() m.items[key] = cacheItem{ value: value, exp: m.nowFn().Add(ttl), } return nil } func (m *MemCache) Get(key string, dst interface{}) error { m.mu.RLock() defer m.mu.RUnlock() item, ok := m.items[key] if !ok || m.nowFn().After(item.exp) { return errors.New("not found or expired") } // 类型拷贝逻辑(简化版) reflect.ValueOf(dst).Elem().Set(reflect.ValueOf(item.value)) return nil }
测试时初始化它并传入可控
nowFn
,所有时间敏感逻辑都变得确定可断言。

相关文章