mirror of https://github.com/tidwall/tile38.git
Better FOLLOW/MONITOR tests
This commit is contained in:
parent
8e0bf1957d
commit
77b4efa55f
|
@ -21,6 +21,8 @@ func follower_follow_test(mc *mockServer) error {
|
|||
Do("SET", "mykey", "truck4", "POINT", 10, 10).OK(),
|
||||
Do("SET", "mykey", "truck5", "POINT", 10, 10).OK(),
|
||||
Do("SET", "mykey", "truck6", "POINT", 10, 10).OK(),
|
||||
Do("CONFIG", "SET", "requirepass", "1234").OK(),
|
||||
Do("AUTH", "1234").OK(),
|
||||
)
|
||||
if err != nil {
|
||||
return err
|
||||
|
@ -31,34 +33,36 @@ func follower_follow_test(mc *mockServer) error {
|
|||
Do("GET", "mykey2", "truck1").Str(`{"type":"Point","coordinates":[10,10]}`),
|
||||
Do("GET", "mykey2", "truck2").Str(`{"type":"Point","coordinates":[10,10]}`),
|
||||
|
||||
Do("CONFIG", "SET", "leaderauth", "1234").OK(),
|
||||
Do("FOLLOW", "localhost", mc.port).OK(),
|
||||
Do("GET", "mykey2", "truck1").Err("catching up to leader"),
|
||||
Sleep(time.Second),
|
||||
Do("GET", "mykey2", "truck1").Err(`<nil>`),
|
||||
Do("GET", "mykey2", "truck2").Err(`<nil>`),
|
||||
Do("GET", "mykey", "truck1").Err("catching up to leader"),
|
||||
Sleep(time.Second/2),
|
||||
|
||||
Do("GET", "mykey", "truck1").Err(`{"type":"Point","coordinates":[10,10]}`),
|
||||
Do("GET", "mykey", "truck2").Err(`{"type":"Point","coordinates":[10,10]}`),
|
||||
)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// err = mc.DoBatch(
|
||||
// Do("SET", "mykey", "truck7", "POINT", 10, 10).OK(),
|
||||
// Do("SET", "mykey", "truck8", "POINT", 10, 10).OK(),
|
||||
// Do("SET", "mykey", "truck9", "POINT", 10, 10).OK(),
|
||||
// )
|
||||
// if err != nil {
|
||||
// return err
|
||||
// }
|
||||
err = mc.DoBatch(
|
||||
Do("SET", "mykey", "truck7", "POINT", 10, 10).OK(),
|
||||
Do("SET", "mykey", "truck8", "POINT", 10, 10).OK(),
|
||||
Do("SET", "mykey", "truck9", "POINT", 10, 10).OK(),
|
||||
)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// err = mc2.DoBatch(
|
||||
// Sleep(time.Second/2),
|
||||
// Do("GET", "mykey1", "truck7").Str(`{"type":"Point","coordinates":[10,10]}`),
|
||||
// Do("GET", "mykey1", "truck8").Str(`{"type":"Point","coordinates":[10,10]}`),
|
||||
// Do("GET", "mykey1", "truck9").Str(`{"type":"Point","coordinates":[10,10]}`),
|
||||
// )
|
||||
// if err != nil {
|
||||
// return err
|
||||
// }
|
||||
err = mc2.DoBatch(
|
||||
Sleep(time.Second/2),
|
||||
Do("GET", "mykey", "truck7").Str(`{"type":"Point","coordinates":[10,10]}`),
|
||||
Do("GET", "mykey", "truck8").Str(`{"type":"Point","coordinates":[10,10]}`),
|
||||
Do("GET", "mykey", "truck9").Str(`{"type":"Point","coordinates":[10,10]}`),
|
||||
)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
|
|
@ -0,0 +1,77 @@
|
|||
package tests
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"strings"
|
||||
"sync"
|
||||
|
||||
"github.com/gomodule/redigo/redis"
|
||||
)
|
||||
|
||||
func subTestMonitor(g *testGroup) {
|
||||
g.regSubTest("monitor", follower_monitor_test)
|
||||
}
|
||||
|
||||
func follower_monitor_test(mc *mockServer) error {
|
||||
N := 1000
|
||||
ch := make(chan error)
|
||||
var wg sync.WaitGroup
|
||||
wg.Add(1)
|
||||
go func() {
|
||||
ch <- func() error {
|
||||
conn, err := redis.Dial("tcp", fmt.Sprintf("localhost:%d", mc.port))
|
||||
if err != nil {
|
||||
wg.Done()
|
||||
return err
|
||||
}
|
||||
defer conn.Close()
|
||||
s, err := redis.String(conn.Do("MONITOR"))
|
||||
if err != nil {
|
||||
wg.Done()
|
||||
return err
|
||||
}
|
||||
if s != "OK" {
|
||||
wg.Done()
|
||||
return fmt.Errorf("expected '%s', got '%s'", "OK", s)
|
||||
}
|
||||
wg.Done()
|
||||
|
||||
for i := 0; i < N; i++ {
|
||||
s, err := redis.String(conn.Receive())
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
ex := fmt.Sprintf(`"mykey" "%d"`, i)
|
||||
if !strings.Contains(s, ex) {
|
||||
return fmt.Errorf("expected '%s', got '%s'", ex, s)
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}()
|
||||
}()
|
||||
|
||||
wg.Wait()
|
||||
|
||||
conn, err := redis.Dial("tcp", fmt.Sprintf("localhost:%d", mc.port))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer conn.Close()
|
||||
|
||||
for i := 0; i < N; i++ {
|
||||
s, err := redis.String(conn.Do("SET", "mykey", i, "POINT", 10, 10))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if s != "OK" {
|
||||
return fmt.Errorf("expected '%s', got '%s'", "OK", s)
|
||||
}
|
||||
}
|
||||
|
||||
err = <-ch
|
||||
if err != nil {
|
||||
err = fmt.Errorf("monitor client: %w", err)
|
||||
}
|
||||
|
||||
return err
|
||||
}
|
|
@ -54,8 +54,9 @@ func TestIntegration(t *testing.T) {
|
|||
regTestGroup("info", subTestInfo)
|
||||
regTestGroup("timeouts", subTestTimeout)
|
||||
regTestGroup("metrics", subTestMetrics)
|
||||
regTestGroup("aof", subTestAOF)
|
||||
regTestGroup("follower", subTestFollower)
|
||||
regTestGroup("aof", subTestAOF)
|
||||
regTestGroup("monitor", subTestMonitor)
|
||||
runTestGroups(t)
|
||||
}
|
||||
|
||||
|
|
Loading…
Reference in New Issue