72 lines
1.5 KiB
Go
72 lines
1.5 KiB
Go
package cache
|
|
|
|
import (
|
|
"runtime"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestCache_Basic(t *testing.T) {
|
|
interval := 200 * time.Millisecond
|
|
ttl := 20 * time.Millisecond
|
|
c := New[string, int](interval)
|
|
c.Put("int", 1, ttl)
|
|
|
|
d := New[string, string](interval)
|
|
d.Put("string", "a", ttl)
|
|
|
|
i := c.Get("int")
|
|
assert.Equal(t, i, 1, "should recv 1")
|
|
|
|
s := d.Get("string")
|
|
assert.Equal(t, s, "a", "should recv 'a'")
|
|
}
|
|
|
|
func TestCache_TTL(t *testing.T) {
|
|
interval := 200 * time.Millisecond
|
|
ttl := 20 * time.Millisecond
|
|
now := time.Now()
|
|
c := New[string, int](interval)
|
|
c.Put("int", 1, ttl)
|
|
c.Put("int2", 2, ttl)
|
|
|
|
i := c.Get("int")
|
|
_, expired := c.GetWithExpire("int2")
|
|
assert.Equal(t, i, 1, "should recv 1")
|
|
assert.True(t, now.Before(expired))
|
|
|
|
time.Sleep(ttl * 2)
|
|
i = c.Get("int")
|
|
j, _ := c.GetWithExpire("int2")
|
|
assert.True(t, i == 0, "should recv 0")
|
|
assert.True(t, j == 0, "should recv 0")
|
|
}
|
|
|
|
func TestCache_AutoCleanup(t *testing.T) {
|
|
interval := 10 * time.Millisecond
|
|
ttl := 15 * time.Millisecond
|
|
c := New[string, int](interval)
|
|
c.Put("int", 1, ttl)
|
|
|
|
time.Sleep(ttl * 2)
|
|
i := c.Get("int")
|
|
j, _ := c.GetWithExpire("int")
|
|
assert.True(t, i == 0, "should recv 0")
|
|
assert.True(t, j == 0, "should recv 0")
|
|
}
|
|
|
|
func TestCache_AutoGC(t *testing.T) {
|
|
sign := make(chan struct{})
|
|
go func() {
|
|
interval := 10 * time.Millisecond
|
|
ttl := 15 * time.Millisecond
|
|
c := New[string, int](interval)
|
|
c.Put("int", 1, ttl)
|
|
sign <- struct{}{}
|
|
}()
|
|
|
|
<-sign
|
|
runtime.GC()
|
|
}
|