mirror of
https://github.com/redis/go-redis.git
synced 2025-07-31 05:04:23 +03:00
Migrates tests to ginkgo/gomega
This commit is contained in:
Dimitrij Denissenko
279
sentinel_test.go
279
sentinel_test.go
@ -5,181 +5,118 @@ import (
|
||||
"os"
|
||||
"os/exec"
|
||||
"path/filepath"
|
||||
"testing"
|
||||
"text/template"
|
||||
"time"
|
||||
|
||||
. "github.com/onsi/ginkgo"
|
||||
. "github.com/onsi/gomega"
|
||||
"gopkg.in/redis.v2"
|
||||
)
|
||||
|
||||
func startRedis(port string) (*exec.Cmd, error) {
|
||||
cmd := exec.Command("redis-server", "--port", port)
|
||||
if false {
|
||||
cmd.Stdout = os.Stdout
|
||||
cmd.Stderr = os.Stderr
|
||||
}
|
||||
if err := cmd.Start(); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return cmd, nil
|
||||
}
|
||||
var _ = Describe("Sentinel", func() {
|
||||
|
||||
func startRedisSlave(port, slave string) (*exec.Cmd, error) {
|
||||
cmd := exec.Command("redis-server", "--port", port, "--slaveof", "127.0.0.1", slave)
|
||||
if false {
|
||||
cmd.Stdout = os.Stdout
|
||||
cmd.Stderr = os.Stderr
|
||||
}
|
||||
if err := cmd.Start(); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return cmd, nil
|
||||
}
|
||||
const masterName = "mymaster"
|
||||
const masterPort = "8123"
|
||||
const sentinelPort = "8124"
|
||||
const sentinelConf = `
|
||||
port ` + sentinelPort + `
|
||||
|
||||
func startRedisSentinel(port, masterName, masterPort string) (*exec.Cmd, error) {
|
||||
dir, err := ioutil.TempDir("", "sentinel")
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
sentinelConfFilepath := filepath.Join(dir, "sentinel.conf")
|
||||
tpl, err := template.New("sentinel.conf").Parse(sentinelConf)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
data := struct {
|
||||
Port string
|
||||
MasterName string
|
||||
MasterPort string
|
||||
}{
|
||||
Port: port,
|
||||
MasterName: masterName,
|
||||
MasterPort: masterPort,
|
||||
}
|
||||
if err := writeTemplateToFile(sentinelConfFilepath, tpl, data); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
cmd := exec.Command("redis-server", sentinelConfFilepath, "--sentinel")
|
||||
if true {
|
||||
cmd.Stdout = os.Stdout
|
||||
cmd.Stderr = os.Stderr
|
||||
}
|
||||
if err := cmd.Start(); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return cmd, nil
|
||||
}
|
||||
|
||||
func writeTemplateToFile(path string, t *template.Template, data interface{}) error {
|
||||
f, err := os.Create(path)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer f.Close()
|
||||
return t.Execute(f, data)
|
||||
}
|
||||
|
||||
func TestSentinel(t *testing.T) {
|
||||
masterName := "mymaster"
|
||||
masterPort := "8123"
|
||||
slavePort := "8124"
|
||||
sentinelPort := "8125"
|
||||
|
||||
masterCmd, err := startRedis(masterPort)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
defer masterCmd.Process.Kill()
|
||||
|
||||
// Wait for master to start.
|
||||
time.Sleep(200 * time.Millisecond)
|
||||
|
||||
master := redis.NewTCPClient(&redis.Options{
|
||||
Addr: ":" + masterPort,
|
||||
})
|
||||
if err := master.Ping().Err(); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
slaveCmd, err := startRedisSlave(slavePort, masterPort)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
defer slaveCmd.Process.Kill()
|
||||
|
||||
// Wait for slave to start.
|
||||
time.Sleep(200 * time.Millisecond)
|
||||
|
||||
slave := redis.NewTCPClient(&redis.Options{
|
||||
Addr: ":" + slavePort,
|
||||
})
|
||||
if err := slave.Ping().Err(); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
sentinelCmd, err := startRedisSentinel(sentinelPort, masterName, masterPort)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
defer sentinelCmd.Process.Kill()
|
||||
|
||||
// Wait for sentinel to start.
|
||||
time.Sleep(200 * time.Millisecond)
|
||||
|
||||
sentinel := redis.NewTCPClient(&redis.Options{
|
||||
Addr: ":" + sentinelPort,
|
||||
})
|
||||
if err := sentinel.Ping().Err(); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
defer sentinel.Shutdown()
|
||||
|
||||
client := redis.NewFailoverClient(&redis.FailoverOptions{
|
||||
MasterName: masterName,
|
||||
SentinelAddrs: []string{":" + sentinelPort},
|
||||
})
|
||||
|
||||
if err := client.Set("foo", "master").Err(); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
val, err := master.Get("foo").Result()
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if val != "master" {
|
||||
t.Fatalf(`got %q, expected "master"`, val)
|
||||
}
|
||||
|
||||
// Kill Redis master.
|
||||
if err := masterCmd.Process.Kill(); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if err := master.Ping().Err(); err == nil {
|
||||
t.Fatalf("master was not killed")
|
||||
}
|
||||
|
||||
// Wait for Redis sentinel to elect new master.
|
||||
time.Sleep(5 * time.Second)
|
||||
|
||||
// Check that client picked up new master.
|
||||
val, err = client.Get("foo").Result()
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if val != "master" {
|
||||
t.Fatalf(`got %q, expected "master"`, val)
|
||||
}
|
||||
}
|
||||
|
||||
var sentinelConf = `
|
||||
port {{ .Port }}
|
||||
|
||||
sentinel monitor {{ .MasterName }} 127.0.0.1 {{ .MasterPort }} 1
|
||||
sentinel down-after-milliseconds {{ .MasterName }} 1000
|
||||
sentinel failover-timeout {{ .MasterName }} 2000
|
||||
sentinel parallel-syncs {{ .MasterName }} 1
|
||||
sentinel monitor ` + masterName + ` 127.0.0.1 ` + masterPort + ` 1
|
||||
sentinel down-after-milliseconds ` + masterName + ` 400
|
||||
sentinel failover-timeout ` + masterName + ` 800
|
||||
sentinel parallel-syncs ` + masterName + ` 1
|
||||
`
|
||||
|
||||
var runCmd = func(name string, args ...string) *os.Process {
|
||||
cmd := exec.Command(name, args...)
|
||||
if false {
|
||||
cmd.Stdout = os.Stdout
|
||||
cmd.Stderr = os.Stderr
|
||||
}
|
||||
err := cmd.Start()
|
||||
Expect(err).NotTo(HaveOccurred())
|
||||
return cmd.Process
|
||||
}
|
||||
|
||||
var connect = func(port string) *redis.Client {
|
||||
client := redis.NewTCPClient(&redis.Options{
|
||||
Addr: ":" + port,
|
||||
})
|
||||
Eventually(func() error {
|
||||
return client.Ping().Err()
|
||||
}, "1s", "100ms").ShouldNot(HaveOccurred())
|
||||
return client
|
||||
}
|
||||
|
||||
var startMaster = func() (*redis.Client, *os.Process) {
|
||||
proc := runCmd("redis-server", "--port", masterPort)
|
||||
return connect(masterPort), proc
|
||||
}
|
||||
|
||||
var startSlave = func(port string) (*redis.Client, *os.Process) {
|
||||
proc := runCmd("redis-server", "--port", port, "--slaveof", "127.0.0.1", masterPort)
|
||||
return connect(port), proc
|
||||
}
|
||||
|
||||
var startSentinel = func() *os.Process {
|
||||
dir, err := ioutil.TempDir("", "sentinel")
|
||||
Expect(err).NotTo(HaveOccurred())
|
||||
|
||||
fname := filepath.Join(dir, "sentinel.conf")
|
||||
err = ioutil.WriteFile(fname, []byte(sentinelConf), 0664)
|
||||
Expect(err).NotTo(HaveOccurred())
|
||||
|
||||
proc := runCmd("redis-server", fname, "--sentinel")
|
||||
client := connect(sentinelPort)
|
||||
client.Close()
|
||||
return proc
|
||||
}
|
||||
|
||||
It("should facilitate failover", func() {
|
||||
master, mproc := startMaster()
|
||||
defer mproc.Kill()
|
||||
slave1, sproc1 := startSlave("8125")
|
||||
defer sproc1.Kill()
|
||||
slave2, sproc2 := startSlave("8126")
|
||||
defer sproc2.Kill()
|
||||
sntproc := startSentinel()
|
||||
defer sntproc.Kill()
|
||||
|
||||
client := redis.NewFailoverClient(&redis.FailoverOptions{
|
||||
MasterName: masterName,
|
||||
SentinelAddrs: []string{":" + sentinelPort},
|
||||
})
|
||||
defer client.Close()
|
||||
|
||||
// Set value on master, verify
|
||||
err := client.Set("foo", "master").Err()
|
||||
Expect(err).NotTo(HaveOccurred())
|
||||
|
||||
val, err := master.Get("foo").Result()
|
||||
Expect(err).NotTo(HaveOccurred())
|
||||
Expect(val).To(Equal("master"))
|
||||
|
||||
// Wait until replicated
|
||||
Eventually(func() string {
|
||||
return slave1.Get("foo").Val()
|
||||
}, "1s", "100ms").Should(Equal("master"))
|
||||
Eventually(func() string {
|
||||
return slave2.Get("foo").Val()
|
||||
}, "1s", "100ms").Should(Equal("master"))
|
||||
|
||||
// Kill master.
|
||||
master.Shutdown()
|
||||
Eventually(func() error {
|
||||
return master.Ping().Err()
|
||||
}, "5s", "100ms").Should(HaveOccurred())
|
||||
|
||||
// Wait for Redis sentinel to elect new master.
|
||||
Eventually(func() string {
|
||||
return slave1.Info().Val() + slave2.Info().Val()
|
||||
}, "30s", "500ms").Should(ContainSubstring("role:master"))
|
||||
|
||||
// Check that client picked up new master.
|
||||
val, err = client.Get("foo").Result()
|
||||
Expect(err).NotTo(HaveOccurred())
|
||||
Expect(val).To(Equal("master"))
|
||||
})
|
||||
|
||||
})
|
||||
|
||||
Reference in New Issue
Block a user