@@ -13,7 +13,6 @@ import (
1313
1414 . "github.com/bsm/ginkgo/v2"
1515 . "github.com/bsm/gomega"
16-
1716 "github.com/redis/go-redis/v9"
1817)
1918
@@ -28,7 +27,7 @@ const (
2827)
2928
3029const (
31- sentinelName = "mymaster "
30+ sentinelName = "go-redis-test "
3231 sentinelMasterPort = "9123"
3332 sentinelSlave1Port = "9124"
3433 sentinelSlave2Port = "9125"
4342)
4443
4544var (
46- rediStackPort = "6379"
47- rediStackAddr = ":" + rediStackPort
45+ redisStackPort = "6379"
46+ redisStackAddr = ":" + redisStackPort
4847)
4948
5049var (
@@ -59,14 +58,17 @@ var (
5958)
6059
6160var cluster = & clusterScenario {
62- ports : []string {"8220 " , "8221 " , "8222 " , "8223 " , "8224 " , "8225 " },
61+ ports : []string {"16600 " , "16601 " , "16602 " , "16603 " , "16604 " , "16605 " },
6362 nodeIDs : make ([]string , 6 ),
6463 processes : make (map [string ]* redisProcess , 6 ),
6564 clients : make (map [string ]* redis.Client , 6 ),
6665}
6766
67+ // Redis Software Cluster
6868var RECluster = false
69- var USE_CONTAINERIZED_REDIS = false
69+
70+ // Redis Community Edition Docker
71+ var RCEDocker = false
7072
7173func registerProcess (port string , p * redisProcess ) {
7274 if processes == nil {
@@ -83,8 +85,10 @@ var _ = BeforeSuite(func() {
8385 }
8486 var err error
8587 RECluster , _ = strconv .ParseBool (os .Getenv ("RE_CLUSTER" ))
86- USE_CONTAINERIZED_REDIS , _ = strconv .ParseBool (os .Getenv ("USE_CONTAINERIZED_REDIS" ))
87- if ! RECluster || ! USE_CONTAINERIZED_REDIS {
88+ RCEDocker , _ = strconv .ParseBool (os .Getenv ("RCE_DOCKER" ))
89+ fmt .Printf ("RECluster: %v\n " , RECluster )
90+ fmt .Printf ("RCEDocker: %v\n " , RCEDocker )
91+ if ! RECluster && ! RCEDocker {
8892
8993 redisMain , err = startRedis (redisPort )
9094 Expect (err ).NotTo (HaveOccurred ())
@@ -121,18 +125,24 @@ var _ = BeforeSuite(func() {
121125 err = startCluster (ctx , cluster )
122126 Expect (err ).NotTo (HaveOccurred ())
123127 } else {
124- redisPort = rediStackPort
125- redisAddr = rediStackAddr
128+ redisPort = redisStackPort
129+ redisAddr = redisStackAddr
130+
131+ if ! RECluster {
132+ // populate cluster node information
133+ Expect (configureClusterTopology (ctx , cluster )).NotTo (HaveOccurred ())
134+ }
126135 }
127136})
128137
129138var _ = AfterSuite (func () {
130139 if ! RECluster {
131140 Expect (cluster .Close ()).NotTo (HaveOccurred ())
141+ }
132142
133- for _ , p := range processes {
134- Expect ( p . Close ()). NotTo ( HaveOccurred ())
135- }
143+ // NOOP if there are no processes registered
144+ for _ , p := range processes {
145+ Expect ( p . Close ()). NotTo ( HaveOccurred ())
136146 }
137147 processes = nil
138148})
@@ -156,8 +166,8 @@ func redisOptions() *redis.Options {
156166 ContextTimeoutEnabled : true ,
157167
158168 MaxRetries : - 1 ,
159- PoolSize : 10 ,
160169
170+ PoolSize : 10 ,
161171 PoolTimeout : 30 * time .Second ,
162172 ConnMaxIdleTime : time .Minute ,
163173 }
0 commit comments