-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathgo_cache_test.go
50 lines (43 loc) · 1.41 KB
/
go_cache_test.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
package cacheit
import (
"context"
"math/rand"
"testing"
"time"
gocache "github.com/patrickmn/go-cache"
"github.com/spf13/cast"
)
func setupGoCacheDriver[V any](t *testing.T) *GoCacheDriver[V] {
memCache := gocache.New(5*time.Minute, 10*time.Minute)
driverIndex = driverIndex + 1
driverName := "mem_test" + cast.ToString(driverIndex)
rand.Seed(time.Now().UnixNano())
prefix := "cache_prefix"
if rand.Intn(2) > 0 {
prefix = ""
}
err := RegisterGoCacheDriver(driverName, memCache, prefix)
if err != nil {
t.Fatalf("Failed register redisDriver: %v", err)
return nil
}
driver, err := Use[V](driverName)
if err != nil {
t.Fatalf("Failed to setup RedisDriver: %v", err)
return nil
}
driver.WithCtx(context.Background())
driver.WithSerializer(&JSONSerializer{})
return driver.(*GoCacheDriver[V])
}
func TestGoCacheDriver(t *testing.T) {
goCacheDriver := setupGoCacheDriver[string](t)
testCache[string](t, goCacheDriver, "test_string_key", "test_string_value")
testNumberCache[string](t, goCacheDriver, "test_string_key", "test_string_value")
goCacheDriverInt := setupGoCacheDriver[int](t)
testNumberCache[int](t, goCacheDriverInt, "test_int_key", 2)
goCacheDriverUint := setupGoCacheDriver[uint](t)
testNumberCache[uint](t, goCacheDriverUint, "test_uint_key", uint(2))
goCacheDriverFloat := setupGoCacheDriver[float32](t)
testNumberCache[float32](t, goCacheDriverFloat, "test_float_key", float32(2.0))
}