mirror of https://github.com/tidwall/tile38.git
103 lines
2.5 KiB
Go
103 lines
2.5 KiB
Go
package tests
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/gomodule/redigo/redis"
|
|
"github.com/tidwall/gjson"
|
|
)
|
|
|
|
func subTestClient(t *testing.T, mc *mockServer) {
|
|
runStep(t, mc, "OUTPUT", client_OUTPUT_test)
|
|
runStep(t, mc, "CLIENT", client_CLIENT_test)
|
|
}
|
|
|
|
func client_OUTPUT_test(mc *mockServer) error {
|
|
if err := mc.DoBatch(
|
|
// tests removal of "elapsed" member.
|
|
Do("OUTPUT", "json").Str(`{"ok":true}`),
|
|
Do("OUTPUT", "resp").OK(),
|
|
); err != nil {
|
|
return err
|
|
}
|
|
|
|
// run direct commands
|
|
if _, err := mc.Do("OUTPUT", "json"); err != nil {
|
|
return err
|
|
}
|
|
res, err := mc.Do("CLIENT", "list")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
bres, ok := res.([]byte)
|
|
if !ok {
|
|
return errors.New("Failed to type assert CLIENT response")
|
|
}
|
|
sres := string(bres)
|
|
if !gjson.Valid(sres) {
|
|
return errors.New("CLIENT response was invalid")
|
|
}
|
|
info := gjson.Get(sres, "list").String()
|
|
if !gjson.Valid(info) {
|
|
return errors.New("CLIENT.list response was invalid")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func client_CLIENT_test(mc *mockServer) error {
|
|
numConns := 20
|
|
var conns []redis.Conn
|
|
defer func() {
|
|
for i := range conns {
|
|
conns[i].Close()
|
|
}
|
|
}()
|
|
for i := 0; i <= numConns; i++ {
|
|
conn, err := redis.Dial("tcp", fmt.Sprintf(":%d", mc.port))
|
|
if err != nil {
|
|
return err
|
|
}
|
|
conns = append(conns, conn)
|
|
}
|
|
if _, err := mc.Do("OUTPUT", "JSON"); err != nil {
|
|
return err
|
|
}
|
|
res, err := mc.Do("CLIENT", "list")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
bres, ok := res.([]byte)
|
|
if !ok {
|
|
return errors.New("Failed to type assert CLIENT response")
|
|
}
|
|
sres := string(bres)
|
|
if len(gjson.Get(sres, "list").Array()) < numConns {
|
|
return errors.New("Invalid number of connections")
|
|
}
|
|
|
|
return mc.DoBatch(
|
|
Do("CLIENT", "list").JSON().Func(func(s string) error {
|
|
if int(gjson.Get(s, "list.#").Int()) < numConns {
|
|
return errors.New("Invalid number of connections")
|
|
}
|
|
return nil
|
|
}),
|
|
Do("CLIENT", "list").Func(func(s string) error {
|
|
if len(strings.Split(strings.TrimSpace(s), "\n")) < numConns {
|
|
return errors.New("Invalid number of connections")
|
|
}
|
|
return nil
|
|
}),
|
|
Do("CLIENT").Err(`wrong number of arguments for 'client' command`),
|
|
Do("CLIENT", "hello").Err(`Syntax error, try CLIENT (LIST | KILL | GETNAME | SETNAME)`),
|
|
Do("CLIENT", "list", "arg3").Err(`wrong number of arguments for 'client' command`),
|
|
Do("CLIENT", "getname", "arg3").Err(`wrong number of arguments for 'client' command`),
|
|
Do("CLIENT", "getname").JSON().Str(`{"ok":true,"name":""}`),
|
|
Do("CLIENT", "getname").Str(``),
|
|
)
|
|
|
|
}
|