Golang缓存机制测试方法详解

type InMemoryCache struct {
    data sync.Map
}
<p>func (c *InMemoryCache) Set(key string, value interface{}) {
c.data.Store(key, value)
}</p><p>func (c *InMemoryCache) Get(key string) (interface{}, bool) {
return c.data.Load(key)
}</p>

func TestInMemoryCache_SetAndGet(t *testing.T) {
    cache := &InMemoryCache{}
    cache.Set("user:123", "alice")
<pre class="brush:php;toolbar:false;">value, ok := cache.Get("user:123")
if !ok {
    t.Error("expected key to exist")
}
if value != "alice" {
    t.Errorf("got %v, want alice", value)
}

}

func TestCache_Expiration(t *testing.T) {
    cache := go_cache.New(50*time.Millisecond, 1*time.Second)
<pre class="brush:php;toolbar:false;">cache.Set("temp", "data", go_cache.DefaultExpiration)
time.Sleep(60 * time.Millisecond) // 超出有效期

_, found := cache.Get("temp")
if found {
    t.Error("expected entry to be expired")
}

}

func TestCache_ConcurrentAccess(t *testing.T) {
    cache := &InMemoryCache{}
    var wg sync.WaitGroup
<pre class="brush:php;toolbar:false;">for i := 0; i < 100; i++ {
    wg.Add(1)
    go func(i int) {
        defer wg.Done()
        key := fmt.Sprintf("key-%d", i%10)
        cache.Set(key, i)
        _, _ = cache.Get(key)
    }(i)
}

wg.Wait()

}

type Cache interface {
    Get(key string) (interface{}, bool)
    Set(key string, val interface{})
}

type MockCache struct {
    data map[string]interface{}
}
<p>func (m *MockCache) Get(key string) (interface{}, bool) {
v, ok := m.data[key]
return v, ok
}</p><p>func (m *MockCache) Set(key string, val interface{}) {
m.data[key] = val
}</p>