-
Notifications
You must be signed in to change notification settings - Fork 350
/
redistest.go
77 lines (61 loc) · 2 KB
/
redistest.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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
package redistest
import (
"context"
"testing"
"time"
"github.com/go-redis/redis/v9"
"github.com/testcontainers/testcontainers-go"
"github.com/testcontainers/testcontainers-go/wait"
)
func NewTestRedis(t *testing.T) (address string, done func()) {
return NewTestRedisWithPassword(t, "")
}
func NewTestRedisWithPassword(t *testing.T, password string) (address string, done func()) {
var args []string
if password != "" {
args = append(args, "--requirepass", password)
}
start := time.Now()
// first testcontainer start takes longer than subsequent
ctx, cancel := context.WithTimeout(context.Background(), 20*time.Second)
defer cancel()
container, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{
ContainerRequest: testcontainers.ContainerRequest{
Image: "redis:6-alpine",
Cmd: args,
ExposedPorts: []string{"6379/tcp"},
WaitingFor: wait.ForLog("* Ready to accept connections"),
},
Started: true,
})
if err != nil {
t.Fatalf("Failed to start redis server: %v", err)
}
ctx, cancel = context.WithTimeout(context.Background(), 2*time.Second)
defer cancel()
address, err = container.Endpoint(ctx, "")
if err != nil {
t.Fatalf("Failed to get redis address: %v", err)
}
t.Logf("Started redis server at %s in %v", address, time.Since(start))
if err := ping(ctx, address, password); err != nil {
t.Fatalf("Failed to ping redis server: %v", err)
}
done = func() {
t.Logf("Stopping redis server at %s", address)
ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
defer cancel()
if err := container.Terminate(ctx); err != nil {
t.Fatalf("Failed to stop redis: %v", err)
}
}
return
}
func ping(ctx context.Context, address, password string) error {
rdb := redis.NewClient(&redis.Options{Addr: address, Password: password})
defer rdb.Close()
for _, err := rdb.Ping(ctx).Result(); ctx.Err() == nil && err != nil; _, err = rdb.Ping(ctx).Result() {
time.Sleep(100 * time.Millisecond)
}
return ctx.Err()
}