From 58c7dacf5be97bc5411a409ce3d77e12fdcd437e Mon Sep 17 00:00:00 2001
From: Vladimir Mihailenco <vladimir.webdev@gmail.com>
Date: Wed, 8 Apr 2015 12:40:45 +0300
Subject: [PATCH] Restore cluster_client_test.go

---
 cluster_client_test.go | 66 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 66 insertions(+)
 create mode 100644 cluster_client_test.go

diff --git a/cluster_client_test.go b/cluster_client_test.go
new file mode 100644
index 0000000..ad5c101
--- /dev/null
+++ b/cluster_client_test.go
@@ -0,0 +1,66 @@
+package redis
+
+import (
+	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/gomega"
+)
+
+var _ = Describe("ClusterClient", func() {
+	var subject *ClusterClient
+
+	var populate = func() {
+		subject.setSlots([]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 = 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._reload).To(Equal(uint32(1)))
+	})
+
+	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([]string{
+			"127.0.0.1:6379",
+			"127.0.0.1:7003",
+			"127.0.0.1:7006",
+			"127.0.0.1:7000",
+			"127.0.0.1:7004",
+			"127.0.0.1:7007",
+			"127.0.0.1:7001",
+			"127.0.0.1:7005",
+			"127.0.0.1:7002",
+		}))
+	})
+
+	It("should check if reload is due", func() {
+		subject._reload = 0
+		Expect(subject._reload).To(Equal(uint32(0)))
+		subject.scheduleReload()
+		Expect(subject._reload).To(Equal(uint32(1)))
+	})
+})