go-pkg/asynq
1
0
Fork 0
mirror of https://github.com/hibiken/asynq.git synced 2025-09-19 13:21:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add benchmark tests for rdb package

Browse Source
This commit is contained in:
Ken Hibino
2021-03-05 15:20:56 -08:00
parent aa936466b3
commit f618f5b1f5
2 changed files with 270 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
internal/rdb/rdb_test.go
View File

@@ -37,12 +37,12 @@ func init() {
flag.StringVar(&redisClusterAddrs, "redis_cluster_addrs", "localhost:7000,localhost:7001,localhost:7002", "comma separated list of redis server addresses")
}
func setup(t *testing.T) (r *RDB) {
t.Helper()
func setup(tb testing.TB) (r *RDB) {
tb.Helper()
if useRedisCluster {
addrs := strings.Split(redisClusterAddrs, ",")
if len(addrs) == 0 {
t.Fatal("No redis cluster addresses provided. Please set addresses using --redis_cluster_addrs flag.")
tb.Fatal("No redis cluster addresses provided. Please set addresses using --redis_cluster_addrs flag.")
}
r = NewRDB(redis.NewClusterClient(&redis.ClusterOptions{
Addrs: addrs,
@@ -54,7 +54,7 @@ func setup(t *testing.T) (r *RDB) {
}))
}
// Start each test with a clean slate.
h.FlushDB(t, r.client)
h.FlushDB(tb, r.client)
return r
}