tile38/tests/client_test.go

package tests

import (
	"errors"
	"fmt"
	"strings"
	"testing"

	"github.com/gomodule/redigo/redis"
	"github.com/tidwall/gjson"
	"github.com/tidwall/pretty"
)

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)
	}

	_, err := conns[1].Do("CLIENT", "setname", "cl1")
	if err != nil {
		return err
	}
	_, err = conns[2].Do("CLIENT", "setname", "cl2")
	if err != nil {
		return err
	}

	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(pretty.Pretty(bres))
	if int(gjson.Get(sres, "list.#").Int()) < numConns {
		return errors.New("Invalid number of connections")
	}

	client13ID := gjson.Get(sres, "list.13.id").String()
	client14Addr := gjson.Get(sres, "list.14.addr").String()
	client15Addr := gjson.Get(sres, "list.15.addr").String()

	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(``),
		Do("CLIENT", "setname", "abc").OK(),
		Do("CLIENT", "getname").Str(`abc`),
		Do("CLIENT", "getname").JSON().Str(`{"ok":true,"name":"abc"}`),
		Do("CLIENT", "setname", "abc", "efg").Err(`wrong number of arguments for 'client' command`),
		Do("CLIENT", "setname", " abc ").Err(`Client names cannot contain spaces, newlines or special characters.`),
		Do("CLIENT", "setname", "abcd").JSON().OK(),
		Do("CLIENT", "kill", "name", "abcd").Err("No such client"),
		Do("CLIENT", "getname").Str(`abcd`),
		Do("CLIENT", "kill").Err(`wrong number of arguments for 'client' command`),
		Do("CLIENT", "kill", "").Err(`No such client`),
		Do("CLIENT", "kill", "abcd").Err(`No such client`),
		Do("CLIENT", "kill", "id", client13ID).OK(),
		Do("CLIENT", "kill", "id").Err("wrong number of arguments for 'client' command"),
		Do("CLIENT", "kill", client14Addr).OK(),
		Do("CLIENT", "kill", client14Addr, "yikes").Err("wrong number of arguments for 'client' command"),
		Do("CLIENT", "kill", "addr").Err("wrong number of arguments for 'client' command"),
		Do("CLIENT", "kill", "addr", client15Addr).JSON().OK(),
		Do("CLIENT", "kill", "addr", client14Addr, "yikes").Err("wrong number of arguments for 'client' command"),
		Do("CLIENT", "kill", "id", "1000").Err("No such client"),
	)

}