forked from mirror/redis
96 lines
2.9 KiB
Go
96 lines
2.9 KiB
Go
|
package redis
|
||
|
|
|
||
|
|
import (
|
||
|
|
. "github.com/onsi/ginkgo"
|
||
|
|
. "github.com/onsi/gomega"
|
||
|
|
)
|
||
|
|
|
||
|
|
var _ = Describe("ClusterClient", func() {
|
||
|
|
|
||
|
|
var subject *ClusterClient
|
||
|
|
var populate = func() {
|
||
|
|
subject.reset()
|
||
|
|
subject.update([]ClusterSlotInfo{
|
||
|
|
{0, 4095, []string{"127.0.0.1:7000", "127.0.0.1:7004"}},
|
||
|
|
{12288, 16383, []string{"127.0.0.1:7003", "127.0.0.1:7007"}},
|
||
|
|
{4096, 8191, []string{"127.0.0.1:7001", "127.0.0.1:7005"}},
|
||
|
|
{8192, 12287, []string{"127.0.0.1:7002", "127.0.0.1:7006"}},
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
BeforeEach(func() {
|
||
|
|
var err error
|
||
|
|
subject, err = NewClusterClient(&ClusterOptions{
|
||
|
|
Addrs: []string{"127.0.0.1:6379", "127.0.0.1:7003", "127.0.0.1:7006"},
|
||
|
|
})
|
||
|
|
Expect(err).NotTo(HaveOccurred())
|
||
|
|
})
|
||
|
|
|
||
|
|
AfterEach(func() {
|
||
|
|
subject.Close()
|
||
|
|
})
|
||
|
|
|
||
|
|
It("should initialize", func() {
|
||
|
|
Expect(subject.addrs).To(HaveLen(3))
|
||
|
|
Expect(subject.slots).To(HaveLen(hashSlots))
|
||
|
|
Expect(subject._reload).To(Equal(uint32(0)))
|
||
|
|
})
|
||
|
|
|
||
|
|
It("should update slots cache", func() {
|
||
|
|
populate()
|
||
|
|
Expect(subject.slots[0]).To(Equal([]string{"127.0.0.1:7000", "127.0.0.1:7004"}))
|
||
|
|
Expect(subject.slots[4095]).To(Equal([]string{"127.0.0.1:7000", "127.0.0.1:7004"}))
|
||
|
|
Expect(subject.slots[4096]).To(Equal([]string{"127.0.0.1:7001", "127.0.0.1:7005"}))
|
||
|
|
Expect(subject.slots[8191]).To(Equal([]string{"127.0.0.1:7001", "127.0.0.1:7005"}))
|
||
|
|
Expect(subject.slots[8192]).To(Equal([]string{"127.0.0.1:7002", "127.0.0.1:7006"}))
|
||
|
|
Expect(subject.slots[12287]).To(Equal([]string{"127.0.0.1:7002", "127.0.0.1:7006"}))
|
||
|
|
Expect(subject.slots[12288]).To(Equal([]string{"127.0.0.1:7003", "127.0.0.1:7007"}))
|
||
|
|
Expect(subject.slots[16383]).To(Equal([]string{"127.0.0.1:7003", "127.0.0.1:7007"}))
|
||
|
|
Expect(subject.addrs).To(Equal(map[string]struct{}{
|
||
|
|
"127.0.0.1:6379": struct{}{},
|
||
|
|
"127.0.0.1:7000": struct{}{},
|
||
|
|
"127.0.0.1:7001": struct{}{},
|
||
|
|
"127.0.0.1:7002": struct{}{},
|
||
|
|
"127.0.0.1:7003": struct{}{},
|
||
|
|
"127.0.0.1:7004": struct{}{},
|
||
|
|
"127.0.0.1:7005": struct{}{},
|
||
|
|
"127.0.0.1:7006": struct{}{},
|
||
|
|
"127.0.0.1:7007": struct{}{},
|
||
|
|
}))
|
||
|
|
})
|
||
|
|
|
||
|
|
It("should find next addresses", func() {
|
||
|
|
populate()
|
||
|
|
seen := map[string]struct{}{
|
||
|
|
"127.0.0.1:7000": struct{}{},
|
||
|
|
"127.0.0.1:7001": struct{}{},
|
||
|
|
"127.0.0.1:7003": struct{}{},
|
||
|
|
}
|
||
|
|
|
||
|
|
addr := subject.findNextAddr(seen)
|
||
|
|
for addr != "" {
|
||
|
|
seen[addr] = struct{}{}
|
||
|
|
addr = subject.findNextAddr(seen)
|
||
|
|
}
|
||
|
|
Expect(subject.findNextAddr(seen)).To(Equal(""))
|
||
|
|
Expect(seen).To(Equal(map[string]struct{}{
|
||
|
|
"127.0.0.1:6379": struct{}{},
|
||
|
|
"127.0.0.1:7000": struct{}{},
|
||
|
|
"127.0.0.1:7001": struct{}{},
|
||
|
|
"127.0.0.1:7002": struct{}{},
|
||
|
|
"127.0.0.1:7003": struct{}{},
|
||
|
|
"127.0.0.1:7004": struct{}{},
|
||
|
|
"127.0.0.1:7005": struct{}{},
|
||
|
|
"127.0.0.1:7006": struct{}{},
|
||
|
|
"127.0.0.1:7007": struct{}{},
|
||
|
|
}))
|
||
|
|
})
|
||
|
|
|
||
|
|
It("should check if reload is due", func() {
|
||
|
|
subject._reload = 0
|
||
|
|
Expect(subject._reload).To(Equal(uint32(0)))
|
||
|
|
subject.forceReload()
|
||
|
|
Expect(subject._reload).To(Equal(uint32(1)))
|
||
|
|
})
|
||
|
|
})
|