cluster_test.go

package redis

import (
	"testing"
	"fmt"
	"time"
	"strings"

	"github.com/stretchr/testify/assert"
)

const (
	addr = "100.81.164.186:40991;100.81.164.186:40992;100.81.164.186:40993"
)

func TestChooseNodeWithCmd(t *testing.T) {
	// test ChooseNodeWithCmd

	var nr int

	// simple test
	{
		fmt.Printf("TestChooseNodeWithCmd case %d.\n", nr)
		nr++

		cluster, err := NewCluster(
			&Options{
				StartNodes:   strings.Split(addr, ";"),
				ConnTimeout:  5 * time.Second,
				KeepAlive:    32,
				AliveTime:    10 * time.Second,
			})
		assert.Equal(t, nil, err, "should be equal")

		node, err := cluster.ChooseNodeWithCmd("set", "a", 1)
		assert.Equal(t, nil, err, "should be equal")
		expect, err := cluster.getNodeByKey("a")
		assert.Equal(t, nil, err, "should be equal")
		assert.Equal(t, node, expect, "should be equal")
	}

	// test mset
	{
		fmt.Printf("TestChooseNodeWithCmd case %d.\n", nr)
		nr++

		cluster, err := NewCluster(
			&Options{
				StartNodes:   strings.Split(addr, ";"),
				ConnTimeout:  5 * time.Second,
				KeepAlive:    32,
				AliveTime:    10 * time.Second,
			})
		assert.Equal(t, nil, err, "should be equal")

		node, err := cluster.ChooseNodeWithCmd("mset", "a", 1, "a", 2)
		assert.Equal(t, nil, err, "should be equal")
		expect, err := cluster.getNodeByKey("a")
		assert.Equal(t, nil, err, "should be equal")
		assert.Equal(t, node, expect, "should be equal")

		node, err = cluster.ChooseNodeWithCmd("mset", "a", 1, "b", 2)
		assert.NotEqual(t, nil, err, "should be equal")

		node, err = cluster.ChooseNodeWithCmd("mset", "a", 1, "d", 2)
		assert.Equal(t, nil, err, "should be equal")
		assert.Equal(t, node, expect, "should be equal")
	}
}