diff --git a/cluster_test.go b/cluster_test.go index a64a0f6a..e94a5099 100644 --- a/cluster_test.go +++ b/cluster_test.go @@ -49,21 +49,51 @@ func (s *clusterScenario) addrs() []string { } func (s *clusterScenario) clusterClient(opt *redis.ClusterOptions) *redis.ClusterClient { + var errBadState = fmt.Errorf("cluster state is not consistent") + opt.Addrs = s.addrs() client := redis.NewClusterClient(opt) + err := eventually(func() error { - state, err := client.GetState() + if opt.ClusterSlots != nil { + return nil + } + + state, err := client.LoadState() if err != nil { return err } + if !state.IsConsistent() { - return fmt.Errorf("cluster state is not conistent") + return errBadState } + + if len(state.Masters) < 3 { + return errBadState + } + for _, master := range state.Masters { + s := master.Client.Info("replication").Val() + if !strings.Contains(s, "role:master") { + return errBadState + } + } + + if len(state.Slaves) < 3 { + return errBadState + } + for _, slave := range state.Slaves { + s := slave.Client.Info("replication").Val() + if !strings.Contains(s, "role:slave") { + return errBadState + } + } + return nil }, 30*time.Second) if err != nil { panic(err) } + return client } @@ -703,7 +733,7 @@ var _ = Describe("ClusterClient", func() { }) Expect(err).NotTo(HaveOccurred()) - state, err := client.GetState() + state, err := client.LoadState() Expect(err).NotTo(HaveOccurred()) Expect(state.IsConsistent()).To(BeTrue()) diff --git a/export_test.go b/export_test.go index 060df102..e9afda93 100644 --- a/export_test.go +++ b/export_test.go @@ -21,10 +21,6 @@ func (c *PubSub) ReceiveMessageTimeout(timeout time.Duration) (*Message, error) return c.receiveMessage(timeout) } -func (c *ClusterClient) GetState() (*clusterState, error) { - return c.state.Get() -} - func (c *ClusterClient) LoadState() (*clusterState, error) { return c.loadState() }