summaryrefslogtreecommitdiffstats
path: root/dependencies/pkg/mod/github.com/go-redis/redis/v8@v8.11.5/sentinel_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'dependencies/pkg/mod/github.com/go-redis/redis/v8@v8.11.5/sentinel_test.go')
-rw-r--r--dependencies/pkg/mod/github.com/go-redis/redis/v8@v8.11.5/sentinel_test.go287
1 files changed, 287 insertions, 0 deletions
diff --git a/dependencies/pkg/mod/github.com/go-redis/redis/v8@v8.11.5/sentinel_test.go b/dependencies/pkg/mod/github.com/go-redis/redis/v8@v8.11.5/sentinel_test.go
new file mode 100644
index 0000000..753e0fc
--- /dev/null
+++ b/dependencies/pkg/mod/github.com/go-redis/redis/v8@v8.11.5/sentinel_test.go
@@ -0,0 +1,287 @@
+package redis_test
+
+import (
+ "net"
+
+ . "github.com/onsi/ginkgo"
+ . "github.com/onsi/gomega"
+
+ "github.com/go-redis/redis/v8"
+)
+
+var _ = Describe("Sentinel", func() {
+ var client *redis.Client
+ var master *redis.Client
+ var masterPort string
+ var sentinel *redis.SentinelClient
+
+ BeforeEach(func() {
+ client = redis.NewFailoverClient(&redis.FailoverOptions{
+ MasterName: sentinelName,
+ SentinelAddrs: sentinelAddrs,
+ MaxRetries: -1,
+ })
+ Expect(client.FlushDB(ctx).Err()).NotTo(HaveOccurred())
+
+ sentinel = redis.NewSentinelClient(&redis.Options{
+ Addr: ":" + sentinelPort1,
+ MaxRetries: -1,
+ })
+
+ addr, err := sentinel.GetMasterAddrByName(ctx, sentinelName).Result()
+ Expect(err).NotTo(HaveOccurred())
+
+ master = redis.NewClient(&redis.Options{
+ Addr: net.JoinHostPort(addr[0], addr[1]),
+ MaxRetries: -1,
+ })
+ masterPort = addr[1]
+
+ // Wait until slaves are picked up by sentinel.
+ Eventually(func() string {
+ return sentinel1.Info(ctx).Val()
+ }, "15s", "100ms").Should(ContainSubstring("slaves=2"))
+ Eventually(func() string {
+ return sentinel2.Info(ctx).Val()
+ }, "15s", "100ms").Should(ContainSubstring("slaves=2"))
+ Eventually(func() string {
+ return sentinel3.Info(ctx).Val()
+ }, "15s", "100ms").Should(ContainSubstring("slaves=2"))
+ })
+
+ AfterEach(func() {
+ _ = client.Close()
+ _ = master.Close()
+ _ = sentinel.Close()
+ })
+
+ It("should facilitate failover", func() {
+ // Set value on master.
+ err := client.Set(ctx, "foo", "master", 0).Err()
+ Expect(err).NotTo(HaveOccurred())
+
+ // Verify.
+ val, err := client.Get(ctx, "foo").Result()
+ Expect(err).NotTo(HaveOccurred())
+ Expect(val).To(Equal("master"))
+
+ // Verify master->slaves sync.
+ var slavesAddr []string
+ Eventually(func() []string {
+ slavesAddr = redis.GetSlavesAddrByName(ctx, sentinel, sentinelName)
+ return slavesAddr
+ }, "15s", "100ms").Should(HaveLen(2))
+ Eventually(func() bool {
+ sync := true
+ for _, addr := range slavesAddr {
+ slave := redis.NewClient(&redis.Options{
+ Addr: addr,
+ MaxRetries: -1,
+ })
+ sync = slave.Get(ctx, "foo").Val() == "master"
+ _ = slave.Close()
+ }
+ return sync
+ }, "15s", "100ms").Should(BeTrue())
+
+ // Create subscription.
+ pub := client.Subscribe(ctx, "foo")
+ ch := pub.Channel()
+
+ // Kill master.
+ err = master.Shutdown(ctx).Err()
+ Expect(err).NotTo(HaveOccurred())
+ Eventually(func() error {
+ return master.Ping(ctx).Err()
+ }, "15s", "100ms").Should(HaveOccurred())
+
+ // Check that client picked up new master.
+ Eventually(func() string {
+ return client.Get(ctx, "foo").Val()
+ }, "15s", "100ms").Should(Equal("master"))
+
+ // Check if subscription is renewed.
+ var msg *redis.Message
+ Eventually(func() <-chan *redis.Message {
+ _ = client.Publish(ctx, "foo", "hello").Err()
+ return ch
+ }, "15s", "100ms").Should(Receive(&msg))
+ Expect(msg.Channel).To(Equal("foo"))
+ Expect(msg.Payload).To(Equal("hello"))
+ Expect(pub.Close()).NotTo(HaveOccurred())
+
+ _, err = startRedis(masterPort)
+ Expect(err).NotTo(HaveOccurred())
+ })
+
+ It("supports DB selection", func() {
+ Expect(client.Close()).NotTo(HaveOccurred())
+
+ client = redis.NewFailoverClient(&redis.FailoverOptions{
+ MasterName: sentinelName,
+ SentinelAddrs: sentinelAddrs,
+ DB: 1,
+ })
+ err := client.Ping(ctx).Err()
+ Expect(err).NotTo(HaveOccurred())
+ })
+})
+
+var _ = Describe("NewFailoverClusterClient", func() {
+ var client *redis.ClusterClient
+ var master *redis.Client
+ var masterPort string
+
+ BeforeEach(func() {
+ client = redis.NewFailoverClusterClient(&redis.FailoverOptions{
+ MasterName: sentinelName,
+ SentinelAddrs: sentinelAddrs,
+
+ RouteRandomly: true,
+ })
+ Expect(client.FlushDB(ctx).Err()).NotTo(HaveOccurred())
+
+ sentinel := redis.NewSentinelClient(&redis.Options{
+ Addr: ":" + sentinelPort1,
+ MaxRetries: -1,
+ })
+
+ addr, err := sentinel.GetMasterAddrByName(ctx, sentinelName).Result()
+ Expect(err).NotTo(HaveOccurred())
+
+ master = redis.NewClient(&redis.Options{
+ Addr: net.JoinHostPort(addr[0], addr[1]),
+ MaxRetries: -1,
+ })
+ masterPort = addr[1]
+
+ // Wait until slaves are picked up by sentinel.
+ Eventually(func() string {
+ return sentinel1.Info(ctx).Val()
+ }, "15s", "100ms").Should(ContainSubstring("slaves=2"))
+ Eventually(func() string {
+ return sentinel2.Info(ctx).Val()
+ }, "15s", "100ms").Should(ContainSubstring("slaves=2"))
+ Eventually(func() string {
+ return sentinel3.Info(ctx).Val()
+ }, "15s", "100ms").Should(ContainSubstring("slaves=2"))
+ })
+
+ AfterEach(func() {
+ _ = client.Close()
+ _ = master.Close()
+ })
+
+ It("should facilitate failover", func() {
+ // Set value.
+ err := client.Set(ctx, "foo", "master", 0).Err()
+ Expect(err).NotTo(HaveOccurred())
+
+ for i := 0; i < 100; i++ {
+ // Verify.
+ Eventually(func() string {
+ return client.Get(ctx, "foo").Val()
+ }, "15s", "1ms").Should(Equal("master"))
+ }
+
+ // Create subscription.
+ ch := client.Subscribe(ctx, "foo").Channel()
+
+ // Kill master.
+ err = master.Shutdown(ctx).Err()
+ Expect(err).NotTo(HaveOccurred())
+ Eventually(func() error {
+ return sentinelMaster.Ping(ctx).Err()
+ }, "15s", "100ms").Should(HaveOccurred())
+
+ // Check that client picked up new master.
+ Eventually(func() string {
+ return client.Get(ctx, "foo").Val()
+ }, "15s", "100ms").Should(Equal("master"))
+
+ // Check if subscription is renewed.
+ var msg *redis.Message
+ Eventually(func() <-chan *redis.Message {
+ _ = client.Publish(ctx, "foo", "hello").Err()
+ return ch
+ }, "15s", "100ms").Should(Receive(&msg))
+ Expect(msg.Channel).To(Equal("foo"))
+ Expect(msg.Payload).To(Equal("hello"))
+
+ _, err = startRedis(masterPort)
+ Expect(err).NotTo(HaveOccurred())
+ })
+})
+
+var _ = Describe("SentinelAclAuth", func() {
+ const (
+ aclSentinelUsername = "sentinel-user"
+ aclSentinelPassword = "sentinel-pass"
+ )
+
+ var client *redis.Client
+ var sentinel *redis.SentinelClient
+ var sentinels = func() []*redisProcess {
+ return []*redisProcess{sentinel1, sentinel2, sentinel3}
+ }
+
+ BeforeEach(func() {
+ authCmd := redis.NewStatusCmd(ctx, "ACL", "SETUSER", aclSentinelUsername, "ON",
+ ">"+aclSentinelPassword, "-@all", "+auth", "+client|getname", "+client|id", "+client|setname",
+ "+command", "+hello", "+ping", "+role", "+sentinel|get-master-addr-by-name", "+sentinel|master",
+ "+sentinel|myid", "+sentinel|replicas", "+sentinel|sentinels")
+
+ for _, process := range sentinels() {
+ err := process.Client.Process(ctx, authCmd)
+ Expect(err).NotTo(HaveOccurred())
+ }
+
+ client = redis.NewFailoverClient(&redis.FailoverOptions{
+ MasterName: sentinelName,
+ SentinelAddrs: sentinelAddrs,
+ MaxRetries: -1,
+ SentinelUsername: aclSentinelUsername,
+ SentinelPassword: aclSentinelPassword,
+ })
+
+ Expect(client.FlushDB(ctx).Err()).NotTo(HaveOccurred())
+
+ sentinel = redis.NewSentinelClient(&redis.Options{
+ Addr: sentinelAddrs[0],
+ MaxRetries: -1,
+ Username: aclSentinelUsername,
+ Password: aclSentinelPassword,
+ })
+
+ _, err := sentinel.GetMasterAddrByName(ctx, sentinelName).Result()
+ Expect(err).NotTo(HaveOccurred())
+
+ // Wait until sentinels are picked up by each other.
+ for _, process := range sentinels() {
+ Eventually(func() string {
+ return process.Info(ctx).Val()
+ }, "15s", "100ms").Should(ContainSubstring("sentinels=3"))
+ }
+ })
+
+ AfterEach(func() {
+ unauthCommand := redis.NewStatusCmd(ctx, "ACL", "DELUSER", aclSentinelUsername)
+
+ for _, process := range sentinels() {
+ err := process.Client.Process(ctx, unauthCommand)
+ Expect(err).NotTo(HaveOccurred())
+ }
+
+ _ = client.Close()
+ _ = sentinel.Close()
+ })
+
+ It("should still facilitate operations", func() {
+ err := client.Set(ctx, "wow", "acl-auth", 0).Err()
+ Expect(err).NotTo(HaveOccurred())
+
+ val, err := client.Get(ctx, "wow").Result()
+ Expect(err).NotTo(HaveOccurred())
+ Expect(val).To(Equal("acl-auth"))
+ })
+})
generated by cgit v1.2.3 (git 2.39.1) at 2024-06-03 07:49:08 +0000